OLD | NEW |
---|---|
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include <string> | 5 #include <string> |
6 | 6 |
7 #include "base/memory/scoped_ptr.h" | 7 #include "base/memory/scoped_ptr.h" |
8 #include "base/message_loop.h" | 8 #include "base/message_loop.h" |
9 #include "base/utf_string_conversions.h" | 9 #include "base/utf_string_conversions.h" |
10 #include "content/renderer/media/mock_media_stream_dependency_factory.h" | 10 #include "content/renderer/media/mock_media_stream_dependency_factory.h" |
11 #include "content/renderer/media/mock_media_stream_impl.h" | 11 #include "content/renderer/media/mock_media_stream_impl.h" |
12 #include "content/renderer/media/mock_web_peer_connection_handler_client.h" | 12 #include "content/renderer/media/mock_web_peer_connection_handler_client.h" |
13 #include "content/renderer/media/mock_peer_connection_impl.h" | 13 #include "content/renderer/media/mock_peer_connection_impl.h" |
14 #include "content/renderer/media/peer_connection_handler.h" | 14 #include "content/renderer/media/peer_connection_handler.h" |
15 #include "content/renderer/media/rtc_video_decoder.h" | 15 #include "content/renderer/media/rtc_video_decoder.h" |
16 #include "jingle/glue/thread_wrapper.h" | 16 #include "jingle/glue/thread_wrapper.h" |
17 #include "testing/gtest/include/gtest/gtest.h" | 17 #include "testing/gtest/include/gtest/gtest.h" |
18 #include "third_party/libjingle/source/talk/app/webrtcv1/peerconnection.h" | 18 #include "third_party/libjingle/source/talk/app/webrtc/peerconnection.h" |
19 //#include "third_party/libjingle/source/talk/base/scoped_ref_ptr.h" | |
tommi (sloooow) - chröme
2012/01/26 15:38:06
remove
Henrik Grunell
2012/01/27 10:23:43
Done.
| |
19 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebMediaStre amDescriptor.h" | 20 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebMediaStre amDescriptor.h" |
20 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebMediaStre amSource.h" | 21 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebMediaStre amSource.h" |
21 | 22 |
23 namespace webrtc { | |
24 | |
tommi (sloooow) - chröme
2012/01/26 15:38:06
run lint (I think it'll complain about this)
Henrik Grunell
2012/01/27 10:23:43
It doesn't. What was your concern?
tommi (sloooow) - chröme
2012/01/27 10:50:53
nevermind me, confused the 'no whitespace at top o
| |
25 class MockVideoRendererWrapper : public VideoRendererWrapperInterface { | |
26 public: | |
27 virtual cricket::VideoRenderer* renderer() OVERRIDE { return NULL; } | |
28 | |
29 protected: | |
30 virtual ~MockVideoRendererWrapper() {} | |
31 }; | |
32 | |
33 } // namespace webrtc | |
34 | |
22 TEST(PeerConnectionHandlerTest, Basic) { | 35 TEST(PeerConnectionHandlerTest, Basic) { |
23 MessageLoop loop; | 36 MessageLoop loop; |
24 | 37 |
25 scoped_ptr<WebKit::MockWebPeerConnectionHandlerClient> mock_client( | 38 scoped_ptr<WebKit::MockWebPeerConnectionHandlerClient> mock_client( |
26 new WebKit::MockWebPeerConnectionHandlerClient()); | 39 new WebKit::MockWebPeerConnectionHandlerClient()); |
27 scoped_refptr<MockMediaStreamImpl> mock_ms_impl(new MockMediaStreamImpl()); | 40 scoped_refptr<MockMediaStreamImpl> mock_ms_impl(new MockMediaStreamImpl()); |
28 scoped_ptr<MockMediaStreamDependencyFactory> mock_dependency_factory( | 41 scoped_ptr<MockMediaStreamDependencyFactory> mock_dependency_factory( |
29 new MockMediaStreamDependencyFactory()); | 42 new MockMediaStreamDependencyFactory()); |
30 mock_dependency_factory->CreatePeerConnectionFactory(NULL, NULL); | 43 mock_dependency_factory->CreatePeerConnectionFactory(NULL, |
44 NULL, | |
45 NULL, | |
46 NULL, | |
47 NULL); | |
31 scoped_ptr<PeerConnectionHandler> pc_handler( | 48 scoped_ptr<PeerConnectionHandler> pc_handler( |
32 new PeerConnectionHandler(mock_client.get(), | 49 new PeerConnectionHandler(mock_client.get(), |
33 mock_ms_impl.get(), | 50 mock_ms_impl.get(), |
34 mock_dependency_factory.get(), | 51 mock_dependency_factory.get())); |
35 NULL, | |
36 NULL, | |
37 NULL, | |
38 NULL)); | |
39 | 52 |
40 WebKit::WebString server_config( | 53 WebKit::WebString server_config( |
41 WebKit::WebString::fromUTF8("STUN stun.l.google.com:19302")); | 54 WebKit::WebString::fromUTF8("STUN stun.l.google.com:19302")); |
42 WebKit::WebSecurityOrigin security_origin; | 55 WebKit::WebSecurityOrigin security_origin; |
43 pc_handler->initialize(server_config, security_origin); | 56 pc_handler->initialize(server_config, security_origin); |
44 EXPECT_TRUE(pc_handler->native_peer_connection_.get()); | 57 EXPECT_TRUE(pc_handler->native_peer_connection_.get()); |
45 webrtc::MockPeerConnectionImpl* mock_peer_connection = | 58 webrtc::MockPeerConnectionImpl* mock_peer_connection = |
46 static_cast<webrtc::MockPeerConnectionImpl*>( | 59 static_cast<webrtc::MockPeerConnectionImpl*>( |
47 pc_handler->native_peer_connection_.get()); | 60 pc_handler->native_peer_connection_.get()); |
48 EXPECT_EQ(static_cast<webrtc::PeerConnectionObserver*>(pc_handler.get()), | |
49 mock_peer_connection->observer()); | |
50 | 61 |
51 std::string label("label"); | 62 // TODO(grunell): Add an audio track as well. |
63 std::string stream_label("stream-label"); | |
64 std::string video_track_label("video-label"); | |
65 talk_base::scoped_refptr<webrtc::LocalVideoTrackInterface> local_video_track( | |
66 mock_dependency_factory->CreateLocalVideoTrack(video_track_label, NULL)); | |
67 mock_ms_impl->AddTrack(video_track_label, local_video_track); | |
52 WebKit::WebVector<WebKit::WebMediaStreamSource> source_vector( | 68 WebKit::WebVector<WebKit::WebMediaStreamSource> source_vector( |
53 static_cast<size_t>(1)); | 69 static_cast<size_t>(1)); |
54 source_vector[0].initialize(WebKit::WebString::fromUTF8(label), | 70 source_vector[0].initialize(WebKit::WebString::fromUTF8(video_track_label), |
55 WebKit::WebMediaStreamSource::TypeVideo, | 71 WebKit::WebMediaStreamSource::TypeVideo, |
56 WebKit::WebString::fromUTF8("RemoteVideo")); | 72 WebKit::WebString::fromUTF8("RemoteVideo")); |
57 WebKit::WebVector<WebKit::WebMediaStreamDescriptor> pendingAddStreams( | 73 WebKit::WebVector<WebKit::WebMediaStreamDescriptor> local_streams( |
58 static_cast<size_t>(1)); | 74 static_cast<size_t>(1)); |
59 pendingAddStreams[0].initialize(UTF8ToUTF16(label), source_vector); | 75 local_streams[0].initialize(UTF8ToUTF16(stream_label), source_vector); |
60 pc_handler->produceInitialOffer(pendingAddStreams); | 76 pc_handler->produceInitialOffer(local_streams); |
61 EXPECT_EQ(label, mock_ms_impl->video_label()); | 77 EXPECT_EQ(stream_label, mock_peer_connection->stream_label()); |
62 EXPECT_EQ(label, mock_peer_connection->stream_id()); | 78 EXPECT_TRUE(mock_peer_connection->stream_changes_committed()); |
63 EXPECT_TRUE(mock_peer_connection->video_stream()); | |
64 EXPECT_TRUE(mock_peer_connection->connected()); | |
65 EXPECT_TRUE(mock_peer_connection->video_capture_set()); | |
66 | 79 |
67 std::string message("message1"); | 80 std::string message("message1"); |
68 pc_handler->handleInitialOffer(WebKit::WebString::fromUTF8(message)); | 81 pc_handler->handleInitialOffer(WebKit::WebString::fromUTF8(message)); |
69 EXPECT_EQ(message, mock_peer_connection->signaling_message()); | 82 EXPECT_EQ(message, mock_peer_connection->signaling_message()); |
70 | 83 |
71 message = "message2"; | 84 message = "message2"; |
72 pc_handler->processSDP(WebKit::WebString::fromUTF8(message)); | 85 pc_handler->processSDP(WebKit::WebString::fromUTF8(message)); |
73 EXPECT_EQ(message, mock_peer_connection->signaling_message()); | 86 EXPECT_EQ(message, mock_peer_connection->signaling_message()); |
74 | 87 |
75 message = "message3"; | 88 message = "message3"; |
76 pc_handler->OnSignalingMessage(message); | 89 pc_handler->OnSignalingMessage(message); |
77 EXPECT_EQ(message, mock_client->sdp()); | 90 EXPECT_EQ(message, mock_client->sdp()); |
78 | 91 |
79 std::string remote_label(label); | 92 std::string remote_stream_label(stream_label); |
80 remote_label.append("-remote"); | 93 remote_stream_label += "-remote"; |
81 pc_handler->OnAddStream(remote_label, true); | 94 std::string remote_video_track_label(video_track_label); |
82 EXPECT_EQ(remote_label, mock_client->stream_label()); | 95 remote_video_track_label += "-remote"; |
96 // We use a local stream as a remote since for testing purposes we really | |
97 // only need the MediaStreamInterface. | |
98 talk_base::scoped_refptr<webrtc::LocalMediaStreamInterface> remote_stream( | |
99 mock_dependency_factory->CreateLocalMediaStream(remote_stream_label)); | |
100 talk_base::scoped_refptr<webrtc::LocalVideoTrackInterface> remote_video_track( | |
101 mock_dependency_factory->CreateLocalVideoTrack(remote_video_track_label, | |
102 NULL)); | |
103 remote_video_track->set_enabled(true); | |
104 remote_stream->AddTrack(remote_video_track); | |
105 mock_peer_connection->AddRemoteStream(remote_stream); | |
106 pc_handler->OnAddStream(remote_stream); | |
107 EXPECT_EQ(remote_stream_label, mock_client->stream_label()); | |
83 | 108 |
84 scoped_refptr<RTCVideoDecoder> rtc_video_decoder( | 109 talk_base::scoped_refptr<webrtc::MockVideoRendererWrapper> renderer( |
85 new RTCVideoDecoder(&loop, "")); | 110 new talk_base::RefCountedObject<webrtc::MockVideoRendererWrapper>()); |
86 pc_handler->SetVideoRenderer(label, rtc_video_decoder.get()); | 111 pc_handler->SetVideoRenderer(remote_stream_label, renderer); |
87 EXPECT_EQ(label, mock_peer_connection->video_renderer_stream_id()); | 112 EXPECT_EQ(renderer, static_cast<webrtc::MockLocalVideoTrack*>( |
113 remote_video_track.get())->renderer()); | |
88 | 114 |
89 pc_handler->OnRemoveStream(remote_label, true); | 115 WebKit::WebVector<WebKit::WebMediaStreamDescriptor> empty_streams( |
116 static_cast<size_t>(0)); | |
117 pc_handler->processPendingStreams(empty_streams, local_streams); | |
118 EXPECT_EQ("", mock_peer_connection->stream_label()); | |
119 mock_peer_connection->ClearStreamChangesCommitted(); | |
120 EXPECT_TRUE(!mock_peer_connection->stream_changes_committed()); | |
121 | |
122 pc_handler->OnRemoveStream(remote_stream); | |
90 EXPECT_TRUE(mock_client->stream_label().empty()); | 123 EXPECT_TRUE(mock_client->stream_label().empty()); |
91 | 124 |
125 pc_handler->processPendingStreams(local_streams, empty_streams); | |
126 EXPECT_EQ(stream_label, mock_peer_connection->stream_label()); | |
127 EXPECT_TRUE(mock_peer_connection->stream_changes_committed()); | |
128 | |
92 pc_handler->stop(); | 129 pc_handler->stop(); |
93 EXPECT_FALSE(pc_handler->native_peer_connection_.get()); | 130 EXPECT_FALSE(pc_handler->native_peer_connection_.get()); |
94 // PC handler is expected to be deleted when stop calls | 131 // PC handler is expected to be deleted when stop calls |
95 // MediaStreamImpl::ClosePeerConnection. We own and delete it here instead of | 132 // MediaStreamImpl::ClosePeerConnection. We own and delete it here instead of |
96 // in the mock. | 133 // in the mock. |
97 pc_handler.reset(); | 134 pc_handler.reset(); |
98 | |
99 // processPendingStreams must be tested on a new PC handler since removing | |
100 // streams is currently not supported. | |
101 pc_handler.reset(new PeerConnectionHandler(mock_client.get(), | |
102 mock_ms_impl.get(), | |
103 mock_dependency_factory.get(), | |
104 NULL, | |
105 NULL, | |
106 NULL, | |
107 NULL)); | |
108 pc_handler->initialize(server_config, security_origin); | |
109 EXPECT_TRUE(pc_handler->native_peer_connection_.get()); | |
110 mock_peer_connection = static_cast<webrtc::MockPeerConnectionImpl*>( | |
111 pc_handler->native_peer_connection_.get()); | |
112 | |
113 WebKit::WebVector<WebKit::WebMediaStreamDescriptor> pendingRemoveStreams( | |
114 static_cast<size_t>(0)); | |
115 pc_handler->processPendingStreams(pendingAddStreams, pendingRemoveStreams); | |
116 EXPECT_EQ(label, mock_ms_impl->video_label()); | |
117 EXPECT_EQ(label, mock_peer_connection->stream_id()); | |
118 EXPECT_TRUE(mock_peer_connection->video_stream()); | |
119 EXPECT_TRUE(mock_peer_connection->connected()); | |
120 EXPECT_TRUE(mock_peer_connection->video_capture_set()); | |
121 | |
122 pc_handler->stop(); | |
123 pc_handler.reset(); | |
124 } | 135 } |
OLD | NEW |