Detect AudioInputStream muting and propagate to MediaStreamAudioSource.

media/audio/audio_input_controller_unittest.cc

 // Copyright (c) 2012 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 "media/audio/audio_input_controller.h"
 #include "base/bind.h"
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "base/single_thread_task_runner.h"
 #include "base/synchronization/waitable_event.h"
 #include "base/test/test_timeouts.h"
 #include "base/threading/thread.h"
 #include "media/audio/audio_device_description.h"
 #include "media/audio/audio_thread_impl.h"
+#include "media/audio/fake_audio_input_stream.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using ::testing::_;
 using ::testing::AnyNumber;
 using ::testing::AtLeast;
 using ::testing::Exactly;
 using ::testing::InvokeWithoutArgs;
 using ::testing::NotNull;
 using base::WaitableEvent;
@@ skipping 28 matching lines @@
     : public AudioInputController::EventHandler {
  public:
   MockAudioInputControllerEventHandler() {}
 
   MOCK_METHOD1(OnCreated, void(AudioInputController* controller));
   MOCK_METHOD2(OnError, void(AudioInputController* controller,
                              AudioInputController::ErrorCode error_code));
   MOCK_METHOD2(OnLog,
                void(AudioInputController* controller,
                     const std::string& message));
+  MOCK_METHOD2(OnMuted, void(AudioInputController* controller, bool is_muted));
 
  private:
   DISALLOW_COPY_AND_ASSIGN(MockAudioInputControllerEventHandler);
 };
 
 class MockSyncWriter : public AudioInputController::SyncWriter {
  public:
   MockSyncWriter() {}
 
   MOCK_METHOD4(Write,
@@ skipping 141 matching lines @@
 
   controller->Record();
 
   controller->Close(base::MessageLoop::QuitWhenIdleClosure());
   base::RunLoop().Run();
 
   controller->Close(base::MessageLoop::QuitWhenIdleClosure());
   base::RunLoop().Run();
 }
 
+// Test that AudioInputController sends OnMute callbacks properly.
+TEST_F(AudioInputControllerTest, TestOnmutedCallbackInitiallyUnmuted) {
+  base::RunLoop run_loop;
+
+  MockAudioInputControllerEventHandler event_handler;
+  MockSyncWriter sync_writer;
+  scoped_refptr<AudioInputController> controller;
+  WaitableEvent callback_event(WaitableEvent::ResetPolicy::AUTOMATIC,
+                               WaitableEvent::InitialState::NOT_SIGNALED);
+
+  AudioParameters params(AudioParameters::AUDIO_FAKE, kChannelLayout,
+                         kSampleRate, kBitsPerSample, kSamplesPerPacket);
+
+  SuspendAudioThread();
+  FakeAudioInputStream::SetGlobalMutedState(false);
+  controller = AudioInputController::Create(
+      audio_manager_.get(), &event_handler, &sync_writer, nullptr, params,
+      AudioDeviceDescription::kDefaultDeviceId, false);
+  ASSERT_TRUE(controller.get());
+  EXPECT_CALL(event_handler, OnCreated(controller.get())).Times(Exactly(1));
+  EXPECT_CALL(event_handler, OnLog(controller.get(), _));
+  EXPECT_CALL(sync_writer, Close()).Times(Exactly(1));
+  EXPECT_CALL(event_handler, OnMuted(controller.get(), true))
+      .WillOnce(InvokeWithoutArgs([&] { callback_event.Signal(); }));
+  EXPECT_CALL(event_handler, OnMuted(controller.get(), false))
+      .WillOnce(InvokeWithoutArgs([&] { callback_event.Signal(); }));
+  ResumeAudioThread();
+
+  // AudioInputController will poll once every second, so wait at most a bit
+  // more than that for the callbacks.
+  FakeAudioInputStream::SetGlobalMutedState(true);
+  callback_event.TimedWait(base::TimeDelta::FromMilliseconds(1500));
+  FakeAudioInputStream::SetGlobalMutedState(false);
+  callback_event.TimedWait(base::TimeDelta::FromMilliseconds(1500));
+
+  CloseAudioController(controller.get());
+}
+
+TEST_F(AudioInputControllerTest, TestOnmutedCallbackInitiallyMuted) {
+  base::RunLoop run_loop;
+
+  MockAudioInputControllerEventHandler event_handler;
+  MockSyncWriter sync_writer;
+  scoped_refptr<AudioInputController> controller;
+  WaitableEvent callback_event(WaitableEvent::ResetPolicy::AUTOMATIC,
+                               WaitableEvent::InitialState::NOT_SIGNALED);
+
+  AudioParameters params(AudioParameters::AUDIO_FAKE, kChannelLayout,
+                         kSampleRate, kBitsPerSample, kSamplesPerPacket);
+
+  SuspendAudioThread();
+  FakeAudioInputStream::SetGlobalMutedState(true);
+  controller = AudioInputController::Create(
+      audio_manager_.get(), &event_handler, &sync_writer, nullptr, params,
+      AudioDeviceDescription::kDefaultDeviceId, false);
+  ASSERT_TRUE(controller.get());
+  EXPECT_CALL(event_handler, OnCreated(controller.get())).Times(Exactly(1));
+  EXPECT_CALL(event_handler, OnLog(controller.get(), _));
+  EXPECT_CALL(sync_writer, Close()).Times(Exactly(1));
+  EXPECT_CALL(event_handler, OnMuted(controller.get(), true))
+      .WillOnce(InvokeWithoutArgs([&] { callback_event.Signal(); }));
+  EXPECT_CALL(event_handler, OnMuted(controller.get(), false))
+      .WillOnce(InvokeWithoutArgs([&] { callback_event.Signal(); }));
+  ResumeAudioThread();
+
+  // AudioInputController will poll once every second, so wait at most a bit
+  // more than that for the callbacks.
+  callback_event.TimedWait(base::TimeDelta::FromMilliseconds(1500));
+  FakeAudioInputStream::SetGlobalMutedState(false);
+  callback_event.TimedWait(base::TimeDelta::FromMilliseconds(1500));
+
+  CloseAudioController(controller.get());
+}
+
 }  // namespace media