Index: webrtc/p2p/base/dtlstransportchannel_unittest.cc |
diff --git a/webrtc/p2p/base/dtlstransportchannel_unittest.cc b/webrtc/p2p/base/dtlstransportchannel_unittest.cc |
index bff2e7da5d27ff18285a402d1f65c78a0a54ea0a..4b21b2f9238e0c5d0d7f7b7d9d56d9279a225ba4 100644 |
--- a/webrtc/p2p/base/dtlstransportchannel_unittest.cc |
+++ b/webrtc/p2p/base/dtlstransportchannel_unittest.cc |
@@ -96,11 +96,12 @@ class DtlsTestClient : public sigslot::has_slots<> { |
fake_ice_channel->SignalReadPacket.connect( |
this, &DtlsTestClient::OnFakeTransportChannelReadPacket); |
- cricket::DtlsTransportChannelWrapper* channel = |
- new cricket::DtlsTransportChannelWrapper(fake_ice_channel); |
+ cricket::DtlsTransport* channel = |
+ new cricket::DtlsTransport(fake_ice_channel); |
channel->SetLocalCertificate(certificate_); |
- channel->SetIceRole(role); |
- channel->SetIceTiebreaker((role == cricket::ICEROLE_CONTROLLING) ? 1 : 2); |
+ channel->ice_transport()->SetIceRole(role); |
+ channel->ice_transport()->SetIceTiebreaker( |
+ (role == cricket::ICEROLE_CONTROLLING) ? 1 : 2); |
channel->SetSslMaxProtocolVersion(ssl_max_version_); |
channel->SignalWritableState.connect(this, |
&DtlsTestClient::OnTransportChannelWritableState); |
@@ -108,8 +109,7 @@ class DtlsTestClient : public sigslot::has_slots<> { |
&DtlsTestClient::OnTransportChannelReadPacket); |
channel->SignalSentPacket.connect( |
this, &DtlsTestClient::OnTransportChannelSentPacket); |
- channels_.push_back( |
- std::unique_ptr<cricket::DtlsTransportChannelWrapper>(channel)); |
+ channels_.push_back(std::unique_ptr<cricket::DtlsTransport>(channel)); |
fake_channels_.push_back( |
std::unique_ptr<cricket::FakeIceTransport>(fake_ice_channel)); |
transport_->AddChannel(channel, i); |
@@ -127,7 +127,7 @@ class DtlsTestClient : public sigslot::has_slots<> { |
return nullptr; |
} |
- cricket::DtlsTransportChannelWrapper* GetDtlsChannel(int component) { |
+ cricket::DtlsTransport* GetDtlsChannel(int component) { |
for (const auto& ch : channels_) { |
if (ch->component() == component) { |
return ch.get(); |
@@ -231,12 +231,12 @@ class DtlsTestClient : public sigslot::has_slots<> { |
return true; |
} |
- bool all_raw_channels_writable() const { |
+ bool all_ice_transport_writable() const { |
pthatcher1
2017/01/17 20:16:48
all_ice_transports_writable()
|
if (channels_.empty()) { |
return false; |
} |
for (const auto& channel : channels_) { |
pthatcher1
2017/01/17 20:16:48
transport : transports_
|
- if (!channel->channel()->writable()) { |
+ if (!channel->ice_transport()->writable()) { |
return false; |
} |
} |
@@ -435,7 +435,7 @@ class DtlsTestClient : public sigslot::has_slots<> { |
std::string name_; |
rtc::scoped_refptr<rtc::RTCCertificate> certificate_; |
std::vector<std::unique_ptr<cricket::FakeIceTransport>> fake_channels_; |
- std::vector<std::unique_ptr<cricket::DtlsTransportChannelWrapper>> channels_; |
+ std::vector<std::unique_ptr<cricket::DtlsTransport>> channels_; |
std::unique_ptr<cricket::JsepTransport> transport_; |
size_t packet_size_ = 0u; |
std::set<int> received_; |
@@ -993,12 +993,12 @@ TEST_F(DtlsTransportChannelTest, TestRetransmissionSchedule) { |
// Make client2_ writable, but not client1_. |
// This means client1_ will send DTLS client hellos but get no response. |
EXPECT_TRUE(client2_.Connect(&client1_, true)); |
- EXPECT_TRUE_SIMULATED_WAIT(client2_.all_raw_channels_writable(), kTimeout, |
+ EXPECT_TRUE_SIMULATED_WAIT(client2_.all_ice_transport_writable(), kTimeout, |
fake_clock_); |
// Wait for the first client hello to be sent. |
EXPECT_EQ_WAIT(1, client1_.received_dtls_client_hellos(), kTimeout); |
- EXPECT_FALSE(client1_.all_raw_channels_writable()); |
+ EXPECT_FALSE(client1_.all_ice_transport_writable()); |
static int timeout_schedule_ms[] = {50, 100, 200, 400, 800, 1600, |
3200, 6400, 12800, 25600, 51200, 60000}; |
@@ -1103,7 +1103,7 @@ class DtlsEventOrderingTest |
break; |
case CALLER_WRITABLE: |
EXPECT_TRUE(client1_.Connect(&client2_, true)); |
- EXPECT_TRUE_SIMULATED_WAIT(client1_.all_raw_channels_writable(), |
+ EXPECT_TRUE_SIMULATED_WAIT(client1_.all_ice_transport_writable(), |
kTimeout, fake_clock_); |
break; |
case CALLER_RECEIVES_CLIENTHELLO: |
@@ -1111,7 +1111,7 @@ class DtlsEventOrderingTest |
EXPECT_EQ(0, client1_.received_dtls_client_hellos()); |
// Making client2_ writable will cause it to send the ClientHello. |
EXPECT_TRUE(client2_.Connect(&client1_, true)); |
- EXPECT_TRUE_SIMULATED_WAIT(client2_.all_raw_channels_writable(), |
+ EXPECT_TRUE_SIMULATED_WAIT(client2_.all_ice_transport_writable(), |
kTimeout, fake_clock_); |
EXPECT_EQ_SIMULATED_WAIT(1, client1_.received_dtls_client_hellos(), |
kTimeout, fake_clock_); |