Move SODA client unit test to Chromium repo

This CL moves the SodaClientUnitTest from the internal SODA repo to the
Chromium repo.

Bug: 1124873
Change-Id: Ib2dabb242f21817a117f379fb0a1b017dd5bcffc
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2443218Reviewed-by: Dale Curtis <dalecurtis@chromium.org>
Commit-Queue: Evan Liu <evliu@google.com>
Cr-Commit-Position: refs/heads/master@{#812968}

chrome/services/speech/soda/BUILD.gn

@@ -11,3 +11,16 @@ source_set("soda") {
 
   deps = [ "//base" ]
 }
+
+source_set("unit_tests") {
+  testonly = true
+
+  sources = [ "soda_client_unittest.cc" ]
+
+  deps = [
+    ":soda",
+    "//base",
+    "//media",
+    "//testing/gtest",
+  ]
+}

chrome/services/speech/soda/soda_client_unittest.cc

+// Copyright 2020 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+#include "chrome/services/speech/soda/soda_client.h"
+
+#include <unistd.h>
+#include <memory>
+
+#include "base/files/file_path.h"
+#include "base/files/file_util.h"
+#include "media/audio/wav_audio_handler.h"
+#include "media/base/audio_bus.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace soda {
+
+constexpr base::FilePath::CharType kSodaResourcesDir[] =
+    FILE_PATH_LITERAL("third_party/soda/resources");
+
+constexpr base::FilePath::CharType kSodaTestBinaryRelativePath[] =
+    FILE_PATH_LITERAL("libsoda_for_testing.so");
+
+constexpr base::FilePath::CharType kSodaTestonfigRelativePath[] =
+    FILE_PATH_LITERAL("en_us");
+
+constexpr base::FilePath::CharType kSodaTestAudioRelativePath[] =
+    FILE_PATH_LITERAL("hey_google.wav");
+
+class SodaClientUnitTest : public testing::Test {
+ public:
+  SodaClientUnitTest() = default;
+  ~SodaClientUnitTest() override = default;
+
+  static void RecognitionCallback(const char* result,
+                                  const bool is_final,
+                                  void* callback_handle);
+
+  void AddRecognitionResult(std::string result);
+
+ protected:
+  void SetUp() override;
+
+  std::unique_ptr<soda::SodaClient> soda_client_;
+  std::vector<std::string> recognition_results_;
+};
+
+// static
+void SodaClientUnitTest::RecognitionCallback(const char* result,
+                                             const bool is_final,
+                                             void* callback_handle) {
+  ASSERT_TRUE(callback_handle);
+  static_cast<soda::SodaClientUnitTest*>(callback_handle)
+      ->AddRecognitionResult(result);
+}
+
+void SodaClientUnitTest::AddRecognitionResult(std::string result) {
+  recognition_results_.push_back(std::move(result));
+}
+
+void SodaClientUnitTest::SetUp() {
+  auto libsoda_path = base::FilePath(kSodaResourcesDir)
+                          .Append(base::FilePath(kSodaTestBinaryRelativePath));
+  ASSERT_TRUE(base::PathExists(libsoda_path));
+  soda_client_ = std::make_unique<soda::SodaClient>(libsoda_path);
+  ASSERT_TRUE(soda_client_.get());
+  ASSERT_FALSE(soda_client_->IsInitialized());
+}
+
+TEST_F(SodaClientUnitTest, CreateSodaClient) {
+  auto audio_file = base::FilePath(kSodaResourcesDir)
+                        .Append(base::FilePath(kSodaTestAudioRelativePath));
+  ASSERT_TRUE(base::PathExists(audio_file));
+
+  std::string buffer;
+  ASSERT_TRUE(base::ReadFileToString(audio_file, &buffer));
+
+  auto handler = media::WavAudioHandler::Create(buffer);
+  ASSERT_TRUE(handler.get());
+  ASSERT_EQ(handler->num_channels(), 1);
+
+  auto config_file_path =
+      base::FilePath(kSodaResourcesDir)
+          .Append(base::FilePath(kSodaTestonfigRelativePath));
+  ASSERT_TRUE(base::PathExists(config_file_path));
+  SodaConfig config;
+  config.channel_count = handler->num_channels();
+  config.sample_rate = handler->sample_rate();
+  config.language_pack_directory = config_file_path.value().c_str();
+  config.callback = SodaClientUnitTest::RecognitionCallback;
+  config.callback_handle = this;
+
+  // The test binary does not verify the execution context.
+  config.api_key = "";
+
+  soda_client_->Reset(config);
+  ASSERT_TRUE(soda_client_->IsInitialized());
+
+  auto bus =
+      media::AudioBus::Create(config.channel_count, handler->total_frames());
+
+  size_t bytes_written = 0u;
+  ASSERT_TRUE(handler->CopyTo(bus.get(), 0, &bytes_written));
+
+  std::vector<int16_t> audio_data(bus->frames());
+  bus->ToInterleaved<media::SignedInt16SampleTypeTraits>(bus->frames(),
+                                                         audio_data.data());
+
+  constexpr size_t kMaxChunkSize = 1024;
+  constexpr int kReplayAudioCount = 2;
+
+  for (int i = 0; i < kReplayAudioCount; i++) {
+    int chunk_start = 0;
+    // Upload chunks of 1024 frames at a time.
+    while (chunk_start < static_cast<int>(audio_data.size())) {
+      int chunk_size = kMaxChunkSize < audio_data.size() - chunk_start
+                           ? kMaxChunkSize
+                           : audio_data.size() - chunk_start;
+      soda_client_->AddAudio(
+          reinterpret_cast<const char*>(&audio_data[chunk_start]),
+          sizeof(int16_t) * chunk_size);
+
+      chunk_start += chunk_size;
+
+      // Sleep for 20ms to simulate real-time audio. SODA requires audio
+      // streaming in order to return events.
+      usleep(20000);
+    }
+  }
+
+  ASSERT_GT(static_cast<int>(recognition_results_.size()), kReplayAudioCount);
+  ASSERT_EQ(recognition_results_.back(), "Hey Google Hey Google");
+}
+
+}  // namespace soda

chrome/test/BUILD.gn

@@ -16,6 +16,7 @@ import("//chrome/browser/page_load_metrics/integration_tests/jsdeps.gni")
 import("//chrome/browser/page_load_metrics/integration_tests/sources.gni")
 import("//chrome/common/features.gni")
 import("//chrome/services/machine_learning/features.gni")
+import("//chrome/services/speech/buildflags.gni")
 import("//chrome/test/base/js2gtest.gni")
 import("//chrome/test/include_js_tests.gni")
 import("//chromeos/assistant/assistant.gni")
@@ -4163,6 +4164,10 @@ test("unit_tests") {
         [ "../browser/android/feed/history/feed_history_helper_unittest.cc" ]
   }
 
+  if (enable_soda) {
+    deps += [ "//chrome/services/speech/soda:unit_tests" ]
+  }
+
   if (is_android) {
     sources += [
       "../browser/android/search_permissions/search_permissions_service_unittest.cc",