blob: e4883f32f61947fcf585043b86f67a7349f73314 [file] [log] [blame]
Amit Hilbuchdd9390c2018-11-13 16:26:05 -08001/*
2 * Copyright 2018 The WebRTC project authors. All Rights Reserved.
3 *
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.
9 */
10
11// This file contains tests for |RtpTransceiver|.
12
Steve Anton10542f22019-01-11 09:11:00 -080013#include "pc/rtp_transceiver.h"
Yves Gerey3e707812018-11-28 16:47:49 +010014
Markus Handell0357b3e2020-03-16 13:40:51 +010015#include <memory>
16
17#include "media/base/fake_media_engine.h"
Steve Anton10542f22019-01-11 09:11:00 -080018#include "pc/test/mock_channel_interface.h"
Markus Handell0357b3e2020-03-16 13:40:51 +010019#include "pc/test/mock_rtp_receiver_internal.h"
20#include "pc/test/mock_rtp_sender_internal.h"
Amit Hilbuchdd9390c2018-11-13 16:26:05 -080021#include "test/gmock.h"
Yves Gerey3e707812018-11-28 16:47:49 +010022#include "test/gtest.h"
Amit Hilbuchdd9390c2018-11-13 16:26:05 -080023
Markus Handell0357b3e2020-03-16 13:40:51 +010024using ::testing::ElementsAre;
25using ::testing::Eq;
26using ::testing::Field;
27using ::testing::Not;
Markus Handell755c65d2020-06-24 01:06:10 +020028using ::testing::Property;
Amit Hilbuchdd9390c2018-11-13 16:26:05 -080029using ::testing::Return;
30using ::testing::ReturnRef;
31
32namespace webrtc {
33
34// Checks that a channel cannot be set on a stopped |RtpTransceiver|.
35TEST(RtpTransceiverTest, CannotSetChannelOnStoppedTransceiver) {
36 RtpTransceiver transceiver(cricket::MediaType::MEDIA_TYPE_AUDIO);
37 cricket::MockChannelInterface channel1;
38 sigslot::signal1<cricket::ChannelInterface*> signal;
39 EXPECT_CALL(channel1, media_type())
40 .WillRepeatedly(Return(cricket::MediaType::MEDIA_TYPE_AUDIO));
41 EXPECT_CALL(channel1, SignalFirstPacketReceived())
42 .WillRepeatedly(ReturnRef(signal));
43
44 transceiver.SetChannel(&channel1);
45 EXPECT_EQ(&channel1, transceiver.channel());
46
47 // Stop the transceiver.
Harald Alvestrand6060df52020-08-11 09:54:02 +020048 transceiver.StopInternal();
Amit Hilbuchdd9390c2018-11-13 16:26:05 -080049 EXPECT_EQ(&channel1, transceiver.channel());
50
51 cricket::MockChannelInterface channel2;
52 EXPECT_CALL(channel2, media_type())
53 .WillRepeatedly(Return(cricket::MediaType::MEDIA_TYPE_AUDIO));
54
55 // Channel can no longer be set, so this call should be a no-op.
56 transceiver.SetChannel(&channel2);
57 EXPECT_EQ(&channel1, transceiver.channel());
58}
59
60// Checks that a channel can be unset on a stopped |RtpTransceiver|
61TEST(RtpTransceiverTest, CanUnsetChannelOnStoppedTransceiver) {
62 RtpTransceiver transceiver(cricket::MediaType::MEDIA_TYPE_VIDEO);
63 cricket::MockChannelInterface channel;
64 sigslot::signal1<cricket::ChannelInterface*> signal;
65 EXPECT_CALL(channel, media_type())
66 .WillRepeatedly(Return(cricket::MediaType::MEDIA_TYPE_VIDEO));
67 EXPECT_CALL(channel, SignalFirstPacketReceived())
68 .WillRepeatedly(ReturnRef(signal));
69
70 transceiver.SetChannel(&channel);
71 EXPECT_EQ(&channel, transceiver.channel());
72
73 // Stop the transceiver.
Harald Alvestrand6060df52020-08-11 09:54:02 +020074 transceiver.StopInternal();
Amit Hilbuchdd9390c2018-11-13 16:26:05 -080075 EXPECT_EQ(&channel, transceiver.channel());
76
77 // Set the channel to |nullptr|.
78 transceiver.SetChannel(nullptr);
79 EXPECT_EQ(nullptr, transceiver.channel());
80}
81
Harald Alvestrandc75c4282020-08-26 12:17:54 +000082class RtpTransceiverUnifiedPlanTest : public ::testing::Test {
83 public:
84 RtpTransceiverUnifiedPlanTest()
85 : channel_manager_(std::make_unique<cricket::FakeMediaEngine>(),
86 std::make_unique<cricket::FakeDataEngine>(),
87 rtc::Thread::Current(),
88 rtc::Thread::Current()),
89 transceiver_(RtpSenderProxyWithInternal<RtpSenderInternal>::Create(
90 rtc::Thread::Current(),
91 new rtc::RefCountedObject<MockRtpSenderInternal>()),
92 RtpReceiverProxyWithInternal<RtpReceiverInternal>::Create(
93 rtc::Thread::Current(),
94 new rtc::RefCountedObject<MockRtpReceiverInternal>()),
95 &channel_manager_,
96 channel_manager_.GetSupportedAudioRtpHeaderExtensions()) {}
97
98 cricket::ChannelManager channel_manager_;
99 RtpTransceiver transceiver_;
100};
101
102// Basic tests for Stop()
103TEST_F(RtpTransceiverUnifiedPlanTest, StopSetsDirection) {
104 EXPECT_EQ(RtpTransceiverDirection::kInactive, transceiver_.direction());
105 EXPECT_FALSE(transceiver_.current_direction());
106 transceiver_.StopStandard();
107 EXPECT_EQ(RtpTransceiverDirection::kStopped, transceiver_.direction());
108 EXPECT_FALSE(transceiver_.current_direction());
109 transceiver_.StopTransceiverProcedure();
110 EXPECT_TRUE(transceiver_.current_direction());
111 EXPECT_EQ(RtpTransceiverDirection::kStopped, transceiver_.direction());
112 EXPECT_EQ(RtpTransceiverDirection::kStopped,
113 *transceiver_.current_direction());
114}
115
Markus Handell755c65d2020-06-24 01:06:10 +0200116class RtpTransceiverTestForHeaderExtensions : public ::testing::Test {
117 public:
118 RtpTransceiverTestForHeaderExtensions()
119 : channel_manager_(std::make_unique<cricket::FakeMediaEngine>(),
120 std::make_unique<cricket::FakeDataEngine>(),
121 rtc::Thread::Current(),
122 rtc::Thread::Current()),
123 extensions_(
124 {RtpHeaderExtensionCapability("uri1",
125 1,
126 RtpTransceiverDirection::kSendOnly),
127 RtpHeaderExtensionCapability("uri2",
128 2,
129 RtpTransceiverDirection::kRecvOnly),
130 RtpHeaderExtensionCapability(RtpExtension::kMidUri,
131 3,
132 RtpTransceiverDirection::kSendRecv),
133 RtpHeaderExtensionCapability(RtpExtension::kVideoRotationUri,
134 4,
135 RtpTransceiverDirection::kSendRecv)}),
136 transceiver_(RtpSenderProxyWithInternal<RtpSenderInternal>::Create(
137 rtc::Thread::Current(),
138 new rtc::RefCountedObject<MockRtpSenderInternal>()),
139 RtpReceiverProxyWithInternal<RtpReceiverInternal>::Create(
140 rtc::Thread::Current(),
141 new rtc::RefCountedObject<MockRtpReceiverInternal>()),
142 &channel_manager_,
143 extensions_) {}
144
145 cricket::ChannelManager channel_manager_;
146 std::vector<RtpHeaderExtensionCapability> extensions_;
147 RtpTransceiver transceiver_;
148};
149
150TEST_F(RtpTransceiverTestForHeaderExtensions, OffersChannelManagerList) {
151 EXPECT_EQ(transceiver_.HeaderExtensionsToOffer(), extensions_);
152}
153
154TEST_F(RtpTransceiverTestForHeaderExtensions, ModifiesDirection) {
155 auto modified_extensions = extensions_;
156 modified_extensions[0].direction = RtpTransceiverDirection::kSendOnly;
157 EXPECT_TRUE(
158 transceiver_.SetOfferedRtpHeaderExtensions(modified_extensions).ok());
159 EXPECT_EQ(transceiver_.HeaderExtensionsToOffer(), modified_extensions);
160 modified_extensions[0].direction = RtpTransceiverDirection::kRecvOnly;
161 EXPECT_TRUE(
162 transceiver_.SetOfferedRtpHeaderExtensions(modified_extensions).ok());
163 EXPECT_EQ(transceiver_.HeaderExtensionsToOffer(), modified_extensions);
164 modified_extensions[0].direction = RtpTransceiverDirection::kSendRecv;
165 EXPECT_TRUE(
166 transceiver_.SetOfferedRtpHeaderExtensions(modified_extensions).ok());
167 EXPECT_EQ(transceiver_.HeaderExtensionsToOffer(), modified_extensions);
168 modified_extensions[0].direction = RtpTransceiverDirection::kInactive;
169 EXPECT_TRUE(
170 transceiver_.SetOfferedRtpHeaderExtensions(modified_extensions).ok());
171 EXPECT_EQ(transceiver_.HeaderExtensionsToOffer(), modified_extensions);
172}
173
174TEST_F(RtpTransceiverTestForHeaderExtensions, AcceptsStoppedExtension) {
175 auto modified_extensions = extensions_;
176 modified_extensions[0].direction = RtpTransceiverDirection::kStopped;
177 EXPECT_TRUE(
178 transceiver_.SetOfferedRtpHeaderExtensions(modified_extensions).ok());
179 EXPECT_EQ(transceiver_.HeaderExtensionsToOffer(), modified_extensions);
180}
181
182TEST_F(RtpTransceiverTestForHeaderExtensions, RejectsUnsupportedExtension) {
183 std::vector<RtpHeaderExtensionCapability> modified_extensions(
184 {RtpHeaderExtensionCapability("uri3", 1,
185 RtpTransceiverDirection::kSendRecv)});
186 EXPECT_THAT(transceiver_.SetOfferedRtpHeaderExtensions(modified_extensions),
187 Property(&RTCError::type, RTCErrorType::INVALID_PARAMETER));
188 EXPECT_EQ(transceiver_.HeaderExtensionsToOffer(), extensions_);
189}
190
191TEST_F(RtpTransceiverTestForHeaderExtensions,
192 RejectsStoppedMandatoryExtensions) {
193 std::vector<RtpHeaderExtensionCapability> modified_extensions = extensions_;
194 // Attempting to stop the mandatory MID extension.
195 modified_extensions[2].direction = RtpTransceiverDirection::kStopped;
196 EXPECT_THAT(transceiver_.SetOfferedRtpHeaderExtensions(modified_extensions),
197 Property(&RTCError::type, RTCErrorType::INVALID_MODIFICATION));
198 EXPECT_EQ(transceiver_.HeaderExtensionsToOffer(), extensions_);
199 modified_extensions = extensions_;
200 // Attempting to stop the mandatory video orientation extension.
201 modified_extensions[3].direction = RtpTransceiverDirection::kStopped;
202 EXPECT_THAT(transceiver_.SetOfferedRtpHeaderExtensions(modified_extensions),
203 Property(&RTCError::type, RTCErrorType::INVALID_MODIFICATION));
204 EXPECT_EQ(transceiver_.HeaderExtensionsToOffer(), extensions_);
Markus Handell0357b3e2020-03-16 13:40:51 +0100205}
206
Amit Hilbuchdd9390c2018-11-13 16:26:05 -0800207} // namespace webrtc