Move SelectMediaType from RampUpTester to BaseTest.
This provides a better default for audio-only tests. BUG=None Review-Url: https://codereview.webrtc.org/2794193003 Cr-Commit-Position: refs/heads/master@{#17536}
This commit is contained in:
@ -89,19 +89,6 @@ void RampUpTester::OnVideoStreamsCreated(
|
||||
send_stream_ = send_stream;
|
||||
}
|
||||
|
||||
MediaType RampUpTester::SelectMediaType() {
|
||||
if (num_video_streams_ > 0) {
|
||||
if (num_audio_streams_ > 0) {
|
||||
// Rely on call to set media type from payload type.
|
||||
return MediaType::ANY;
|
||||
} else {
|
||||
return MediaType::VIDEO;
|
||||
}
|
||||
} else {
|
||||
return MediaType::AUDIO;
|
||||
}
|
||||
}
|
||||
|
||||
test::PacketTransport* RampUpTester::CreateSendTransport(Call* sender_call) {
|
||||
send_transport_ = new test::PacketTransport(sender_call, this,
|
||||
test::PacketTransport::kSender,
|
||||
@ -110,13 +97,6 @@ test::PacketTransport* RampUpTester::CreateSendTransport(Call* sender_call) {
|
||||
return send_transport_;
|
||||
}
|
||||
|
||||
test::PacketTransport* RampUpTester::CreateReceiveTransport() {
|
||||
return new test::PacketTransport(nullptr, this,
|
||||
test::PacketTransport::kReceiver,
|
||||
SelectMediaType(),
|
||||
FakeNetworkPipe::Config());
|
||||
}
|
||||
|
||||
size_t RampUpTester::GetNumVideoStreams() const {
|
||||
return num_video_streams_;
|
||||
}
|
||||
|
||||
@ -84,9 +84,7 @@ class RampUpTester : public test::EndToEndTest {
|
||||
void OnVideoStreamsCreated(
|
||||
VideoSendStream* send_stream,
|
||||
const std::vector<VideoReceiveStream*>& receive_streams) override;
|
||||
MediaType SelectMediaType();
|
||||
test::PacketTransport* CreateSendTransport(Call* sender_call) override;
|
||||
test::PacketTransport* CreateReceiveTransport() override;
|
||||
void ModifyVideoConfigs(
|
||||
VideoSendStream::Config* send_config,
|
||||
std::vector<VideoReceiveStream::Config>* receive_configs,
|
||||
|
||||
@ -492,15 +492,28 @@ Call::Config BaseTest::GetReceiverCallConfig() {
|
||||
void BaseTest::OnCallsCreated(Call* sender_call, Call* receiver_call) {
|
||||
}
|
||||
|
||||
MediaType BaseTest::SelectMediaType() {
|
||||
if (GetNumVideoStreams() > 0) {
|
||||
if (GetNumAudioStreams() > 0) {
|
||||
// Relies on PayloadDemuxer to set media type from payload type.
|
||||
return MediaType::ANY;
|
||||
} else {
|
||||
return MediaType::VIDEO;
|
||||
}
|
||||
} else {
|
||||
return MediaType::AUDIO;
|
||||
}
|
||||
}
|
||||
|
||||
test::PacketTransport* BaseTest::CreateSendTransport(Call* sender_call) {
|
||||
return new PacketTransport(sender_call, this, test::PacketTransport::kSender,
|
||||
MediaType::VIDEO,
|
||||
SelectMediaType(),
|
||||
FakeNetworkPipe::Config());
|
||||
}
|
||||
|
||||
test::PacketTransport* BaseTest::CreateReceiveTransport() {
|
||||
return new PacketTransport(nullptr, this, test::PacketTransport::kReceiver,
|
||||
MediaType::VIDEO,
|
||||
SelectMediaType(),
|
||||
FakeNetworkPipe::Config());
|
||||
}
|
||||
|
||||
|
||||
@ -192,7 +192,10 @@ class BaseTest : public RtpRtcpObserver {
|
||||
virtual Call::Config GetReceiverCallConfig();
|
||||
virtual void OnCallsCreated(Call* sender_call, Call* receiver_call);
|
||||
|
||||
// The default implementation creates MediaType::VIDEO transports.
|
||||
// Returns VIDEO for video-only tests, AUDIO for audio-only tests,
|
||||
// and ANY for tests sending audio and video over the same
|
||||
// transport.
|
||||
virtual MediaType SelectMediaType();
|
||||
virtual test::PacketTransport* CreateSendTransport(Call* sender_call);
|
||||
virtual test::PacketTransport* CreateReceiveTransport();
|
||||
|
||||
|
||||
Reference in New Issue
Block a user