OLD | NEW |
1 /* | 1 /* |
2 * Copyright 2016 The WebRTC Project Authors. All rights reserved. | 2 * Copyright 2016 The WebRTC Project Authors. All rights reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #include "webrtc/api/trackmediainfomap.h" | 11 #include "webrtc/pc/trackmediainfomap.h" |
12 | 12 |
13 #include <initializer_list> | 13 #include <initializer_list> |
14 #include <memory> | 14 #include <memory> |
15 #include <utility> | 15 #include <utility> |
16 #include <vector> | 16 #include <vector> |
17 | 17 |
18 #include "webrtc/api/audiotrack.h" | |
19 #include "webrtc/api/rtpreceiverinterface.h" | 18 #include "webrtc/api/rtpreceiverinterface.h" |
20 #include "webrtc/api/rtpsenderinterface.h" | 19 #include "webrtc/api/rtpsenderinterface.h" |
21 #include "webrtc/api/test/mock_rtpreceiver.h" | 20 #include "webrtc/api/test/mock_rtpreceiver.h" |
22 #include "webrtc/api/test/mock_rtpsender.h" | 21 #include "webrtc/api/test/mock_rtpsender.h" |
23 #include "webrtc/api/test/fakevideotracksource.h" | |
24 #include "webrtc/api/videotrack.h" | |
25 #include "webrtc/base/refcount.h" | 22 #include "webrtc/base/refcount.h" |
26 #include "webrtc/media/base/mediachannel.h" | 23 #include "webrtc/media/base/mediachannel.h" |
| 24 #include "webrtc/pc/audiotrack.h" |
| 25 #include "webrtc/pc/test/fakevideotracksource.h" |
| 26 #include "webrtc/pc/videotrack.h" |
27 #include "webrtc/test/gtest.h" | 27 #include "webrtc/test/gtest.h" |
28 | 28 |
29 namespace webrtc { | 29 namespace webrtc { |
30 | 30 |
31 namespace { | 31 namespace { |
32 | 32 |
33 RtpParameters CreateRtpParametersWithSsrcs( | 33 RtpParameters CreateRtpParametersWithSsrcs( |
34 std::initializer_list<uint32_t> ssrcs) { | 34 std::initializer_list<uint32_t> ssrcs) { |
35 RtpParameters params; | 35 RtpParameters params; |
36 for (uint32_t ssrc : ssrcs) { | 36 for (uint32_t ssrc : ssrcs) { |
37 RtpEncodingParameters encoding_params; | 37 RtpEncodingParameters encoding_params; |
38 encoding_params.ssrc = rtc::Optional<uint32_t>(ssrc); | 38 encoding_params.ssrc = rtc::Optional<uint32_t>(ssrc); |
39 params.encodings.push_back(encoding_params); | 39 params.encodings.push_back(encoding_params); |
40 } | 40 } |
41 return params; | 41 return params; |
42 } | 42 } |
43 | 43 |
44 rtc::scoped_refptr<MockRtpSender> CreateMockRtpSender( | 44 rtc::scoped_refptr<MockRtpSender> CreateMockRtpSender( |
45 cricket::MediaType media_type, std::initializer_list<uint32_t> ssrcs, | 45 cricket::MediaType media_type, |
| 46 std::initializer_list<uint32_t> ssrcs, |
46 rtc::scoped_refptr<MediaStreamTrackInterface> track) { | 47 rtc::scoped_refptr<MediaStreamTrackInterface> track) { |
47 uint32_t first_ssrc; | 48 uint32_t first_ssrc; |
48 if (ssrcs.size()) { | 49 if (ssrcs.size()) { |
49 first_ssrc = *ssrcs.begin(); | 50 first_ssrc = *ssrcs.begin(); |
50 } else { | 51 } else { |
51 first_ssrc = 0; | 52 first_ssrc = 0; |
52 } | 53 } |
53 rtc::scoped_refptr<MockRtpSender> sender( | 54 rtc::scoped_refptr<MockRtpSender> sender( |
54 new rtc::RefCountedObject<MockRtpSender>()); | 55 new rtc::RefCountedObject<MockRtpSender>()); |
55 EXPECT_CALL(*sender, track()).WillRepeatedly(testing::Return(track)); | 56 EXPECT_CALL(*sender, track()).WillRepeatedly(testing::Return(track)); |
56 EXPECT_CALL(*sender, ssrc()).WillRepeatedly(testing::Return(first_ssrc)); | 57 EXPECT_CALL(*sender, ssrc()).WillRepeatedly(testing::Return(first_ssrc)); |
57 EXPECT_CALL(*sender, media_type()).WillRepeatedly(testing::Return( | 58 EXPECT_CALL(*sender, media_type()) |
58 media_type)); | 59 .WillRepeatedly(testing::Return(media_type)); |
59 EXPECT_CALL(*sender, GetParameters()).WillRepeatedly(testing::Return( | 60 EXPECT_CALL(*sender, GetParameters()) |
60 CreateRtpParametersWithSsrcs(ssrcs))); | 61 .WillRepeatedly(testing::Return(CreateRtpParametersWithSsrcs(ssrcs))); |
61 return sender; | 62 return sender; |
62 } | 63 } |
63 | 64 |
64 rtc::scoped_refptr<MockRtpReceiver> CreateMockRtpReceiver( | 65 rtc::scoped_refptr<MockRtpReceiver> CreateMockRtpReceiver( |
65 cricket::MediaType media_type, std::initializer_list<uint32_t> ssrcs, | 66 cricket::MediaType media_type, |
| 67 std::initializer_list<uint32_t> ssrcs, |
66 rtc::scoped_refptr<MediaStreamTrackInterface> track) { | 68 rtc::scoped_refptr<MediaStreamTrackInterface> track) { |
67 rtc::scoped_refptr<MockRtpReceiver> receiver( | 69 rtc::scoped_refptr<MockRtpReceiver> receiver( |
68 new rtc::RefCountedObject<MockRtpReceiver>()); | 70 new rtc::RefCountedObject<MockRtpReceiver>()); |
69 EXPECT_CALL(*receiver, track()).WillRepeatedly(testing::Return(track)); | 71 EXPECT_CALL(*receiver, track()).WillRepeatedly(testing::Return(track)); |
70 EXPECT_CALL(*receiver, media_type()).WillRepeatedly(testing::Return( | 72 EXPECT_CALL(*receiver, media_type()) |
71 media_type)); | 73 .WillRepeatedly(testing::Return(media_type)); |
72 EXPECT_CALL(*receiver, GetParameters()).WillRepeatedly(testing::Return( | 74 EXPECT_CALL(*receiver, GetParameters()) |
73 CreateRtpParametersWithSsrcs(ssrcs))); | 75 .WillRepeatedly(testing::Return(CreateRtpParametersWithSsrcs(ssrcs))); |
74 return receiver; | 76 return receiver; |
75 } | 77 } |
76 | 78 |
77 class TrackMediaInfoMapTest : public testing::Test { | 79 class TrackMediaInfoMapTest : public testing::Test { |
78 public: | 80 public: |
79 TrackMediaInfoMapTest() | 81 TrackMediaInfoMapTest() |
80 : voice_media_info_(new cricket::VoiceMediaInfo()), | 82 : voice_media_info_(new cricket::VoiceMediaInfo()), |
81 video_media_info_(new cricket::VideoMediaInfo()), | 83 video_media_info_(new cricket::VideoMediaInfo()), |
82 local_audio_track_(AudioTrack::Create("LocalAudioTrack", nullptr)), | 84 local_audio_track_(AudioTrack::Create("LocalAudioTrack", nullptr)), |
83 remote_audio_track_(AudioTrack::Create("RemoteAudioTrack", nullptr)), | 85 remote_audio_track_(AudioTrack::Create("RemoteAudioTrack", nullptr)), |
84 local_video_track_( | 86 local_video_track_( |
85 VideoTrack::Create("LocalVideoTrack", | 87 VideoTrack::Create("LocalVideoTrack", |
86 FakeVideoTrackSource::Create(false))), | 88 FakeVideoTrackSource::Create(false))), |
87 remote_video_track_( | 89 remote_video_track_( |
88 VideoTrack::Create("RemoteVideoTrack", | 90 VideoTrack::Create("RemoteVideoTrack", |
89 FakeVideoTrackSource::Create(false))) { | 91 FakeVideoTrackSource::Create(false))) {} |
90 } | |
91 | 92 |
92 ~TrackMediaInfoMapTest() { | 93 ~TrackMediaInfoMapTest() { |
93 // If we have a map the ownership has been passed to the map, only delete if | 94 // If we have a map the ownership has been passed to the map, only delete if |
94 // |CreateMap| has not been called. | 95 // |CreateMap| has not been called. |
95 if (!map_) { | 96 if (!map_) { |
96 delete voice_media_info_; | 97 delete voice_media_info_; |
97 delete video_media_info_; | 98 delete video_media_info_; |
98 } | 99 } |
99 } | 100 } |
100 | 101 |
101 void AddRtpSenderWithSsrcs(std::initializer_list<uint32_t> ssrcs, | 102 void AddRtpSenderWithSsrcs(std::initializer_list<uint32_t> ssrcs, |
102 MediaStreamTrackInterface* local_track) { | 103 MediaStreamTrackInterface* local_track) { |
103 rtc::scoped_refptr<MockRtpSender> rtp_sender = CreateMockRtpSender( | 104 rtc::scoped_refptr<MockRtpSender> rtp_sender = CreateMockRtpSender( |
104 local_track->kind() == MediaStreamTrackInterface::kAudioKind ? | 105 local_track->kind() == MediaStreamTrackInterface::kAudioKind |
105 cricket::MEDIA_TYPE_AUDIO : cricket::MEDIA_TYPE_VIDEO, | 106 ? cricket::MEDIA_TYPE_AUDIO |
| 107 : cricket::MEDIA_TYPE_VIDEO, |
106 ssrcs, local_track); | 108 ssrcs, local_track); |
107 rtp_senders_.push_back(rtp_sender); | 109 rtp_senders_.push_back(rtp_sender); |
108 | 110 |
109 if (local_track->kind() == MediaStreamTrackInterface::kAudioKind) { | 111 if (local_track->kind() == MediaStreamTrackInterface::kAudioKind) { |
110 cricket::VoiceSenderInfo voice_sender_info; | 112 cricket::VoiceSenderInfo voice_sender_info; |
111 size_t i = 0; | 113 size_t i = 0; |
112 for (uint32_t ssrc : ssrcs) { | 114 for (uint32_t ssrc : ssrcs) { |
113 voice_sender_info.local_stats.push_back(cricket::SsrcSenderInfo()); | 115 voice_sender_info.local_stats.push_back(cricket::SsrcSenderInfo()); |
114 voice_sender_info.local_stats[i++].ssrc = ssrc; | 116 voice_sender_info.local_stats[i++].ssrc = ssrc; |
115 } | 117 } |
116 voice_media_info_->senders.push_back(voice_sender_info); | 118 voice_media_info_->senders.push_back(voice_sender_info); |
117 } else { | 119 } else { |
118 cricket::VideoSenderInfo video_sender_info; | 120 cricket::VideoSenderInfo video_sender_info; |
119 size_t i = 0; | 121 size_t i = 0; |
120 for (uint32_t ssrc : ssrcs) { | 122 for (uint32_t ssrc : ssrcs) { |
121 video_sender_info.local_stats.push_back(cricket::SsrcSenderInfo()); | 123 video_sender_info.local_stats.push_back(cricket::SsrcSenderInfo()); |
122 video_sender_info.local_stats[i++].ssrc = ssrc; | 124 video_sender_info.local_stats[i++].ssrc = ssrc; |
123 } | 125 } |
124 video_media_info_->senders.push_back(video_sender_info); | 126 video_media_info_->senders.push_back(video_sender_info); |
125 } | 127 } |
126 } | 128 } |
127 | 129 |
128 void AddRtpReceiverWithSsrcs(std::initializer_list<uint32_t> ssrcs, | 130 void AddRtpReceiverWithSsrcs(std::initializer_list<uint32_t> ssrcs, |
129 MediaStreamTrackInterface* remote_track) { | 131 MediaStreamTrackInterface* remote_track) { |
130 rtc::scoped_refptr<MockRtpReceiver> rtp_receiver = CreateMockRtpReceiver( | 132 rtc::scoped_refptr<MockRtpReceiver> rtp_receiver = CreateMockRtpReceiver( |
131 remote_track->kind() == MediaStreamTrackInterface::kAudioKind ? | 133 remote_track->kind() == MediaStreamTrackInterface::kAudioKind |
132 cricket::MEDIA_TYPE_AUDIO : cricket::MEDIA_TYPE_VIDEO, | 134 ? cricket::MEDIA_TYPE_AUDIO |
| 135 : cricket::MEDIA_TYPE_VIDEO, |
133 ssrcs, remote_track); | 136 ssrcs, remote_track); |
134 rtp_receivers_.push_back(rtp_receiver); | 137 rtp_receivers_.push_back(rtp_receiver); |
135 | 138 |
136 if (remote_track->kind() == MediaStreamTrackInterface::kAudioKind) { | 139 if (remote_track->kind() == MediaStreamTrackInterface::kAudioKind) { |
137 cricket::VoiceReceiverInfo voice_receiver_info; | 140 cricket::VoiceReceiverInfo voice_receiver_info; |
138 size_t i = 0; | 141 size_t i = 0; |
139 for (uint32_t ssrc : ssrcs) { | 142 for (uint32_t ssrc : ssrcs) { |
140 voice_receiver_info.local_stats.push_back(cricket::SsrcReceiverInfo()); | 143 voice_receiver_info.local_stats.push_back(cricket::SsrcReceiverInfo()); |
141 voice_receiver_info.local_stats[i++].ssrc = ssrc; | 144 voice_receiver_info.local_stats[i++].ssrc = ssrc; |
142 } | 145 } |
143 voice_media_info_->receivers.push_back(voice_receiver_info); | 146 voice_media_info_->receivers.push_back(voice_receiver_info); |
144 } else { | 147 } else { |
145 cricket::VideoReceiverInfo video_receiver_info; | 148 cricket::VideoReceiverInfo video_receiver_info; |
146 size_t i = 0; | 149 size_t i = 0; |
147 for (uint32_t ssrc : ssrcs) { | 150 for (uint32_t ssrc : ssrcs) { |
148 video_receiver_info.local_stats.push_back(cricket::SsrcReceiverInfo()); | 151 video_receiver_info.local_stats.push_back(cricket::SsrcReceiverInfo()); |
149 video_receiver_info.local_stats[i++].ssrc = ssrc; | 152 video_receiver_info.local_stats[i++].ssrc = ssrc; |
150 } | 153 } |
151 video_media_info_->receivers.push_back(video_receiver_info); | 154 video_media_info_->receivers.push_back(video_receiver_info); |
152 } | 155 } |
153 } | 156 } |
154 | 157 |
155 void CreateMap() { | 158 void CreateMap() { |
156 RTC_DCHECK(!map_); | 159 RTC_DCHECK(!map_); |
157 map_.reset(new TrackMediaInfoMap( | 160 map_.reset(new TrackMediaInfoMap( |
158 std::unique_ptr<cricket::VoiceMediaInfo>(voice_media_info_), | 161 std::unique_ptr<cricket::VoiceMediaInfo>(voice_media_info_), |
159 std::unique_ptr<cricket::VideoMediaInfo>(video_media_info_), | 162 std::unique_ptr<cricket::VideoMediaInfo>(video_media_info_), |
160 rtp_senders_, | 163 rtp_senders_, rtp_receivers_)); |
161 rtp_receivers_)); | |
162 } | 164 } |
163 | 165 |
164 protected: | 166 protected: |
165 cricket::VoiceMediaInfo* voice_media_info_; | 167 cricket::VoiceMediaInfo* voice_media_info_; |
166 cricket::VideoMediaInfo* video_media_info_; | 168 cricket::VideoMediaInfo* video_media_info_; |
167 std::vector<rtc::scoped_refptr<RtpSenderInterface>> rtp_senders_; | 169 std::vector<rtc::scoped_refptr<RtpSenderInterface>> rtp_senders_; |
168 std::vector<rtc::scoped_refptr<RtpReceiverInterface>> rtp_receivers_; | 170 std::vector<rtc::scoped_refptr<RtpReceiverInterface>> rtp_receivers_; |
169 std::unique_ptr<TrackMediaInfoMap> map_; | 171 std::unique_ptr<TrackMediaInfoMap> map_; |
170 rtc::scoped_refptr<AudioTrack> local_audio_track_; | 172 rtc::scoped_refptr<AudioTrack> local_audio_track_; |
171 rtc::scoped_refptr<AudioTrack> remote_audio_track_; | 173 rtc::scoped_refptr<AudioTrack> remote_audio_track_; |
172 rtc::scoped_refptr<VideoTrack> local_video_track_; | 174 rtc::scoped_refptr<VideoTrack> local_video_track_; |
173 rtc::scoped_refptr<VideoTrack> remote_video_track_; | 175 rtc::scoped_refptr<VideoTrack> remote_video_track_; |
174 }; | 176 }; |
175 | 177 |
176 } // namespace | 178 } // namespace |
177 | 179 |
178 TEST_F(TrackMediaInfoMapTest, SingleSenderReceiverPerTrackWithOneSsrc) { | 180 TEST_F(TrackMediaInfoMapTest, SingleSenderReceiverPerTrackWithOneSsrc) { |
179 AddRtpSenderWithSsrcs({ 1 }, local_audio_track_); | 181 AddRtpSenderWithSsrcs({1}, local_audio_track_); |
180 AddRtpReceiverWithSsrcs({ 2 }, remote_audio_track_); | 182 AddRtpReceiverWithSsrcs({2}, remote_audio_track_); |
181 AddRtpSenderWithSsrcs({ 3 }, local_video_track_); | 183 AddRtpSenderWithSsrcs({3}, local_video_track_); |
182 AddRtpReceiverWithSsrcs({ 4 }, remote_video_track_); | 184 AddRtpReceiverWithSsrcs({4}, remote_video_track_); |
183 CreateMap(); | 185 CreateMap(); |
184 | 186 |
185 // Local audio track <-> RTP audio sender | 187 // Local audio track <-> RTP audio sender |
186 ASSERT_TRUE(map_->GetVoiceSenderInfos(*local_audio_track_)); | 188 ASSERT_TRUE(map_->GetVoiceSenderInfos(*local_audio_track_)); |
187 EXPECT_EQ(*map_->GetVoiceSenderInfos(*local_audio_track_), | 189 EXPECT_EQ( |
188 std::vector<cricket::VoiceSenderInfo*>({ | 190 *map_->GetVoiceSenderInfos(*local_audio_track_), |
189 &voice_media_info_->senders[0] })); | 191 std::vector<cricket::VoiceSenderInfo*>({&voice_media_info_->senders[0]})); |
190 EXPECT_EQ(map_->GetAudioTrack(voice_media_info_->senders[0]), | 192 EXPECT_EQ(map_->GetAudioTrack(voice_media_info_->senders[0]), |
191 local_audio_track_.get()); | 193 local_audio_track_.get()); |
192 | 194 |
193 // Remote audio track <-> RTP audio receiver | 195 // Remote audio track <-> RTP audio receiver |
194 EXPECT_EQ(map_->GetVoiceReceiverInfo(*remote_audio_track_), | 196 EXPECT_EQ(map_->GetVoiceReceiverInfo(*remote_audio_track_), |
195 &voice_media_info_->receivers[0]); | 197 &voice_media_info_->receivers[0]); |
196 EXPECT_EQ(map_->GetAudioTrack(voice_media_info_->receivers[0]), | 198 EXPECT_EQ(map_->GetAudioTrack(voice_media_info_->receivers[0]), |
197 remote_audio_track_.get()); | 199 remote_audio_track_.get()); |
198 | 200 |
199 // Local video track <-> RTP video sender | 201 // Local video track <-> RTP video sender |
200 ASSERT_TRUE(map_->GetVideoSenderInfos(*local_video_track_)); | 202 ASSERT_TRUE(map_->GetVideoSenderInfos(*local_video_track_)); |
201 EXPECT_EQ(*map_->GetVideoSenderInfos(*local_video_track_), | 203 EXPECT_EQ( |
202 std::vector<cricket::VideoSenderInfo*>({ | 204 *map_->GetVideoSenderInfos(*local_video_track_), |
203 &video_media_info_->senders[0] })); | 205 std::vector<cricket::VideoSenderInfo*>({&video_media_info_->senders[0]})); |
204 EXPECT_EQ(map_->GetVideoTrack(video_media_info_->senders[0]), | 206 EXPECT_EQ(map_->GetVideoTrack(video_media_info_->senders[0]), |
205 local_video_track_.get()); | 207 local_video_track_.get()); |
206 | 208 |
207 // Remote video track <-> RTP video receiver | 209 // Remote video track <-> RTP video receiver |
208 EXPECT_EQ(map_->GetVideoReceiverInfo(*remote_video_track_), | 210 EXPECT_EQ(map_->GetVideoReceiverInfo(*remote_video_track_), |
209 &video_media_info_->receivers[0]); | 211 &video_media_info_->receivers[0]); |
210 EXPECT_EQ(map_->GetVideoTrack(video_media_info_->receivers[0]), | 212 EXPECT_EQ(map_->GetVideoTrack(video_media_info_->receivers[0]), |
211 remote_video_track_.get()); | 213 remote_video_track_.get()); |
212 } | 214 } |
213 | 215 |
214 TEST_F(TrackMediaInfoMapTest, SingleSenderReceiverPerTrackWithMissingSsrc) { | 216 TEST_F(TrackMediaInfoMapTest, SingleSenderReceiverPerTrackWithMissingSsrc) { |
215 AddRtpSenderWithSsrcs({}, local_audio_track_); | 217 AddRtpSenderWithSsrcs({}, local_audio_track_); |
216 AddRtpSenderWithSsrcs({}, local_video_track_); | 218 AddRtpSenderWithSsrcs({}, local_video_track_); |
217 AddRtpReceiverWithSsrcs({}, remote_audio_track_); | 219 AddRtpReceiverWithSsrcs({}, remote_audio_track_); |
218 AddRtpReceiverWithSsrcs({}, remote_video_track_); | 220 AddRtpReceiverWithSsrcs({}, remote_video_track_); |
219 CreateMap(); | 221 CreateMap(); |
220 | 222 |
221 EXPECT_FALSE(map_->GetVoiceSenderInfos(*local_audio_track_)); | 223 EXPECT_FALSE(map_->GetVoiceSenderInfos(*local_audio_track_)); |
222 EXPECT_FALSE(map_->GetVideoSenderInfos(*local_video_track_)); | 224 EXPECT_FALSE(map_->GetVideoSenderInfos(*local_video_track_)); |
223 EXPECT_FALSE(map_->GetVoiceReceiverInfo(*remote_audio_track_)); | 225 EXPECT_FALSE(map_->GetVoiceReceiverInfo(*remote_audio_track_)); |
224 EXPECT_FALSE(map_->GetVideoReceiverInfo(*remote_video_track_)); | 226 EXPECT_FALSE(map_->GetVideoReceiverInfo(*remote_video_track_)); |
225 } | 227 } |
226 | 228 |
227 TEST_F(TrackMediaInfoMapTest, | 229 TEST_F(TrackMediaInfoMapTest, |
228 SingleSenderReceiverPerTrackWithAudioAndVideoUseSameSsrc) { | 230 SingleSenderReceiverPerTrackWithAudioAndVideoUseSameSsrc) { |
229 AddRtpSenderWithSsrcs({ 1 }, local_audio_track_); | 231 AddRtpSenderWithSsrcs({1}, local_audio_track_); |
230 AddRtpReceiverWithSsrcs({ 2 }, remote_audio_track_); | 232 AddRtpReceiverWithSsrcs({2}, remote_audio_track_); |
231 AddRtpSenderWithSsrcs({ 1 }, local_video_track_); | 233 AddRtpSenderWithSsrcs({1}, local_video_track_); |
232 AddRtpReceiverWithSsrcs({ 2 }, remote_video_track_); | 234 AddRtpReceiverWithSsrcs({2}, remote_video_track_); |
233 CreateMap(); | 235 CreateMap(); |
234 | 236 |
235 // Local audio track <-> RTP audio sender | 237 // Local audio track <-> RTP audio sender |
236 ASSERT_TRUE(map_->GetVoiceSenderInfos(*local_audio_track_)); | 238 ASSERT_TRUE(map_->GetVoiceSenderInfos(*local_audio_track_)); |
237 EXPECT_EQ(*map_->GetVoiceSenderInfos(*local_audio_track_), | 239 EXPECT_EQ( |
238 std::vector<cricket::VoiceSenderInfo*>({ | 240 *map_->GetVoiceSenderInfos(*local_audio_track_), |
239 &voice_media_info_->senders[0] })); | 241 std::vector<cricket::VoiceSenderInfo*>({&voice_media_info_->senders[0]})); |
240 EXPECT_EQ(map_->GetAudioTrack(voice_media_info_->senders[0]), | 242 EXPECT_EQ(map_->GetAudioTrack(voice_media_info_->senders[0]), |
241 local_audio_track_.get()); | 243 local_audio_track_.get()); |
242 | 244 |
243 // Remote audio track <-> RTP audio receiver | 245 // Remote audio track <-> RTP audio receiver |
244 EXPECT_EQ(map_->GetVoiceReceiverInfo(*remote_audio_track_), | 246 EXPECT_EQ(map_->GetVoiceReceiverInfo(*remote_audio_track_), |
245 &voice_media_info_->receivers[0]); | 247 &voice_media_info_->receivers[0]); |
246 EXPECT_EQ(map_->GetAudioTrack(voice_media_info_->receivers[0]), | 248 EXPECT_EQ(map_->GetAudioTrack(voice_media_info_->receivers[0]), |
247 remote_audio_track_.get()); | 249 remote_audio_track_.get()); |
248 | 250 |
249 // Local video track <-> RTP video sender | 251 // Local video track <-> RTP video sender |
250 ASSERT_TRUE(map_->GetVideoSenderInfos(*local_video_track_)); | 252 ASSERT_TRUE(map_->GetVideoSenderInfos(*local_video_track_)); |
251 EXPECT_EQ(*map_->GetVideoSenderInfos(*local_video_track_), | 253 EXPECT_EQ( |
252 std::vector<cricket::VideoSenderInfo*>({ | 254 *map_->GetVideoSenderInfos(*local_video_track_), |
253 &video_media_info_->senders[0] })); | 255 std::vector<cricket::VideoSenderInfo*>({&video_media_info_->senders[0]})); |
254 EXPECT_EQ(map_->GetVideoTrack(video_media_info_->senders[0]), | 256 EXPECT_EQ(map_->GetVideoTrack(video_media_info_->senders[0]), |
255 local_video_track_.get()); | 257 local_video_track_.get()); |
256 | 258 |
257 // Remote video track <-> RTP video receiver | 259 // Remote video track <-> RTP video receiver |
258 EXPECT_EQ(map_->GetVideoReceiverInfo(*remote_video_track_), | 260 EXPECT_EQ(map_->GetVideoReceiverInfo(*remote_video_track_), |
259 &video_media_info_->receivers[0]); | 261 &video_media_info_->receivers[0]); |
260 EXPECT_EQ(map_->GetVideoTrack(video_media_info_->receivers[0]), | 262 EXPECT_EQ(map_->GetVideoTrack(video_media_info_->receivers[0]), |
261 remote_video_track_.get()); | 263 remote_video_track_.get()); |
262 } | 264 } |
263 | 265 |
264 TEST_F(TrackMediaInfoMapTest, SingleMultiSsrcSenderPerTrack) { | 266 TEST_F(TrackMediaInfoMapTest, SingleMultiSsrcSenderPerTrack) { |
265 AddRtpSenderWithSsrcs({ 1, 2 }, local_audio_track_); | 267 AddRtpSenderWithSsrcs({1, 2}, local_audio_track_); |
266 AddRtpSenderWithSsrcs({ 3, 4 }, local_video_track_); | 268 AddRtpSenderWithSsrcs({3, 4}, local_video_track_); |
267 CreateMap(); | 269 CreateMap(); |
268 | 270 |
269 // Local audio track <-> RTP audio senders | 271 // Local audio track <-> RTP audio senders |
270 ASSERT_TRUE(map_->GetVoiceSenderInfos(*local_audio_track_)); | 272 ASSERT_TRUE(map_->GetVoiceSenderInfos(*local_audio_track_)); |
271 EXPECT_EQ(*map_->GetVoiceSenderInfos(*local_audio_track_), | 273 EXPECT_EQ( |
272 std::vector<cricket::VoiceSenderInfo*>({ | 274 *map_->GetVoiceSenderInfos(*local_audio_track_), |
273 &voice_media_info_->senders[0] })); | 275 std::vector<cricket::VoiceSenderInfo*>({&voice_media_info_->senders[0]})); |
274 EXPECT_EQ(map_->GetAudioTrack(voice_media_info_->senders[0]), | 276 EXPECT_EQ(map_->GetAudioTrack(voice_media_info_->senders[0]), |
275 local_audio_track_.get()); | 277 local_audio_track_.get()); |
276 | 278 |
277 // Local video track <-> RTP video senders | 279 // Local video track <-> RTP video senders |
278 ASSERT_TRUE(map_->GetVideoSenderInfos(*local_video_track_)); | 280 ASSERT_TRUE(map_->GetVideoSenderInfos(*local_video_track_)); |
279 EXPECT_EQ(*map_->GetVideoSenderInfos(*local_video_track_), | 281 EXPECT_EQ( |
280 std::vector<cricket::VideoSenderInfo*>({ | 282 *map_->GetVideoSenderInfos(*local_video_track_), |
281 &video_media_info_->senders[0] })); | 283 std::vector<cricket::VideoSenderInfo*>({&video_media_info_->senders[0]})); |
282 EXPECT_EQ(map_->GetVideoTrack(video_media_info_->senders[0]), | 284 EXPECT_EQ(map_->GetVideoTrack(video_media_info_->senders[0]), |
283 local_video_track_.get()); | 285 local_video_track_.get()); |
284 } | 286 } |
285 | 287 |
286 TEST_F(TrackMediaInfoMapTest, MultipleOneSsrcSendersPerTrack) { | 288 TEST_F(TrackMediaInfoMapTest, MultipleOneSsrcSendersPerTrack) { |
287 AddRtpSenderWithSsrcs({ 1 }, local_audio_track_); | 289 AddRtpSenderWithSsrcs({1}, local_audio_track_); |
288 AddRtpSenderWithSsrcs({ 2 }, local_audio_track_); | 290 AddRtpSenderWithSsrcs({2}, local_audio_track_); |
289 AddRtpSenderWithSsrcs({ 3 }, local_video_track_); | 291 AddRtpSenderWithSsrcs({3}, local_video_track_); |
290 AddRtpSenderWithSsrcs({ 4 }, local_video_track_); | 292 AddRtpSenderWithSsrcs({4}, local_video_track_); |
291 CreateMap(); | 293 CreateMap(); |
292 | 294 |
293 // Local audio track <-> RTP audio senders | 295 // Local audio track <-> RTP audio senders |
294 ASSERT_TRUE(map_->GetVoiceSenderInfos(*local_audio_track_)); | 296 ASSERT_TRUE(map_->GetVoiceSenderInfos(*local_audio_track_)); |
295 EXPECT_EQ(*map_->GetVoiceSenderInfos(*local_audio_track_), | 297 EXPECT_EQ( |
296 std::vector<cricket::VoiceSenderInfo*>({ | 298 *map_->GetVoiceSenderInfos(*local_audio_track_), |
297 &voice_media_info_->senders[0], | 299 std::vector<cricket::VoiceSenderInfo*>( |
298 &voice_media_info_->senders[1] })); | 300 {&voice_media_info_->senders[0], &voice_media_info_->senders[1]})); |
299 EXPECT_EQ(map_->GetAudioTrack(voice_media_info_->senders[0]), | 301 EXPECT_EQ(map_->GetAudioTrack(voice_media_info_->senders[0]), |
300 local_audio_track_.get()); | 302 local_audio_track_.get()); |
301 EXPECT_EQ(map_->GetAudioTrack(voice_media_info_->senders[1]), | 303 EXPECT_EQ(map_->GetAudioTrack(voice_media_info_->senders[1]), |
302 local_audio_track_.get()); | 304 local_audio_track_.get()); |
303 | 305 |
304 // Local video track <-> RTP video senders | 306 // Local video track <-> RTP video senders |
305 ASSERT_TRUE(map_->GetVideoSenderInfos(*local_video_track_)); | 307 ASSERT_TRUE(map_->GetVideoSenderInfos(*local_video_track_)); |
306 EXPECT_EQ(*map_->GetVideoSenderInfos(*local_video_track_), | 308 EXPECT_EQ( |
307 std::vector<cricket::VideoSenderInfo*>({ | 309 *map_->GetVideoSenderInfos(*local_video_track_), |
308 &video_media_info_->senders[0], | 310 std::vector<cricket::VideoSenderInfo*>( |
309 &video_media_info_->senders[1] })); | 311 {&video_media_info_->senders[0], &video_media_info_->senders[1]})); |
310 EXPECT_EQ(map_->GetVideoTrack(video_media_info_->senders[0]), | 312 EXPECT_EQ(map_->GetVideoTrack(video_media_info_->senders[0]), |
311 local_video_track_.get()); | 313 local_video_track_.get()); |
312 EXPECT_EQ(map_->GetVideoTrack(video_media_info_->senders[1]), | 314 EXPECT_EQ(map_->GetVideoTrack(video_media_info_->senders[1]), |
313 local_video_track_.get()); | 315 local_video_track_.get()); |
314 } | 316 } |
315 | 317 |
316 TEST_F(TrackMediaInfoMapTest, MultipleMultiSsrcSendersPerTrack) { | 318 TEST_F(TrackMediaInfoMapTest, MultipleMultiSsrcSendersPerTrack) { |
317 AddRtpSenderWithSsrcs({ 1, 2 }, local_audio_track_); | 319 AddRtpSenderWithSsrcs({1, 2}, local_audio_track_); |
318 AddRtpSenderWithSsrcs({ 3, 4 }, local_audio_track_); | 320 AddRtpSenderWithSsrcs({3, 4}, local_audio_track_); |
319 AddRtpSenderWithSsrcs({ 5, 6 }, local_video_track_); | 321 AddRtpSenderWithSsrcs({5, 6}, local_video_track_); |
320 AddRtpSenderWithSsrcs({ 7, 8 }, local_video_track_); | 322 AddRtpSenderWithSsrcs({7, 8}, local_video_track_); |
321 CreateMap(); | 323 CreateMap(); |
322 | 324 |
323 // Local audio track <-> RTP audio senders | 325 // Local audio track <-> RTP audio senders |
324 ASSERT_TRUE(map_->GetVoiceSenderInfos(*local_audio_track_)); | 326 ASSERT_TRUE(map_->GetVoiceSenderInfos(*local_audio_track_)); |
325 EXPECT_EQ(*map_->GetVoiceSenderInfos(*local_audio_track_), | 327 EXPECT_EQ( |
326 std::vector<cricket::VoiceSenderInfo*>({ | 328 *map_->GetVoiceSenderInfos(*local_audio_track_), |
327 &voice_media_info_->senders[0], | 329 std::vector<cricket::VoiceSenderInfo*>( |
328 &voice_media_info_->senders[1] })); | 330 {&voice_media_info_->senders[0], &voice_media_info_->senders[1]})); |
329 EXPECT_EQ(map_->GetAudioTrack(voice_media_info_->senders[0]), | 331 EXPECT_EQ(map_->GetAudioTrack(voice_media_info_->senders[0]), |
330 local_audio_track_.get()); | 332 local_audio_track_.get()); |
331 EXPECT_EQ(map_->GetAudioTrack(voice_media_info_->senders[1]), | 333 EXPECT_EQ(map_->GetAudioTrack(voice_media_info_->senders[1]), |
332 local_audio_track_.get()); | 334 local_audio_track_.get()); |
333 | 335 |
334 // Local video track <-> RTP video senders | 336 // Local video track <-> RTP video senders |
335 ASSERT_TRUE(map_->GetVideoSenderInfos(*local_video_track_)); | 337 ASSERT_TRUE(map_->GetVideoSenderInfos(*local_video_track_)); |
336 EXPECT_EQ(*map_->GetVideoSenderInfos(*local_video_track_), | 338 EXPECT_EQ( |
337 std::vector<cricket::VideoSenderInfo*>({ | 339 *map_->GetVideoSenderInfos(*local_video_track_), |
338 &video_media_info_->senders[0], | 340 std::vector<cricket::VideoSenderInfo*>( |
339 &video_media_info_->senders[1] })); | 341 {&video_media_info_->senders[0], &video_media_info_->senders[1]})); |
340 EXPECT_EQ(map_->GetVideoTrack(video_media_info_->senders[0]), | 342 EXPECT_EQ(map_->GetVideoTrack(video_media_info_->senders[0]), |
341 local_video_track_.get()); | 343 local_video_track_.get()); |
342 EXPECT_EQ(map_->GetVideoTrack(video_media_info_->senders[1]), | 344 EXPECT_EQ(map_->GetVideoTrack(video_media_info_->senders[1]), |
343 local_video_track_.get()); | 345 local_video_track_.get()); |
344 } | 346 } |
345 | 347 |
346 // Death tests. | 348 // Death tests. |
347 // Disabled on Android because death tests misbehave on Android, see | 349 // Disabled on Android because death tests misbehave on Android, see |
348 // base/test/gtest_util.h. | 350 // base/test/gtest_util.h. |
349 #if RTC_DCHECK_IS_ON && GTEST_HAS_DEATH_TEST && !defined(WEBRTC_ANDROID) | 351 #if RTC_DCHECK_IS_ON && GTEST_HAS_DEATH_TEST && !defined(WEBRTC_ANDROID) |
350 | 352 |
351 class TrackMediaInfoMapDeathTest : public TrackMediaInfoMapTest { | 353 class TrackMediaInfoMapDeathTest : public TrackMediaInfoMapTest {}; |
352 }; | |
353 | 354 |
354 TEST_F(TrackMediaInfoMapDeathTest, MultipleOneSsrcReceiversPerTrack) { | 355 TEST_F(TrackMediaInfoMapDeathTest, MultipleOneSsrcReceiversPerTrack) { |
355 AddRtpReceiverWithSsrcs({ 1 }, remote_audio_track_); | 356 AddRtpReceiverWithSsrcs({1}, remote_audio_track_); |
356 AddRtpReceiverWithSsrcs({ 2 }, remote_audio_track_); | 357 AddRtpReceiverWithSsrcs({2}, remote_audio_track_); |
357 AddRtpReceiverWithSsrcs({ 3 }, remote_video_track_); | 358 AddRtpReceiverWithSsrcs({3}, remote_video_track_); |
358 AddRtpReceiverWithSsrcs({ 4 }, remote_video_track_); | 359 AddRtpReceiverWithSsrcs({4}, remote_video_track_); |
359 EXPECT_DEATH(CreateMap(), ""); | 360 EXPECT_DEATH(CreateMap(), ""); |
360 } | 361 } |
361 | 362 |
362 TEST_F(TrackMediaInfoMapDeathTest, MultipleMultiSsrcReceiversPerTrack) { | 363 TEST_F(TrackMediaInfoMapDeathTest, MultipleMultiSsrcReceiversPerTrack) { |
363 AddRtpReceiverWithSsrcs({ 1, 2 }, remote_audio_track_); | 364 AddRtpReceiverWithSsrcs({1, 2}, remote_audio_track_); |
364 AddRtpReceiverWithSsrcs({ 3, 4 }, remote_audio_track_); | 365 AddRtpReceiverWithSsrcs({3, 4}, remote_audio_track_); |
365 AddRtpReceiverWithSsrcs({ 5, 6 }, remote_video_track_); | 366 AddRtpReceiverWithSsrcs({5, 6}, remote_video_track_); |
366 AddRtpReceiverWithSsrcs({ 7, 8 }, remote_video_track_); | 367 AddRtpReceiverWithSsrcs({7, 8}, remote_video_track_); |
367 EXPECT_DEATH(CreateMap(), ""); | 368 EXPECT_DEATH(CreateMap(), ""); |
368 } | 369 } |
369 | 370 |
370 TEST_F(TrackMediaInfoMapDeathTest, | 371 TEST_F(TrackMediaInfoMapDeathTest, |
371 SingleSenderReceiverPerTrackWithSsrcNotUnique) { | 372 SingleSenderReceiverPerTrackWithSsrcNotUnique) { |
372 AddRtpSenderWithSsrcs({ 1 }, local_audio_track_); | 373 AddRtpSenderWithSsrcs({1}, local_audio_track_); |
373 AddRtpReceiverWithSsrcs({ 1 }, remote_audio_track_); | 374 AddRtpReceiverWithSsrcs({1}, remote_audio_track_); |
374 AddRtpSenderWithSsrcs({ 2 }, local_video_track_); | 375 AddRtpSenderWithSsrcs({2}, local_video_track_); |
375 AddRtpReceiverWithSsrcs({ 2 }, remote_video_track_); | 376 AddRtpReceiverWithSsrcs({2}, remote_video_track_); |
376 EXPECT_DEATH(CreateMap(), ""); | 377 EXPECT_DEATH(CreateMap(), ""); |
377 } | 378 } |
378 | 379 |
379 #endif // RTC_DCHECK_IS_ON && GTEST_HAS_DEATH_TEST && !defined(WEBRTC_ANDROID) | 380 #endif // RTC_DCHECK_IS_ON && GTEST_HAS_DEATH_TEST && !defined(WEBRTC_ANDROID) |
380 | 381 |
381 } // namespace webrtc | 382 } // namespace webrtc |
OLD | NEW |