Index: webrtc/p2p/base/dtlstransportchannel_unittest.cc |
diff --git a/webrtc/p2p/base/dtlstransportchannel_unittest.cc b/webrtc/p2p/base/dtlstransportchannel_unittest.cc |
index 3efc1e7df9fddf23130d94be7a3ba363014bebe0..bc237d1e1abcbd258a8ac592f074d379539dddd5 100644 |
--- a/webrtc/p2p/base/dtlstransportchannel_unittest.cc |
+++ b/webrtc/p2p/base/dtlstransportchannel_unittest.cc |
@@ -88,8 +88,8 @@ class DtlsTestClient : public sigslot::has_slots<> { |
transport_.reset( |
new cricket::JsepTransport("dtls content name", certificate_)); |
for (int i = 0; i < count; ++i) { |
- cricket::FakeTransportChannel* fake_ice_channel = |
- new cricket::FakeTransportChannel(transport_->mid(), i); |
+ cricket::FakeIceTransport* fake_ice_channel = |
+ new cricket::FakeIceTransport(transport_->mid(), i); |
fake_ice_channel->SetAsync(true); |
fake_ice_channel->SetAsyncDelay(async_delay_ms); |
// Hook the raw packets so that we can verify they are encrypted. |
@@ -99,8 +99,9 @@ class DtlsTestClient : public sigslot::has_slots<> { |
cricket::DtlsTransportChannelWrapper* channel = |
new cricket::DtlsTransportChannelWrapper(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); |
@@ -111,14 +112,14 @@ class DtlsTestClient : public sigslot::has_slots<> { |
channels_.push_back( |
std::unique_ptr<cricket::DtlsTransportChannelWrapper>(channel)); |
fake_channels_.push_back( |
- std::unique_ptr<cricket::FakeTransportChannel>(fake_ice_channel)); |
+ std::unique_ptr<cricket::FakeIceTransport>(fake_ice_channel)); |
transport_->AddChannel(channel, i); |
} |
} |
cricket::JsepTransport* transport() { return transport_.get(); } |
- cricket::FakeTransportChannel* GetFakeChannel(int component) { |
+ cricket::FakeIceTransport* GetFakeChannel(int component) { |
for (const auto& ch : fake_channels_) { |
if (ch->component() == component) { |
return ch.get(); |
@@ -231,12 +232,12 @@ class DtlsTestClient : public sigslot::has_slots<> { |
return true; |
} |
- bool all_raw_channels_writable() const { |
+ bool all_ice_transport_writable() const { |
if (channels_.empty()) { |
return false; |
} |
for (const auto& channel : channels_) { |
- if (!channel->channel()->writable()) { |
+ if (!channel->ice_transport()->writable()) { |
return false; |
} |
} |
@@ -434,7 +435,7 @@ class DtlsTestClient : public sigslot::has_slots<> { |
private: |
std::string name_; |
rtc::scoped_refptr<rtc::RTCCertificate> certificate_; |
- std::vector<std::unique_ptr<cricket::FakeTransportChannel>> fake_channels_; |
+ std::vector<std::unique_ptr<cricket::FakeIceTransport>> fake_channels_; |
std::vector<std::unique_ptr<cricket::DtlsTransportChannelWrapper>> channels_; |
std::unique_ptr<cricket::JsepTransport> transport_; |
size_t packet_size_ = 0u; |
@@ -639,8 +640,8 @@ class DtlsTransportChannelTest : public DtlsTransportChannelTestBase, |
// Test that transport negotiation of ICE, no DTLS works properly. |
TEST_F(DtlsTransportChannelTest, TestChannelSetupIce) { |
Negotiate(); |
- cricket::FakeTransportChannel* channel1 = client1_.GetFakeChannel(0); |
- cricket::FakeTransportChannel* channel2 = client2_.GetFakeChannel(0); |
+ cricket::FakeIceTransport* channel1 = client1_.GetFakeChannel(0); |
+ cricket::FakeIceTransport* channel2 = client2_.GetFakeChannel(0); |
ASSERT_TRUE(channel1 != NULL); |
ASSERT_TRUE(channel2 != NULL); |
EXPECT_EQ(cricket::ICEROLE_CONTROLLING, channel1->GetIceRole()); |
@@ -993,12 +994,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 +1104,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 +1112,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_); |