Create a private API for controlling WebRTC's AEC3

content/renderer/media/media_stream_audio_processor_unittest.cc

 // Copyright 2013 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 <stddef.h>
 #include <stdint.h>
 
 #include <vector>
 
 #include "base/files/file_path.h"
@@ skipping 181 matching lines @@
     EXPECT_FALSE(audio_processing->voice_detection()->is_enabled());
 #else
     EXPECT_TRUE(audio_processing->gain_control()->mode() ==
         webrtc::GainControl::kAdaptiveAnalog);
     EXPECT_TRUE(audio_processing->voice_detection()->is_enabled());
     EXPECT_TRUE(audio_processing->voice_detection()->likelihood() ==
         webrtc::VoiceDetection::kVeryLowLikelihood);
 #endif
   }
 
+  bool GetAec3ConfigState(MediaStreamAudioProcessor* audio_processor) {
+    DCHECK(audio_processor);
+    if (!audio_processor->audio_processing_) {
+      ADD_FAILURE() << "AudioProcessing object missing where it shouldn't be";
+      return false;
+    }
+    return audio_processor->audio_processing_->GetConfig()
+        .echo_canceller3.enabled;
+  }
+
   base::MessageLoop main_thread_message_loop_;
   media::AudioParameters params_;
   MediaStreamDevice::AudioDeviceParameters input_device_params_;
 };
 
 // Test crashing with ASAN on Android. crbug.com/468762
 #if defined(OS_ANDROID) && defined(ADDRESS_SANITIZER)
 #define MAYBE_WithAudioProcessing DISABLED_WithAudioProcessing
 #else
 #define MAYBE_WithAudioProcessing WithAudioProcessing
@@ skipping 373 matching lines @@
   ProcessDataAndVerifyFormat(audio_processor.get(),
                              kAudioProcessingSampleRate,
                              kAudioProcessingNumberOfChannel,
                              kAudioProcessingSampleRate / 100);
 
   // Stop |audio_processor| so that it removes itself from
   // |webrtc_audio_device| and clears its pointer to it.
   audio_processor->Stop();
 }
 
+// Test that the OnAec3Enable method has the desired effect on the APM config.
+TEST_F(MediaStreamAudioProcessorTest, TestAec3Switch) {
+  MockConstraintFactory constraint_factory;
+  scoped_refptr<WebRtcAudioDeviceImpl> webrtc_audio_device(
+      new WebRtcAudioDeviceImpl());
+  scoped_refptr<MediaStreamAudioProcessor> audio_processor(
+      new rtc::RefCountedObject<MediaStreamAudioProcessor>(
+          constraint_factory.CreateWebMediaConstraints(), input_device_params_,
+          webrtc_audio_device.get()));
+
+  audio_processor->OnAec3Enable(true);
+  EXPECT_TRUE(GetAec3ConfigState(audio_processor.get()));
+
+  audio_processor->OnAec3Enable(false);
+  EXPECT_FALSE(GetAec3ConfigState(audio_processor.get()));
+
+  // Stop |audio_processor| so that it removes itself from
+  // |webrtc_audio_device| and clears its pointer to it.
+  audio_processor->Stop();
+}
+
+// Same test as above, but when AEC is disabled in the constrants. The expected
+// outcome is that AEC3 should be disabled in all cases.
+TEST_F(MediaStreamAudioProcessorTest, TestAec3Switch_AecOff) {
+  MockConstraintFactory constraint_factory;
+  // Disable the AEC.
+  constraint_factory.DisableAecAudioConstraints();
+  scoped_refptr<WebRtcAudioDeviceImpl> webrtc_audio_device(
+      new WebRtcAudioDeviceImpl());
+  scoped_refptr<MediaStreamAudioProcessor> audio_processor(
+      new rtc::RefCountedObject<MediaStreamAudioProcessor>(
+          constraint_factory.CreateWebMediaConstraints(), input_device_params_,
+          webrtc_audio_device.get()));
+
+  EXPECT_FALSE(GetAec3ConfigState(audio_processor.get()));
+
+  audio_processor->OnAec3Enable(true);
+  EXPECT_FALSE(GetAec3ConfigState(audio_processor.get()));
+
+  audio_processor->OnAec3Enable(false);
+  EXPECT_FALSE(GetAec3ConfigState(audio_processor.get()));
+
+  // Stop |audio_processor| so that it removes itself from
+  // |webrtc_audio_device| and clears its pointer to it.
+  audio_processor->Stop();
+}
+
 }  // namespace content