You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
190 lines
7.8 KiB
190 lines
7.8 KiB
4 months ago
|
/*
|
||
|
|
* Copyright 2020 The WebRTC project authors. All Rights Reserved.
|
||
|
|
*
|
||
|
|
* Use of this source code is governed by a BSD-style license
|
||
|
|
* that can be found in the LICENSE file in the root of the source
|
||
|
|
* tree. An additional intellectual property rights grant can be found
|
||
|
|
* in the file PATENTS. All contributing project authors may
|
||
|
|
* be found in the AUTHORS file in the root of the source tree.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <memory>
|
||
|
|
#include <tuple>
|
||
|
|
|
||
|
|
#include "api/rtc_event_log/rtc_event_log_factory.h"
|
||
|
|
#include "api/task_queue/default_task_queue_factory.h"
|
||
|
|
#include "media/base/fake_media_engine.h"
|
||
|
|
#include "p2p/base/fake_port_allocator.h"
|
||
|
|
#include "pc/media_session.h"
|
||
|
|
#include "pc/peer_connection_wrapper.h"
|
||
|
|
#include "rtc_base/gunit.h"
|
||
|
|
#include "rtc_base/strings/string_builder.h"
|
||
|
|
#include "test/gmock.h"
|
||
|
|
|
||
|
|
namespace webrtc {
|
||
|
|
|
||
|
|
using ::testing::Combine;
|
||
|
|
using ::testing::ElementsAre;
|
||
|
|
using ::testing::Field;
|
||
|
|
using ::testing::Return;
|
||
|
|
using ::testing::Values;
|
||
|
|
|
||
|
|
class PeerConnectionHeaderExtensionTest
|
||
|
|
: public ::testing::TestWithParam<
|
||
|
|
std::tuple<cricket::MediaType, SdpSemantics>> {
|
||
|
|
protected:
|
||
|
|
PeerConnectionHeaderExtensionTest()
|
||
|
|
: extensions_(
|
||
|
|
{RtpHeaderExtensionCapability("uri1",
|
||
|
|
1,
|
||
|
|
RtpTransceiverDirection::kStopped),
|
||
|
|
RtpHeaderExtensionCapability("uri2",
|
||
|
|
2,
|
||
|
|
RtpTransceiverDirection::kSendOnly),
|
||
|
|
RtpHeaderExtensionCapability("uri3",
|
||
|
|
3,
|
||
|
|
RtpTransceiverDirection::kRecvOnly),
|
||
|
|
RtpHeaderExtensionCapability(
|
||
|
|
"uri4",
|
||
|
|
4,
|
||
|
|
RtpTransceiverDirection::kSendRecv)}) {}
|
||
|
|
|
||
|
|
std::unique_ptr<PeerConnectionWrapper> CreatePeerConnection(
|
||
|
|
cricket::MediaType media_type,
|
||
|
|
absl::optional<SdpSemantics> semantics) {
|
||
|
|
auto voice = std::make_unique<cricket::FakeVoiceEngine>();
|
||
|
|
auto video = std::make_unique<cricket::FakeVideoEngine>();
|
||
|
|
if (media_type == cricket::MediaType::MEDIA_TYPE_AUDIO)
|
||
|
|
voice->SetRtpHeaderExtensions(extensions_);
|
||
|
|
else
|
||
|
|
video->SetRtpHeaderExtensions(extensions_);
|
||
|
|
auto media_engine = std::make_unique<cricket::CompositeMediaEngine>(
|
||
|
|
std::move(voice), std::move(video));
|
||
|
|
PeerConnectionFactoryDependencies factory_dependencies;
|
||
|
|
factory_dependencies.network_thread = rtc::Thread::Current();
|
||
|
|
factory_dependencies.worker_thread = rtc::Thread::Current();
|
||
|
|
factory_dependencies.signaling_thread = rtc::Thread::Current();
|
||
|
|
factory_dependencies.task_queue_factory = CreateDefaultTaskQueueFactory();
|
||
|
|
factory_dependencies.media_engine = std::move(media_engine);
|
||
|
|
factory_dependencies.call_factory = CreateCallFactory();
|
||
|
|
factory_dependencies.event_log_factory =
|
||
|
|
std::make_unique<RtcEventLogFactory>(
|
||
|
|
factory_dependencies.task_queue_factory.get());
|
||
|
|
|
||
|
|
auto pc_factory =
|
||
|
|
CreateModularPeerConnectionFactory(std::move(factory_dependencies));
|
||
|
|
|
||
|
|
auto fake_port_allocator = std::make_unique<cricket::FakePortAllocator>(
|
||
|
|
rtc::Thread::Current(), nullptr);
|
||
|
|
auto observer = std::make_unique<MockPeerConnectionObserver>();
|
||
|
|
PeerConnectionInterface::RTCConfiguration config;
|
||
|
|
if (semantics)
|
||
|
|
config.sdp_semantics = *semantics;
|
||
|
|
auto pc = pc_factory->CreatePeerConnection(
|
||
|
|
config, std::move(fake_port_allocator), nullptr, observer.get());
|
||
|
|
observer->SetPeerConnectionInterface(pc.get());
|
||
|
|
return std::make_unique<PeerConnectionWrapper>(pc_factory, pc,
|
||
|
|
std::move(observer));
|
||
|
|
}
|
||
|
|
|
||
|
|
std::vector<RtpHeaderExtensionCapability> extensions_;
|
||
|
|
};
|
||
|
|
|
||
|
|
TEST_P(PeerConnectionHeaderExtensionTest, TransceiverOffersHeaderExtensions) {
|
||
|
|
cricket::MediaType media_type;
|
||
|
|
SdpSemantics semantics;
|
||
|
|
std::tie(media_type, semantics) = GetParam();
|
||
|
|
if (semantics != SdpSemantics::kUnifiedPlan)
|
||
|
|
return;
|
||
|
|
std::unique_ptr<PeerConnectionWrapper> wrapper =
|
||
|
|
CreatePeerConnection(media_type, semantics);
|
||
|
|
auto transceiver = wrapper->AddTransceiver(media_type);
|
||
|
|
EXPECT_EQ(transceiver->HeaderExtensionsToOffer(), extensions_);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_P(PeerConnectionHeaderExtensionTest,
|
||
|
|
SenderReceiverCapabilitiesReturnNotStoppedExtensions) {
|
||
|
|
cricket::MediaType media_type;
|
||
|
|
SdpSemantics semantics;
|
||
|
|
std::tie(media_type, semantics) = GetParam();
|
||
|
|
std::unique_ptr<PeerConnectionWrapper> wrapper =
|
||
|
|
CreatePeerConnection(media_type, semantics);
|
||
|
|
EXPECT_THAT(wrapper->pc_factory()
|
||
|
|
->GetRtpSenderCapabilities(media_type)
|
||
|
|
.header_extensions,
|
||
|
|
ElementsAre(Field(&RtpHeaderExtensionCapability::uri, "uri2"),
|
||
|
|
Field(&RtpHeaderExtensionCapability::uri, "uri3"),
|
||
|
|
Field(&RtpHeaderExtensionCapability::uri, "uri4")));
|
||
|
|
EXPECT_EQ(wrapper->pc_factory()
|
||
|
|
->GetRtpReceiverCapabilities(media_type)
|
||
|
|
.header_extensions,
|
||
|
|
wrapper->pc_factory()
|
||
|
|
->GetRtpSenderCapabilities(media_type)
|
||
|
|
.header_extensions);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_P(PeerConnectionHeaderExtensionTest, OffersUnstoppedDefaultExtensions) {
|
||
|
|
cricket::MediaType media_type;
|
||
|
|
SdpSemantics semantics;
|
||
|
|
std::tie(media_type, semantics) = GetParam();
|
||
|
|
if (semantics != SdpSemantics::kUnifiedPlan)
|
||
|
|
return;
|
||
|
|
std::unique_ptr<PeerConnectionWrapper> wrapper =
|
||
|
|
CreatePeerConnection(media_type, semantics);
|
||
|
|
auto transceiver = wrapper->AddTransceiver(media_type);
|
||
|
|
auto session_description = wrapper->CreateOffer();
|
||
|
|
EXPECT_THAT(session_description->description()
|
||
|
|
->contents()[0]
|
||
|
|
.media_description()
|
||
|
|
->rtp_header_extensions(),
|
||
|
|
ElementsAre(Field(&RtpExtension::uri, "uri2"),
|
||
|
|
Field(&RtpExtension::uri, "uri3"),
|
||
|
|
Field(&RtpExtension::uri, "uri4")));
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_P(PeerConnectionHeaderExtensionTest, OffersUnstoppedModifiedExtensions) {
|
||
|
|
cricket::MediaType media_type;
|
||
|
|
SdpSemantics semantics;
|
||
|
|
std::tie(media_type, semantics) = GetParam();
|
||
|
|
if (semantics != SdpSemantics::kUnifiedPlan)
|
||
|
|
return;
|
||
|
|
std::unique_ptr<PeerConnectionWrapper> wrapper =
|
||
|
|
CreatePeerConnection(media_type, semantics);
|
||
|
|
auto transceiver = wrapper->AddTransceiver(media_type);
|
||
|
|
auto modified_extensions = transceiver->HeaderExtensionsToOffer();
|
||
|
|
modified_extensions[0].direction = RtpTransceiverDirection::kSendRecv;
|
||
|
|
modified_extensions[3].direction = RtpTransceiverDirection::kStopped;
|
||
|
|
EXPECT_TRUE(
|
||
|
|
transceiver->SetOfferedRtpHeaderExtensions(modified_extensions).ok());
|
||
|
|
auto session_description = wrapper->CreateOffer();
|
||
|
|
EXPECT_THAT(session_description->description()
|
||
|
|
->contents()[0]
|
||
|
|
.media_description()
|
||
|
|
->rtp_header_extensions(),
|
||
|
|
ElementsAre(Field(&RtpExtension::uri, "uri1"),
|
||
|
|
Field(&RtpExtension::uri, "uri2"),
|
||
|
|
Field(&RtpExtension::uri, "uri3")));
|
||
|
|
}
|
||
|
|
|
||
|
|
INSTANTIATE_TEST_SUITE_P(
|
||
|
|
,
|
||
|
|
PeerConnectionHeaderExtensionTest,
|
||
|
|
Combine(Values(SdpSemantics::kPlanB, SdpSemantics::kUnifiedPlan),
|
||
|
|
Values(cricket::MediaType::MEDIA_TYPE_AUDIO,
|
||
|
|
cricket::MediaType::MEDIA_TYPE_VIDEO)),
|
||
|
|
[](const testing::TestParamInfo<
|
||
|
|
PeerConnectionHeaderExtensionTest::ParamType>& info) {
|
||
|
|
cricket::MediaType media_type;
|
||
|
|
SdpSemantics semantics;
|
||
|
|
std::tie(media_type, semantics) = info.param;
|
||
|
|
return (rtc::StringBuilder("With")
|
||
|
|
<< (semantics == SdpSemantics::kPlanB ? "PlanB" : "UnifiedPlan")
|
||
|
|
<< "And"
|
||
|
|
<< (media_type == cricket::MediaType::MEDIA_TYPE_AUDIO ? "Voice"
|
||
|
|
: "Video")
|
||
|
|
<< "Engine")
|
||
|
|
.str();
|
||
|
|
});
|
||
|
|
|
||
|
|
} // namespace webrtc
|