deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 1 | /* |
kjellander | b24317b | 2016-02-10 07:54:43 -0800 | [diff] [blame] | 2 | * Copyright 2012 The WebRTC project authors. All Rights Reserved. |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 3 | * |
kjellander | b24317b | 2016-02-10 07:54:43 -0800 | [diff] [blame] | 4 | * Use of this source code is governed by a BSD-style license |
| 5 | * that can be found in the LICENSE file in the root of the source |
| 6 | * tree. An additional intellectual property rights grant can be found |
| 7 | * in the file PATENTS. All contributing project authors may |
| 8 | * be found in the AUTHORS file in the root of the source tree. |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 9 | */ |
| 10 | |
kwiberg | d1fe281 | 2016-04-27 06:47:29 -0700 | [diff] [blame] | 11 | #include <memory> |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 12 | #include <string> |
Tommi | f888bb5 | 2015-12-12 01:37:01 +0100 | [diff] [blame] | 13 | #include <utility> |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 14 | |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 15 | #include "testing/gmock/include/gmock/gmock.h" |
| 16 | #include "testing/gtest/include/gtest/gtest.h" |
Henrik Kjellander | 15583c1 | 2016-02-10 10:53:12 +0100 | [diff] [blame] | 17 | #include "webrtc/api/audiotrack.h" |
| 18 | #include "webrtc/api/mediastream.h" |
| 19 | #include "webrtc/api/remoteaudiosource.h" |
| 20 | #include "webrtc/api/rtpreceiver.h" |
| 21 | #include "webrtc/api/rtpsender.h" |
| 22 | #include "webrtc/api/streamcollection.h" |
nisse | af510af | 2016-03-21 08:20:42 -0700 | [diff] [blame] | 23 | #include "webrtc/api/test/fakevideotracksource.h" |
perkj | 0d3eef2 | 2016-03-09 02:39:17 +0100 | [diff] [blame] | 24 | #include "webrtc/api/videotracksource.h" |
Henrik Kjellander | 15583c1 | 2016-02-10 10:53:12 +0100 | [diff] [blame] | 25 | #include "webrtc/api/videotrack.h" |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 26 | #include "webrtc/base/gunit.h" |
kjellander | a96e2d7 | 2016-02-04 23:52:28 -0800 | [diff] [blame] | 27 | #include "webrtc/media/base/mediachannel.h" |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 28 | |
| 29 | using ::testing::_; |
| 30 | using ::testing::Exactly; |
deadbeef | 5dd42fd | 2016-05-02 16:20:01 -0700 | [diff] [blame] | 31 | using ::testing::InvokeWithoutArgs; |
skvlad | dc1c62c | 2016-03-16 19:07:43 -0700 | [diff] [blame] | 32 | using ::testing::Return; |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 33 | |
| 34 | static const char kStreamLabel1[] = "local_stream_1"; |
| 35 | static const char kVideoTrackId[] = "video_1"; |
| 36 | static const char kAudioTrackId[] = "audio_1"; |
Peter Boström | 0c4e06b | 2015-10-07 12:23:21 +0200 | [diff] [blame] | 37 | static const uint32_t kVideoSsrc = 98; |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 38 | static const uint32_t kVideoSsrc2 = 100; |
Peter Boström | 0c4e06b | 2015-10-07 12:23:21 +0200 | [diff] [blame] | 39 | static const uint32_t kAudioSsrc = 99; |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 40 | static const uint32_t kAudioSsrc2 = 101; |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 41 | |
| 42 | namespace webrtc { |
| 43 | |
| 44 | // Helper class to test RtpSender/RtpReceiver. |
| 45 | class MockAudioProvider : public AudioProviderInterface { |
| 46 | public: |
nisse | ef8b61e | 2016-04-29 06:09:15 -0700 | [diff] [blame] | 47 | // TODO(nisse): Valid overrides commented out, because the gmock |
| 48 | // methods don't use any override declarations, and we want to avoid |
| 49 | // warnings from -Winconsistent-missing-override. See |
| 50 | // http://crbug.com/428099. |
| 51 | ~MockAudioProvider() /* override */ {} |
Tommi | f888bb5 | 2015-12-12 01:37:01 +0100 | [diff] [blame] | 52 | |
solenberg | d4cec0d | 2015-10-09 08:55:48 -0700 | [diff] [blame] | 53 | MOCK_METHOD2(SetAudioPlayout, |
Peter Boström | 0c4e06b | 2015-10-07 12:23:21 +0200 | [diff] [blame] | 54 | void(uint32_t ssrc, |
solenberg | d4cec0d | 2015-10-09 08:55:48 -0700 | [diff] [blame] | 55 | bool enable)); |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 56 | MOCK_METHOD4(SetAudioSend, |
Peter Boström | 0c4e06b | 2015-10-07 12:23:21 +0200 | [diff] [blame] | 57 | void(uint32_t ssrc, |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 58 | bool enable, |
| 59 | const cricket::AudioOptions& options, |
Taylor Brandstetter | 1a018dc | 2016-03-08 12:37:39 -0800 | [diff] [blame] | 60 | cricket::AudioSource* source)); |
Peter Boström | 0c4e06b | 2015-10-07 12:23:21 +0200 | [diff] [blame] | 61 | MOCK_METHOD2(SetAudioPlayoutVolume, void(uint32_t ssrc, double volume)); |
Taylor Brandstetter | db0cd9e | 2016-05-16 11:40:30 -0700 | [diff] [blame^] | 62 | MOCK_CONST_METHOD1(GetAudioRtpSendParameters, RtpParameters(uint32_t ssrc)); |
| 63 | MOCK_METHOD2(SetAudioRtpSendParameters, |
| 64 | bool(uint32_t ssrc, const RtpParameters&)); |
| 65 | MOCK_CONST_METHOD1(GetAudioRtpReceiveParameters, |
| 66 | RtpParameters(uint32_t ssrc)); |
| 67 | MOCK_METHOD2(SetAudioRtpReceiveParameters, |
skvlad | dc1c62c | 2016-03-16 19:07:43 -0700 | [diff] [blame] | 68 | bool(uint32_t ssrc, const RtpParameters&)); |
Tommi | f888bb5 | 2015-12-12 01:37:01 +0100 | [diff] [blame] | 69 | |
nisse | ef8b61e | 2016-04-29 06:09:15 -0700 | [diff] [blame] | 70 | void SetRawAudioSink( |
| 71 | uint32_t, std::unique_ptr<AudioSinkInterface> sink) /* override */ { |
deadbeef | 2d110be | 2016-01-13 12:00:26 -0800 | [diff] [blame] | 72 | sink_ = std::move(sink); |
Tommi | f888bb5 | 2015-12-12 01:37:01 +0100 | [diff] [blame] | 73 | } |
| 74 | |
| 75 | private: |
kwiberg | d1fe281 | 2016-04-27 06:47:29 -0700 | [diff] [blame] | 76 | std::unique_ptr<AudioSinkInterface> sink_; |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 77 | }; |
| 78 | |
| 79 | // Helper class to test RtpSender/RtpReceiver. |
| 80 | class MockVideoProvider : public VideoProviderInterface { |
| 81 | public: |
| 82 | virtual ~MockVideoProvider() {} |
nisse | 2ded9b1 | 2016-04-08 02:23:55 -0700 | [diff] [blame] | 83 | MOCK_METHOD2(SetSource, |
| 84 | bool(uint32_t ssrc, |
| 85 | rtc::VideoSourceInterface<cricket::VideoFrame>* source)); |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 86 | MOCK_METHOD3(SetVideoPlayout, |
Peter Boström | 0c4e06b | 2015-10-07 12:23:21 +0200 | [diff] [blame] | 87 | void(uint32_t ssrc, |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 88 | bool enable, |
nisse | 08582ff | 2016-02-04 01:24:52 -0800 | [diff] [blame] | 89 | rtc::VideoSinkInterface<cricket::VideoFrame>* sink)); |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 90 | MOCK_METHOD3(SetVideoSend, |
Peter Boström | 0c4e06b | 2015-10-07 12:23:21 +0200 | [diff] [blame] | 91 | void(uint32_t ssrc, |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 92 | bool enable, |
| 93 | const cricket::VideoOptions* options)); |
skvlad | dc1c62c | 2016-03-16 19:07:43 -0700 | [diff] [blame] | 94 | |
Taylor Brandstetter | db0cd9e | 2016-05-16 11:40:30 -0700 | [diff] [blame^] | 95 | MOCK_CONST_METHOD1(GetVideoRtpSendParameters, RtpParameters(uint32_t ssrc)); |
| 96 | MOCK_METHOD2(SetVideoRtpSendParameters, |
| 97 | bool(uint32_t ssrc, const RtpParameters&)); |
| 98 | MOCK_CONST_METHOD1(GetVideoRtpReceiveParameters, |
| 99 | RtpParameters(uint32_t ssrc)); |
| 100 | MOCK_METHOD2(SetVideoRtpReceiveParameters, |
skvlad | dc1c62c | 2016-03-16 19:07:43 -0700 | [diff] [blame] | 101 | bool(uint32_t ssrc, const RtpParameters&)); |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 102 | }; |
| 103 | |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 104 | class RtpSenderReceiverTest : public testing::Test { |
| 105 | public: |
| 106 | virtual void SetUp() { |
| 107 | stream_ = MediaStream::Create(kStreamLabel1); |
tommi | 6eca7e3 | 2015-12-15 04:27:11 -0800 | [diff] [blame] | 108 | } |
| 109 | |
nisse | af510af | 2016-03-21 08:20:42 -0700 | [diff] [blame] | 110 | void AddVideoTrack() { |
perkj | a3ede6c | 2016-03-08 01:27:48 +0100 | [diff] [blame] | 111 | rtc::scoped_refptr<VideoTrackSourceInterface> source( |
nisse | af510af | 2016-03-21 08:20:42 -0700 | [diff] [blame] | 112 | FakeVideoTrackSource::Create()); |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 113 | video_track_ = VideoTrack::Create(kVideoTrackId, source); |
| 114 | EXPECT_TRUE(stream_->AddTrack(video_track_)); |
| 115 | } |
| 116 | |
| 117 | void CreateAudioRtpSender() { |
| 118 | audio_track_ = AudioTrack::Create(kAudioTrackId, NULL); |
| 119 | EXPECT_TRUE(stream_->AddTrack(audio_track_)); |
| 120 | EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, true, _, _)); |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 121 | audio_rtp_sender_ = |
| 122 | new AudioRtpSender(stream_->GetAudioTracks()[0], stream_->label(), |
| 123 | &audio_provider_, nullptr); |
| 124 | audio_rtp_sender_->SetSsrc(kAudioSsrc); |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 125 | } |
| 126 | |
| 127 | void CreateVideoRtpSender() { |
nisse | af510af | 2016-03-21 08:20:42 -0700 | [diff] [blame] | 128 | AddVideoTrack(); |
nisse | 2ded9b1 | 2016-04-08 02:23:55 -0700 | [diff] [blame] | 129 | EXPECT_CALL(video_provider_, SetSource(kVideoSsrc, video_track_.get())); |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 130 | EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _)); |
| 131 | video_rtp_sender_ = new VideoRtpSender(stream_->GetVideoTracks()[0], |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 132 | stream_->label(), &video_provider_); |
| 133 | video_rtp_sender_->SetSsrc(kVideoSsrc); |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 134 | } |
| 135 | |
| 136 | void DestroyAudioRtpSender() { |
| 137 | EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, false, _, _)) |
| 138 | .Times(1); |
| 139 | audio_rtp_sender_ = nullptr; |
| 140 | } |
| 141 | |
| 142 | void DestroyVideoRtpSender() { |
nisse | 2ded9b1 | 2016-04-08 02:23:55 -0700 | [diff] [blame] | 143 | EXPECT_CALL(video_provider_, SetSource(kVideoSsrc, NULL)).Times(1); |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 144 | EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _)).Times(1); |
| 145 | video_rtp_sender_ = nullptr; |
| 146 | } |
| 147 | |
| 148 | void CreateAudioRtpReceiver() { |
tommi | 6eca7e3 | 2015-12-15 04:27:11 -0800 | [diff] [blame] | 149 | audio_track_ = AudioTrack::Create( |
Tommi | f888bb5 | 2015-12-12 01:37:01 +0100 | [diff] [blame] | 150 | kAudioTrackId, RemoteAudioSource::Create(kAudioSsrc, NULL)); |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 151 | EXPECT_TRUE(stream_->AddTrack(audio_track_)); |
solenberg | d4cec0d | 2015-10-09 08:55:48 -0700 | [diff] [blame] | 152 | EXPECT_CALL(audio_provider_, SetAudioPlayout(kAudioSsrc, true)); |
perkj | d61bf80 | 2016-03-24 03:16:19 -0700 | [diff] [blame] | 153 | audio_rtp_receiver_ = new AudioRtpReceiver(stream_, kAudioTrackId, |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 154 | kAudioSsrc, &audio_provider_); |
perkj | d61bf80 | 2016-03-24 03:16:19 -0700 | [diff] [blame] | 155 | audio_track_ = audio_rtp_receiver_->audio_track(); |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 156 | } |
| 157 | |
| 158 | void CreateVideoRtpReceiver() { |
perkj | f0dcfe2 | 2016-03-10 18:32:00 +0100 | [diff] [blame] | 159 | EXPECT_CALL(video_provider_, SetVideoPlayout(kVideoSsrc, true, _)); |
| 160 | video_rtp_receiver_ = |
| 161 | new VideoRtpReceiver(stream_, kVideoTrackId, rtc::Thread::Current(), |
| 162 | kVideoSsrc, &video_provider_); |
| 163 | video_track_ = video_rtp_receiver_->video_track(); |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 164 | } |
| 165 | |
| 166 | void DestroyAudioRtpReceiver() { |
solenberg | d4cec0d | 2015-10-09 08:55:48 -0700 | [diff] [blame] | 167 | EXPECT_CALL(audio_provider_, SetAudioPlayout(kAudioSsrc, false)); |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 168 | audio_rtp_receiver_ = nullptr; |
| 169 | } |
| 170 | |
| 171 | void DestroyVideoRtpReceiver() { |
| 172 | EXPECT_CALL(video_provider_, SetVideoPlayout(kVideoSsrc, false, NULL)); |
| 173 | video_rtp_receiver_ = nullptr; |
| 174 | } |
| 175 | |
| 176 | protected: |
| 177 | MockAudioProvider audio_provider_; |
| 178 | MockVideoProvider video_provider_; |
| 179 | rtc::scoped_refptr<AudioRtpSender> audio_rtp_sender_; |
| 180 | rtc::scoped_refptr<VideoRtpSender> video_rtp_sender_; |
| 181 | rtc::scoped_refptr<AudioRtpReceiver> audio_rtp_receiver_; |
| 182 | rtc::scoped_refptr<VideoRtpReceiver> video_rtp_receiver_; |
| 183 | rtc::scoped_refptr<MediaStreamInterface> stream_; |
| 184 | rtc::scoped_refptr<VideoTrackInterface> video_track_; |
| 185 | rtc::scoped_refptr<AudioTrackInterface> audio_track_; |
| 186 | }; |
| 187 | |
| 188 | // Test that |audio_provider_| is notified when an audio track is associated |
| 189 | // and disassociated with an AudioRtpSender. |
| 190 | TEST_F(RtpSenderReceiverTest, AddAndDestroyAudioRtpSender) { |
| 191 | CreateAudioRtpSender(); |
| 192 | DestroyAudioRtpSender(); |
| 193 | } |
| 194 | |
| 195 | // Test that |video_provider_| is notified when a video track is associated and |
| 196 | // disassociated with a VideoRtpSender. |
| 197 | TEST_F(RtpSenderReceiverTest, AddAndDestroyVideoRtpSender) { |
| 198 | CreateVideoRtpSender(); |
| 199 | DestroyVideoRtpSender(); |
| 200 | } |
| 201 | |
| 202 | // Test that |audio_provider_| is notified when a remote audio and track is |
| 203 | // associated and disassociated with an AudioRtpReceiver. |
| 204 | TEST_F(RtpSenderReceiverTest, AddAndDestroyAudioRtpReceiver) { |
| 205 | CreateAudioRtpReceiver(); |
| 206 | DestroyAudioRtpReceiver(); |
| 207 | } |
| 208 | |
| 209 | // Test that |video_provider_| is notified when a remote |
| 210 | // video track is associated and disassociated with a VideoRtpReceiver. |
| 211 | TEST_F(RtpSenderReceiverTest, AddAndDestroyVideoRtpReceiver) { |
| 212 | CreateVideoRtpReceiver(); |
| 213 | DestroyVideoRtpReceiver(); |
| 214 | } |
| 215 | |
| 216 | TEST_F(RtpSenderReceiverTest, LocalAudioTrackDisable) { |
| 217 | CreateAudioRtpSender(); |
| 218 | |
| 219 | EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, false, _, _)); |
| 220 | audio_track_->set_enabled(false); |
| 221 | |
| 222 | EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, true, _, _)); |
| 223 | audio_track_->set_enabled(true); |
| 224 | |
| 225 | DestroyAudioRtpSender(); |
| 226 | } |
| 227 | |
| 228 | TEST_F(RtpSenderReceiverTest, RemoteAudioTrackDisable) { |
| 229 | CreateAudioRtpReceiver(); |
| 230 | |
solenberg | d4cec0d | 2015-10-09 08:55:48 -0700 | [diff] [blame] | 231 | EXPECT_CALL(audio_provider_, SetAudioPlayout(kAudioSsrc, false)); |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 232 | audio_track_->set_enabled(false); |
| 233 | |
solenberg | d4cec0d | 2015-10-09 08:55:48 -0700 | [diff] [blame] | 234 | EXPECT_CALL(audio_provider_, SetAudioPlayout(kAudioSsrc, true)); |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 235 | audio_track_->set_enabled(true); |
| 236 | |
| 237 | DestroyAudioRtpReceiver(); |
| 238 | } |
| 239 | |
| 240 | TEST_F(RtpSenderReceiverTest, LocalVideoTrackDisable) { |
| 241 | CreateVideoRtpSender(); |
| 242 | |
| 243 | EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _)); |
| 244 | video_track_->set_enabled(false); |
| 245 | |
| 246 | EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _)); |
| 247 | video_track_->set_enabled(true); |
| 248 | |
| 249 | DestroyVideoRtpSender(); |
| 250 | } |
| 251 | |
perkj | f0dcfe2 | 2016-03-10 18:32:00 +0100 | [diff] [blame] | 252 | TEST_F(RtpSenderReceiverTest, RemoteVideoTrackState) { |
| 253 | CreateVideoRtpReceiver(); |
| 254 | |
| 255 | EXPECT_EQ(webrtc::MediaStreamTrackInterface::kLive, video_track_->state()); |
| 256 | EXPECT_EQ(webrtc::MediaSourceInterface::kLive, |
| 257 | video_track_->GetSource()->state()); |
| 258 | |
| 259 | DestroyVideoRtpReceiver(); |
| 260 | |
| 261 | EXPECT_EQ(webrtc::MediaStreamTrackInterface::kEnded, video_track_->state()); |
| 262 | EXPECT_EQ(webrtc::MediaSourceInterface::kEnded, |
| 263 | video_track_->GetSource()->state()); |
| 264 | } |
| 265 | |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 266 | TEST_F(RtpSenderReceiverTest, RemoteVideoTrackDisable) { |
| 267 | CreateVideoRtpReceiver(); |
| 268 | |
| 269 | video_track_->set_enabled(false); |
| 270 | |
| 271 | video_track_->set_enabled(true); |
| 272 | |
| 273 | DestroyVideoRtpReceiver(); |
| 274 | } |
| 275 | |
| 276 | TEST_F(RtpSenderReceiverTest, RemoteAudioTrackSetVolume) { |
| 277 | CreateAudioRtpReceiver(); |
| 278 | |
| 279 | double volume = 0.5; |
| 280 | EXPECT_CALL(audio_provider_, SetAudioPlayoutVolume(kAudioSsrc, volume)); |
| 281 | audio_track_->GetSource()->SetVolume(volume); |
| 282 | |
| 283 | // Disable the audio track, this should prevent setting the volume. |
solenberg | d4cec0d | 2015-10-09 08:55:48 -0700 | [diff] [blame] | 284 | EXPECT_CALL(audio_provider_, SetAudioPlayout(kAudioSsrc, false)); |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 285 | audio_track_->set_enabled(false); |
| 286 | audio_track_->GetSource()->SetVolume(1.0); |
| 287 | |
solenberg | d4cec0d | 2015-10-09 08:55:48 -0700 | [diff] [blame] | 288 | EXPECT_CALL(audio_provider_, SetAudioPlayout(kAudioSsrc, true)); |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 289 | audio_track_->set_enabled(true); |
| 290 | |
| 291 | double new_volume = 0.8; |
| 292 | EXPECT_CALL(audio_provider_, SetAudioPlayoutVolume(kAudioSsrc, new_volume)); |
| 293 | audio_track_->GetSource()->SetVolume(new_volume); |
| 294 | |
| 295 | DestroyAudioRtpReceiver(); |
| 296 | } |
| 297 | |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 298 | // Test that provider methods aren't called without both a track and an SSRC. |
| 299 | TEST_F(RtpSenderReceiverTest, AudioSenderWithoutTrackAndSsrc) { |
| 300 | rtc::scoped_refptr<AudioRtpSender> sender = |
| 301 | new AudioRtpSender(&audio_provider_, nullptr); |
| 302 | rtc::scoped_refptr<AudioTrackInterface> track = |
| 303 | AudioTrack::Create(kAudioTrackId, nullptr); |
| 304 | EXPECT_TRUE(sender->SetTrack(track)); |
| 305 | EXPECT_TRUE(sender->SetTrack(nullptr)); |
| 306 | sender->SetSsrc(kAudioSsrc); |
| 307 | sender->SetSsrc(0); |
| 308 | // Just let it get destroyed and make sure it doesn't call any methods on the |
| 309 | // provider interface. |
| 310 | } |
| 311 | |
| 312 | // Test that provider methods aren't called without both a track and an SSRC. |
| 313 | TEST_F(RtpSenderReceiverTest, VideoSenderWithoutTrackAndSsrc) { |
| 314 | rtc::scoped_refptr<VideoRtpSender> sender = |
| 315 | new VideoRtpSender(&video_provider_); |
| 316 | EXPECT_TRUE(sender->SetTrack(video_track_)); |
| 317 | EXPECT_TRUE(sender->SetTrack(nullptr)); |
| 318 | sender->SetSsrc(kVideoSsrc); |
| 319 | sender->SetSsrc(0); |
| 320 | // Just let it get destroyed and make sure it doesn't call any methods on the |
| 321 | // provider interface. |
| 322 | } |
| 323 | |
| 324 | // Test that an audio sender calls the expected methods on the provider once |
| 325 | // it has a track and SSRC, when the SSRC is set first. |
| 326 | TEST_F(RtpSenderReceiverTest, AudioSenderEarlyWarmupSsrcThenTrack) { |
| 327 | rtc::scoped_refptr<AudioRtpSender> sender = |
| 328 | new AudioRtpSender(&audio_provider_, nullptr); |
| 329 | rtc::scoped_refptr<AudioTrackInterface> track = |
| 330 | AudioTrack::Create(kAudioTrackId, nullptr); |
| 331 | sender->SetSsrc(kAudioSsrc); |
| 332 | EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, true, _, _)); |
| 333 | sender->SetTrack(track); |
| 334 | |
| 335 | // Calls expected from destructor. |
| 336 | EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, false, _, _)).Times(1); |
| 337 | } |
| 338 | |
| 339 | // Test that an audio sender calls the expected methods on the provider once |
| 340 | // it has a track and SSRC, when the SSRC is set last. |
| 341 | TEST_F(RtpSenderReceiverTest, AudioSenderEarlyWarmupTrackThenSsrc) { |
| 342 | rtc::scoped_refptr<AudioRtpSender> sender = |
| 343 | new AudioRtpSender(&audio_provider_, nullptr); |
| 344 | rtc::scoped_refptr<AudioTrackInterface> track = |
| 345 | AudioTrack::Create(kAudioTrackId, nullptr); |
| 346 | sender->SetTrack(track); |
| 347 | EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, true, _, _)); |
| 348 | sender->SetSsrc(kAudioSsrc); |
| 349 | |
| 350 | // Calls expected from destructor. |
| 351 | EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, false, _, _)).Times(1); |
| 352 | } |
| 353 | |
| 354 | // Test that a video sender calls the expected methods on the provider once |
| 355 | // it has a track and SSRC, when the SSRC is set first. |
| 356 | TEST_F(RtpSenderReceiverTest, VideoSenderEarlyWarmupSsrcThenTrack) { |
nisse | af510af | 2016-03-21 08:20:42 -0700 | [diff] [blame] | 357 | AddVideoTrack(); |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 358 | rtc::scoped_refptr<VideoRtpSender> sender = |
| 359 | new VideoRtpSender(&video_provider_); |
| 360 | sender->SetSsrc(kVideoSsrc); |
nisse | 2ded9b1 | 2016-04-08 02:23:55 -0700 | [diff] [blame] | 361 | EXPECT_CALL(video_provider_, SetSource(kVideoSsrc, video_track_.get())); |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 362 | EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _)); |
| 363 | sender->SetTrack(video_track_); |
| 364 | |
| 365 | // Calls expected from destructor. |
nisse | 2ded9b1 | 2016-04-08 02:23:55 -0700 | [diff] [blame] | 366 | EXPECT_CALL(video_provider_, SetSource(kVideoSsrc, nullptr)).Times(1); |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 367 | EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _)).Times(1); |
| 368 | } |
| 369 | |
| 370 | // Test that a video sender calls the expected methods on the provider once |
| 371 | // it has a track and SSRC, when the SSRC is set last. |
| 372 | TEST_F(RtpSenderReceiverTest, VideoSenderEarlyWarmupTrackThenSsrc) { |
nisse | af510af | 2016-03-21 08:20:42 -0700 | [diff] [blame] | 373 | AddVideoTrack(); |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 374 | rtc::scoped_refptr<VideoRtpSender> sender = |
| 375 | new VideoRtpSender(&video_provider_); |
| 376 | sender->SetTrack(video_track_); |
nisse | 2ded9b1 | 2016-04-08 02:23:55 -0700 | [diff] [blame] | 377 | EXPECT_CALL(video_provider_, SetSource(kVideoSsrc, video_track_.get())); |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 378 | EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _)); |
| 379 | sender->SetSsrc(kVideoSsrc); |
| 380 | |
| 381 | // Calls expected from destructor. |
nisse | 2ded9b1 | 2016-04-08 02:23:55 -0700 | [diff] [blame] | 382 | EXPECT_CALL(video_provider_, SetSource(kVideoSsrc, nullptr)).Times(1); |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 383 | EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _)).Times(1); |
| 384 | } |
| 385 | |
| 386 | // Test that the sender is disconnected from the provider when its SSRC is |
| 387 | // set to 0. |
| 388 | TEST_F(RtpSenderReceiverTest, AudioSenderSsrcSetToZero) { |
| 389 | rtc::scoped_refptr<AudioTrackInterface> track = |
| 390 | AudioTrack::Create(kAudioTrackId, nullptr); |
| 391 | EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, true, _, _)); |
| 392 | rtc::scoped_refptr<AudioRtpSender> sender = |
| 393 | new AudioRtpSender(track, kStreamLabel1, &audio_provider_, nullptr); |
| 394 | sender->SetSsrc(kAudioSsrc); |
| 395 | |
| 396 | EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, false, _, _)).Times(1); |
| 397 | sender->SetSsrc(0); |
| 398 | |
| 399 | // Make sure it's SetSsrc that called methods on the provider, and not the |
| 400 | // destructor. |
| 401 | EXPECT_CALL(audio_provider_, SetAudioSend(_, _, _, _)).Times(0); |
| 402 | } |
| 403 | |
| 404 | // Test that the sender is disconnected from the provider when its SSRC is |
| 405 | // set to 0. |
| 406 | TEST_F(RtpSenderReceiverTest, VideoSenderSsrcSetToZero) { |
nisse | af510af | 2016-03-21 08:20:42 -0700 | [diff] [blame] | 407 | AddVideoTrack(); |
nisse | 2ded9b1 | 2016-04-08 02:23:55 -0700 | [diff] [blame] | 408 | EXPECT_CALL(video_provider_, SetSource(kVideoSsrc, video_track_.get())); |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 409 | EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _)); |
| 410 | rtc::scoped_refptr<VideoRtpSender> sender = |
| 411 | new VideoRtpSender(video_track_, kStreamLabel1, &video_provider_); |
| 412 | sender->SetSsrc(kVideoSsrc); |
| 413 | |
nisse | 2ded9b1 | 2016-04-08 02:23:55 -0700 | [diff] [blame] | 414 | EXPECT_CALL(video_provider_, SetSource(kVideoSsrc, nullptr)).Times(1); |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 415 | EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _)).Times(1); |
| 416 | sender->SetSsrc(0); |
| 417 | |
| 418 | // Make sure it's SetSsrc that called methods on the provider, and not the |
| 419 | // destructor. |
nisse | 2ded9b1 | 2016-04-08 02:23:55 -0700 | [diff] [blame] | 420 | EXPECT_CALL(video_provider_, SetSource(_, _)).Times(0); |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 421 | EXPECT_CALL(video_provider_, SetVideoSend(_, _, _)).Times(0); |
| 422 | } |
| 423 | |
| 424 | TEST_F(RtpSenderReceiverTest, AudioSenderTrackSetToNull) { |
| 425 | rtc::scoped_refptr<AudioTrackInterface> track = |
| 426 | AudioTrack::Create(kAudioTrackId, nullptr); |
| 427 | EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, true, _, _)); |
| 428 | rtc::scoped_refptr<AudioRtpSender> sender = |
| 429 | new AudioRtpSender(track, kStreamLabel1, &audio_provider_, nullptr); |
| 430 | sender->SetSsrc(kAudioSsrc); |
| 431 | |
deadbeef | 5dd42fd | 2016-05-02 16:20:01 -0700 | [diff] [blame] | 432 | // Expect that SetAudioSend will be called before the reference to the track |
| 433 | // is released. |
| 434 | EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, false, _, nullptr)) |
| 435 | .Times(1) |
| 436 | .WillOnce(InvokeWithoutArgs([&track] { |
| 437 | EXPECT_LT(2, track->AddRef()); |
| 438 | track->Release(); |
| 439 | })); |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 440 | EXPECT_TRUE(sender->SetTrack(nullptr)); |
| 441 | |
| 442 | // Make sure it's SetTrack that called methods on the provider, and not the |
| 443 | // destructor. |
| 444 | EXPECT_CALL(audio_provider_, SetAudioSend(_, _, _, _)).Times(0); |
| 445 | } |
| 446 | |
| 447 | TEST_F(RtpSenderReceiverTest, VideoSenderTrackSetToNull) { |
deadbeef | 5dd42fd | 2016-05-02 16:20:01 -0700 | [diff] [blame] | 448 | rtc::scoped_refptr<VideoTrackSourceInterface> source( |
| 449 | FakeVideoTrackSource::Create()); |
| 450 | rtc::scoped_refptr<VideoTrackInterface> track = |
| 451 | VideoTrack::Create(kVideoTrackId, source); |
| 452 | EXPECT_CALL(video_provider_, SetSource(kVideoSsrc, track.get())); |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 453 | EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _)); |
| 454 | rtc::scoped_refptr<VideoRtpSender> sender = |
deadbeef | 5dd42fd | 2016-05-02 16:20:01 -0700 | [diff] [blame] | 455 | new VideoRtpSender(track, kStreamLabel1, &video_provider_); |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 456 | sender->SetSsrc(kVideoSsrc); |
| 457 | |
deadbeef | 5dd42fd | 2016-05-02 16:20:01 -0700 | [diff] [blame] | 458 | // Expect that SetSource will be called before the reference to the track |
| 459 | // is released. |
| 460 | EXPECT_CALL(video_provider_, SetSource(kVideoSsrc, nullptr)) |
| 461 | .Times(1) |
| 462 | .WillOnce(InvokeWithoutArgs([&track] { |
| 463 | EXPECT_LT(2, track->AddRef()); |
| 464 | track->Release(); |
| 465 | return true; |
| 466 | })); |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 467 | EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _)).Times(1); |
| 468 | EXPECT_TRUE(sender->SetTrack(nullptr)); |
| 469 | |
| 470 | // Make sure it's SetTrack that called methods on the provider, and not the |
| 471 | // destructor. |
nisse | 2ded9b1 | 2016-04-08 02:23:55 -0700 | [diff] [blame] | 472 | EXPECT_CALL(video_provider_, SetSource(_, _)).Times(0); |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 473 | EXPECT_CALL(video_provider_, SetVideoSend(_, _, _)).Times(0); |
| 474 | } |
| 475 | |
| 476 | TEST_F(RtpSenderReceiverTest, AudioSenderSsrcChanged) { |
nisse | af510af | 2016-03-21 08:20:42 -0700 | [diff] [blame] | 477 | AddVideoTrack(); |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 478 | rtc::scoped_refptr<AudioTrackInterface> track = |
| 479 | AudioTrack::Create(kAudioTrackId, nullptr); |
| 480 | EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, true, _, _)); |
| 481 | rtc::scoped_refptr<AudioRtpSender> sender = |
| 482 | new AudioRtpSender(track, kStreamLabel1, &audio_provider_, nullptr); |
| 483 | sender->SetSsrc(kAudioSsrc); |
| 484 | |
| 485 | EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, false, _, _)).Times(1); |
| 486 | EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc2, true, _, _)).Times(1); |
| 487 | sender->SetSsrc(kAudioSsrc2); |
| 488 | |
| 489 | // Calls expected from destructor. |
| 490 | EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc2, false, _, _)).Times(1); |
| 491 | } |
| 492 | |
| 493 | TEST_F(RtpSenderReceiverTest, VideoSenderSsrcChanged) { |
nisse | af510af | 2016-03-21 08:20:42 -0700 | [diff] [blame] | 494 | AddVideoTrack(); |
nisse | 2ded9b1 | 2016-04-08 02:23:55 -0700 | [diff] [blame] | 495 | EXPECT_CALL(video_provider_, SetSource(kVideoSsrc, video_track_.get())); |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 496 | EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _)); |
| 497 | rtc::scoped_refptr<VideoRtpSender> sender = |
| 498 | new VideoRtpSender(video_track_, kStreamLabel1, &video_provider_); |
| 499 | sender->SetSsrc(kVideoSsrc); |
| 500 | |
nisse | 2ded9b1 | 2016-04-08 02:23:55 -0700 | [diff] [blame] | 501 | EXPECT_CALL(video_provider_, SetSource(kVideoSsrc, nullptr)).Times(1); |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 502 | EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _)).Times(1); |
nisse | 2ded9b1 | 2016-04-08 02:23:55 -0700 | [diff] [blame] | 503 | EXPECT_CALL(video_provider_, SetSource(kVideoSsrc2, video_track_.get())); |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 504 | EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc2, true, _)); |
| 505 | sender->SetSsrc(kVideoSsrc2); |
| 506 | |
| 507 | // Calls expected from destructor. |
nisse | 2ded9b1 | 2016-04-08 02:23:55 -0700 | [diff] [blame] | 508 | EXPECT_CALL(video_provider_, SetSource(kVideoSsrc2, nullptr)).Times(1); |
deadbeef | fac0655 | 2015-11-25 11:26:01 -0800 | [diff] [blame] | 509 | EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc2, false, _)).Times(1); |
| 510 | } |
| 511 | |
skvlad | dc1c62c | 2016-03-16 19:07:43 -0700 | [diff] [blame] | 512 | TEST_F(RtpSenderReceiverTest, AudioSenderCanSetParameters) { |
| 513 | CreateAudioRtpSender(); |
| 514 | |
Taylor Brandstetter | db0cd9e | 2016-05-16 11:40:30 -0700 | [diff] [blame^] | 515 | EXPECT_CALL(audio_provider_, GetAudioRtpSendParameters(kAudioSsrc)) |
skvlad | dc1c62c | 2016-03-16 19:07:43 -0700 | [diff] [blame] | 516 | .WillOnce(Return(RtpParameters())); |
Taylor Brandstetter | db0cd9e | 2016-05-16 11:40:30 -0700 | [diff] [blame^] | 517 | EXPECT_CALL(audio_provider_, SetAudioRtpSendParameters(kAudioSsrc, _)) |
skvlad | dc1c62c | 2016-03-16 19:07:43 -0700 | [diff] [blame] | 518 | .WillOnce(Return(true)); |
| 519 | RtpParameters params = audio_rtp_sender_->GetParameters(); |
| 520 | EXPECT_TRUE(audio_rtp_sender_->SetParameters(params)); |
| 521 | |
| 522 | DestroyAudioRtpSender(); |
| 523 | } |
| 524 | |
| 525 | TEST_F(RtpSenderReceiverTest, VideoSenderCanSetParameters) { |
| 526 | CreateVideoRtpSender(); |
| 527 | |
Taylor Brandstetter | db0cd9e | 2016-05-16 11:40:30 -0700 | [diff] [blame^] | 528 | EXPECT_CALL(video_provider_, GetVideoRtpSendParameters(kVideoSsrc)) |
skvlad | dc1c62c | 2016-03-16 19:07:43 -0700 | [diff] [blame] | 529 | .WillOnce(Return(RtpParameters())); |
Taylor Brandstetter | db0cd9e | 2016-05-16 11:40:30 -0700 | [diff] [blame^] | 530 | EXPECT_CALL(video_provider_, SetVideoRtpSendParameters(kVideoSsrc, _)) |
skvlad | dc1c62c | 2016-03-16 19:07:43 -0700 | [diff] [blame] | 531 | .WillOnce(Return(true)); |
| 532 | RtpParameters params = video_rtp_sender_->GetParameters(); |
| 533 | EXPECT_TRUE(video_rtp_sender_->SetParameters(params)); |
| 534 | |
| 535 | DestroyVideoRtpSender(); |
| 536 | } |
| 537 | |
Taylor Brandstetter | db0cd9e | 2016-05-16 11:40:30 -0700 | [diff] [blame^] | 538 | TEST_F(RtpSenderReceiverTest, AudioReceiverCanSetParameters) { |
| 539 | CreateAudioRtpReceiver(); |
| 540 | |
| 541 | EXPECT_CALL(audio_provider_, GetAudioRtpReceiveParameters(kAudioSsrc)) |
| 542 | .WillOnce(Return(RtpParameters())); |
| 543 | EXPECT_CALL(audio_provider_, SetAudioRtpReceiveParameters(kAudioSsrc, _)) |
| 544 | .WillOnce(Return(true)); |
| 545 | RtpParameters params = audio_rtp_receiver_->GetParameters(); |
| 546 | EXPECT_TRUE(audio_rtp_receiver_->SetParameters(params)); |
| 547 | |
| 548 | DestroyAudioRtpReceiver(); |
| 549 | } |
| 550 | |
| 551 | TEST_F(RtpSenderReceiverTest, VideoReceiverCanSetParameters) { |
| 552 | CreateVideoRtpReceiver(); |
| 553 | |
| 554 | EXPECT_CALL(video_provider_, GetVideoRtpReceiveParameters(kVideoSsrc)) |
| 555 | .WillOnce(Return(RtpParameters())); |
| 556 | EXPECT_CALL(video_provider_, SetVideoRtpReceiveParameters(kVideoSsrc, _)) |
| 557 | .WillOnce(Return(true)); |
| 558 | RtpParameters params = video_rtp_receiver_->GetParameters(); |
| 559 | EXPECT_TRUE(video_rtp_receiver_->SetParameters(params)); |
| 560 | |
| 561 | DestroyVideoRtpReceiver(); |
| 562 | } |
| 563 | |
deadbeef | 70ab1a1 | 2015-09-28 16:53:55 -0700 | [diff] [blame] | 564 | } // namespace webrtc |