1 files changed, 116 insertions, 0 deletions
diff --git a/third_party/libwebrtc/modules/audio_processing/aec3/config_selector_unittest.cc b/third_party/libwebrtc/modules/audio_processing/aec3/config_selector_unittest.cc
new file mode 100644
index 0000000000..1826bfcace
--- /dev/null
+++ b/third_party/libwebrtc/modules/audio_processing/aec3/config_selector_unittest.cc
@@ -0,0 +1,116 @@
+/*
+ *  Copyright (c) 2022 The WebRTC project authors. All Rights Reserved.
+ *
+ *  Use of this source code is governed by a BSD-style license
+ *  that can be found in the LICENSE file in the root of the source
+ *  tree. An additional intellectual property rights grant can be found
+ *  in the file PATENTS.  All contributing project authors may
+ *  be found in the AUTHORS file in the root of the source tree.
+ */
+
+#include "modules/audio_processing/aec3/config_selector.h"
+
+#include <tuple>
+
+#include "absl/types/optional.h"
+#include "api/audio/echo_canceller3_config.h"
+#include "test/gtest.h"
+
+namespace webrtc {
+
+class ConfigSelectorChannelsAndContentDetection
+    : public ::testing::Test,
+      public ::testing::WithParamInterface<std::tuple<int, bool>> {};
+
+INSTANTIATE_TEST_SUITE_P(ConfigSelectorMultiParameters,
+                         ConfigSelectorChannelsAndContentDetection,
+                         ::testing::Combine(::testing::Values(1, 2, 8),
+                                            ::testing::Values(false, true)));
+
+class ConfigSelectorChannels : public ::testing::Test,
+                               public ::testing::WithParamInterface<int> {};
+
+INSTANTIATE_TEST_SUITE_P(ConfigSelectorMultiParameters,
+                         ConfigSelectorChannels,
+                         ::testing::Values(1, 2, 8));
+
+TEST_P(ConfigSelectorChannelsAndContentDetection,
+       MonoConfigIsSelectedWhenNoMultiChannelConfigPresent) {
+  const auto [num_channels, detect_stereo_content] = GetParam();
+  EchoCanceller3Config config;
+  config.multi_channel.detect_stereo_content = detect_stereo_content;
+  absl::optional<EchoCanceller3Config> multichannel_config;
+
+  config.delay.default_delay = config.delay.default_delay + 1;
+  const size_t custom_delay_value_in_config = config.delay.default_delay;
+
+  ConfigSelector cs(config, multichannel_config,
+                    /*num_render_input_channels=*/num_channels);
+  EXPECT_EQ(cs.active_config().delay.default_delay,
+            custom_delay_value_in_config);
+
+  cs.Update(/*multichannel_content=*/false);
+  EXPECT_EQ(cs.active_config().delay.default_delay,
+            custom_delay_value_in_config);
+
+  cs.Update(/*multichannel_content=*/true);
+  EXPECT_EQ(cs.active_config().delay.default_delay,
+            custom_delay_value_in_config);
+}
+
+TEST_P(ConfigSelectorChannelsAndContentDetection,
+       CorrectInitialConfigIsSelected) {
+  const auto [num_channels, detect_stereo_content] = GetParam();
+  EchoCanceller3Config config;
+  config.multi_channel.detect_stereo_content = detect_stereo_content;
+  absl::optional<EchoCanceller3Config> multichannel_config = config;
+
+  config.delay.default_delay += 1;
+  const size_t custom_delay_value_in_config = config.delay.default_delay;
+  multichannel_config->delay.default_delay += 2;
+  const size_t custom_delay_value_in_multichannel_config =
+      multichannel_config->delay.default_delay;
+
+  ConfigSelector cs(config, multichannel_config,
+                    /*num_render_input_channels=*/num_channels);
+
+  if (num_channels == 1 || detect_stereo_content) {
+    EXPECT_EQ(cs.active_config().delay.default_delay,
+              custom_delay_value_in_config);
+  } else {
+    EXPECT_EQ(cs.active_config().delay.default_delay,
+              custom_delay_value_in_multichannel_config);
+  }
+}
+
+TEST_P(ConfigSelectorChannels, CorrectConfigUpdateBehavior) {
+  const int num_channels = GetParam();
+  EchoCanceller3Config config;
+  config.multi_channel.detect_stereo_content = true;
+  absl::optional<EchoCanceller3Config> multichannel_config = config;
+
+  config.delay.default_delay += 1;
+  const size_t custom_delay_value_in_config = config.delay.default_delay;
+  multichannel_config->delay.default_delay += 2;
+  const size_t custom_delay_value_in_multichannel_config =
+      multichannel_config->delay.default_delay;
+
+  ConfigSelector cs(config, multichannel_config,
+                    /*num_render_input_channels=*/num_channels);
+
+  cs.Update(/*multichannel_content=*/false);
+  EXPECT_EQ(cs.active_config().delay.default_delay,
+            custom_delay_value_in_config);
+
+  if (num_channels == 1) {
+    cs.Update(/*multichannel_content=*/false);
+    EXPECT_EQ(cs.active_config().delay.default_delay,
+              custom_delay_value_in_config);
+  } else {
+    cs.Update(/*multichannel_content=*/true);
+    EXPECT_EQ(cs.active_config().delay.default_delay,
+              custom_delay_value_in_multichannel_config);
+  }
+}
+
+}  // namespace webrtc