Implement audio constraints for PeerConneciton API

content/renderer/media/media_stream_impl_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 "base/memory/scoped_ptr.h"
 #include "base/message_loop.h"
 #include "base/utf_string_conversions.h"
 #include "content/renderer/media/media_stream_extra_data.h"
 #include "content/renderer/media/media_stream_impl.h"
 #include "content/renderer/media/mock_media_stream_dependency_factory.h"
@@ skipping 73 matching lines @@
     dependency_factory_.reset(new MockMediaStreamDependencyFactory());
     ms_impl_.reset(new MediaStreamImplUnderTest(ms_dispatcher_.get(),
                                                 vc_manager.get(),
                                                 dependency_factory_.get()));
   }
 
   WebKit::WebMediaStreamDescriptor RequestLocalMediaStream(bool audio,
                                                            bool video) {
     ms_impl_->RequestUserMedia(audio, video);
     FakeMediaStreamDispatcherComplete();
-    if (video) {
-      // TODO(perkj): Only change the video source at the moment since audio
-      // sources are not implemented.
-      ChangeSourceStateToLive();
-    }
+    if (video)
+      ChangeVideoSourceStateToLive();
+    if (audio)
+      ChangeAudioSourceStateToLive();
+
     EXPECT_EQ(MediaStreamImplUnderTest::REQUEST_SUCCEEDED,
               ms_impl_->request_state());
 
     WebKit::WebMediaStreamDescriptor desc = ms_impl_->last_generated_stream();
     content::MediaStreamExtraData* extra_data =
         static_cast<content::MediaStreamExtraData*>(desc.extraData());
     if (!extra_data || !extra_data->local_stream()) {
       ADD_FAILURE();
       return desc;
     }
@@ skipping 10 matching lines @@
   }
 
 
   void FakeMediaStreamDispatcherComplete() {
     ms_impl_->OnStreamGenerated(ms_dispatcher_->request_id(),
                                 ms_dispatcher_->stream_label(),
                                 ms_dispatcher_->audio_array(),
                                 ms_dispatcher_->video_array());
   }
 
-  void ChangeSourceStateToLive() {
+  void ChangeVideoSourceStateToLive() {
     if (dependency_factory_->last_video_source() != NULL) {
       dependency_factory_->last_video_source()->SetLive();
     }
   }
 
-  void ChangeSourceStateToEnded() {
+  void ChangeAudioSourceStateToLive() {
+    if (dependency_factory_->last_audio_source() != NULL) {
+      dependency_factory_->last_audio_source()->SetLive();
+    }
+  }
+
+  void ChangeVideoSourceStateToEnded() {
     if (dependency_factory_->last_video_source() != NULL) {
       dependency_factory_->last_video_source()->SetEnded();
     }
   }
 
+  void ChangeAudioSourceStateToEnded() {
+    if (dependency_factory_->last_audio_source() != NULL) {
+      dependency_factory_->last_audio_source()->SetEnded();
+    }
+  }
+
  protected:
   scoped_ptr<MockMediaStreamDispatcher> ms_dispatcher_;
   scoped_ptr<MediaStreamImplUnderTest> ms_impl_;
   scoped_ptr<MockMediaStreamDependencyFactory> dependency_factory_;
 };
 
 TEST_F(MediaStreamImplTest, LocalMediaStream) {
   // Test a stream with both audio and video.
   WebKit::WebMediaStreamDescriptor mixed_desc = RequestLocalMediaStream(true,
                                                                         true);
@@ skipping 25 matching lines @@
   // Test that the MediaStreams are deleted if the owning WebFrame is deleted.
   // In the unit test the owning frame is NULL.
   ms_impl_->FrameWillClose(NULL);
   EXPECT_EQ(3, ms_dispatcher_->stop_stream_counter());
 }
 
 // This test what happens if a source to a MediaSteam fails to start.
 TEST_F(MediaStreamImplTest, MediaSourceFailToStart) {
   ms_impl_->RequestUserMedia(true, true);
   FakeMediaStreamDispatcherComplete();
-  ChangeSourceStateToEnded();
+  ChangeVideoSourceStateToEnded();
+  ChangeAudioSourceStateToEnded();
   EXPECT_EQ(MediaStreamImplUnderTest::REQUEST_FAILED,
             ms_impl_->request_state());
   EXPECT_EQ(1, ms_dispatcher_->request_stream_counter());
   EXPECT_EQ(1, ms_dispatcher_->stop_stream_counter());
 }
 
 // This test what happens if MediaStreamImpl is deleted while the sources of a
 // MediaStream is being started.
 TEST_F(MediaStreamImplTest, MediaStreamImplShutDown) {
   ms_impl_->RequestUserMedia(true, true);
   FakeMediaStreamDispatcherComplete();
   EXPECT_EQ(1, ms_dispatcher_->request_stream_counter());
   EXPECT_EQ(MediaStreamImplUnderTest::REQUEST_NOT_COMPLETE,
             ms_impl_->request_state());
   ms_impl_.reset();
-  ChangeSourceStateToLive();
+  ChangeAudioSourceStateToLive();
+  ChangeVideoSourceStateToLive();
 }
 
 // This test what happens if the WebFrame is closed while the MediaStream is
 // being generated by the MediaStreamDispatcher.
 TEST_F(MediaStreamImplTest, ReloadFrameWhileGeneratingStream) {
   ms_impl_->RequestUserMedia(true, true);
   ms_impl_->FrameWillClose(NULL);
   EXPECT_EQ(1, ms_dispatcher_->request_stream_counter());
   EXPECT_EQ(0, ms_dispatcher_->stop_stream_counter());
-  ChangeSourceStateToLive();
+  ChangeAudioSourceStateToLive();
+  ChangeVideoSourceStateToLive();
   EXPECT_EQ(MediaStreamImplUnderTest::REQUEST_NOT_COMPLETE,
             ms_impl_->request_state());
 }
 
 // This test what happens if the WebFrame is closed while the sources are being
 // started by MediaStreamDependencyFactory.
 TEST_F(MediaStreamImplTest, ReloadFrameWhileGeneratingSources) {
   ms_impl_->RequestUserMedia(true, true);
   FakeMediaStreamDispatcherComplete();
   EXPECT_EQ(0, ms_dispatcher_->stop_stream_counter());
   EXPECT_EQ(1, ms_dispatcher_->request_stream_counter());
   ms_impl_->FrameWillClose(NULL);
   EXPECT_EQ(1, ms_dispatcher_->stop_stream_counter());
-  ChangeSourceStateToLive();
+  ChangeAudioSourceStateToLive();
+  ChangeVideoSourceStateToLive();
   EXPECT_EQ(MediaStreamImplUnderTest::REQUEST_NOT_COMPLETE,
             ms_impl_->request_state());
 }
 
 // This test what happens if stop is called on a stream after the frame has
 // been reloaded.
 TEST_F(MediaStreamImplTest, StopStreamAfterReload) {
   WebKit::WebMediaStreamDescriptor mixed_desc = RequestLocalMediaStream(true,
                                                                         true);
   EXPECT_EQ(0, ms_dispatcher_->stop_stream_counter());
   EXPECT_EQ(1, ms_dispatcher_->request_stream_counter());
   ms_impl_->FrameWillClose(NULL);
   EXPECT_EQ(1, ms_dispatcher_->stop_stream_counter());
   ms_impl_->OnLocalMediaStreamStop(mixed_desc.label().utf8());
   EXPECT_EQ(1, ms_dispatcher_->stop_stream_counter());
 }
 
 }  // namespace content