Refactoring that removes P2PTransport and DtlsTransport classes.

webrtc/p2p/base/transportcontroller_unittest.cc

 /*
  *  Copyright 2015 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.
  */
 
@@ skipping 731 matching lines @@
 
   // The endpoint that initiated an ICE restart should keep the existing role.
   TransportDescription ice_restart_desc(std::vector<std::string>(), kIceUfrag3,
                                         kIcePwd3, ICEMODE_FULL,
                                         CONNECTIONROLE_ACTPASS, nullptr);
   EXPECT_TRUE(transport_controller_->SetLocalTransportDescription(
       "audio", ice_restart_desc, CA_OFFER, &err));
   EXPECT_EQ(ICEROLE_CONTROLLED, channel->GetIceRole());
 }
 
+// Tests channel role is reversed after receiving ice-lite from remote.
+TEST_F(TransportControllerTest, TestSetRemoteIceLiteInOffer) {
+  FakeTransportChannel* channel = CreateChannel("audio", 1);
+  ASSERT_NE(nullptr, channel);
+  std::string err;
+
+  transport_controller_->SetIceRole(ICEROLE_CONTROLLED);
+  TransportDescription remote_desc(std::vector<std::string>(), kIceUfrag1,
+                                   kIcePwd1, ICEMODE_LITE,
+                                   CONNECTIONROLE_ACTPASS, nullptr);
+  EXPECT_TRUE(transport_controller_->SetRemoteTransportDescription(
+      "audio", remote_desc, CA_OFFER, &err));
+  TransportDescription local_desc(kIceUfrag1, kIcePwd1);
+  ASSERT_TRUE(transport_controller_->SetLocalTransportDescription(
+      "audio", local_desc, CA_ANSWER, nullptr));
+
+  EXPECT_EQ(ICEROLE_CONTROLLING, channel->GetIceRole());
+  EXPECT_EQ(ICEMODE_LITE, channel->remote_ice_mode());
+}
+
+// Tests ice-lite in remote answer.
+TEST_F(TransportControllerTest, TestSetRemoteIceLiteInAnswer) {
+  FakeTransportChannel* channel = CreateChannel("audio", 1);
+  ASSERT_NE(nullptr, channel);
+  std::string err;
+
+  transport_controller_->SetIceRole(ICEROLE_CONTROLLING);
+  TransportDescription local_desc(kIceUfrag1, kIcePwd1);
+  ASSERT_TRUE(transport_controller_->SetLocalTransportDescription(
+      "audio", local_desc, CA_OFFER, nullptr));
+  EXPECT_EQ(ICEROLE_CONTROLLING, channel->GetIceRole());
+  // Channels will be created in ICEFULL_MODE.
+  EXPECT_EQ(ICEMODE_FULL, channel->remote_ice_mode());
+  TransportDescription remote_desc(std::vector<std::string>(), kIceUfrag1,
+                                   kIcePwd1, ICEMODE_LITE, CONNECTIONROLE_NONE,
+                                   nullptr);
+  ASSERT_TRUE(transport_controller_->SetRemoteTransportDescription(
+      "audio", remote_desc, CA_ANSWER, nullptr));
+  EXPECT_EQ(ICEROLE_CONTROLLING, channel->GetIceRole());
+  // After receiving remote description with ICEMODE_LITE, channel should
+  // have mode set to ICEMODE_LITE.
+  EXPECT_EQ(ICEMODE_LITE, channel->remote_ice_mode());
+}
+
 }  // namespace cricket {