diff --git a/talk/p2p/base/relayserver_unittest.cc b/talk/p2p/base/relayserver_unittest.cc index 821c373a13..5d77ca6a9a 100644 --- a/talk/p2p/base/relayserver_unittest.cc +++ b/talk/p2p/base/relayserver_unittest.cc @@ -36,7 +36,6 @@ #include "webrtc/base/ssladapter.h" #include "webrtc/base/testclient.h" #include "webrtc/base/thread.h" -#include "webrtc/base/virtualsocketserver.h" using rtc::SocketAddress; using namespace cricket; @@ -55,25 +54,23 @@ static const char* msg2 = "Lobster Thermidor a Crevette with a mornay sauce..."; class RelayServerTest : public testing::Test { public: RelayServerTest() - : pss_(new rtc::PhysicalSocketServer), - ss_(new rtc::VirtualSocketServer(pss_.get())), - ss_scope_(ss_.get()), + : main_(rtc::Thread::Current()), ss_(main_->socketserver()), username_(rtc::CreateRandomString(12)), - password_(rtc::CreateRandomString(12)) {} - + password_(rtc::CreateRandomString(12)) { + } protected: virtual void SetUp() { - server_.reset(new RelayServer(rtc::Thread::Current())); + server_.reset(new RelayServer(main_)); server_->AddInternalSocket( - rtc::AsyncUDPSocket::Create(ss_.get(), server_int_addr)); + rtc::AsyncUDPSocket::Create(ss_, server_int_addr)); server_->AddExternalSocket( - rtc::AsyncUDPSocket::Create(ss_.get(), server_ext_addr)); + rtc::AsyncUDPSocket::Create(ss_, server_ext_addr)); client1_.reset(new rtc::TestClient( - rtc::AsyncUDPSocket::Create(ss_.get(), client1_addr))); + rtc::AsyncUDPSocket::Create(ss_, client1_addr))); client2_.reset(new rtc::TestClient( - rtc::AsyncUDPSocket::Create(ss_.get(), client2_addr))); + rtc::AsyncUDPSocket::Create(ss_, client2_addr))); } void Allocate() { @@ -179,9 +176,8 @@ class RelayServerTest : public testing::Test { msg->AddAttribute(attr); } - rtc::scoped_ptr pss_; - rtc::scoped_ptr ss_; - rtc::SocketServerScope ss_scope_; + rtc::Thread* main_; + rtc::SocketServer* ss_; rtc::scoped_ptr server_; rtc::scoped_ptr client1_; rtc::scoped_ptr client2_; diff --git a/talk/p2p/base/session_unittest.cc b/talk/p2p/base/session_unittest.cc index 9bd41a6f1c..4674d2c232 100644 --- a/talk/p2p/base/session_unittest.cc +++ b/talk/p2p/base/session_unittest.cc @@ -54,9 +54,7 @@ #include "webrtc/base/logging.h" #include "webrtc/base/natserver.h" #include "webrtc/base/natsocketfactory.h" -#include "webrtc/base/physicalsocketserver.h" #include "webrtc/base/stringencode.h" -#include "webrtc/base/virtualsocketserver.h" using cricket::SignalingProtocol; using cricket::PROTOCOL_HYBRID; @@ -1123,10 +1121,6 @@ class TestClient : public sigslot::has_slots<> { class SessionTest : public testing::Test { protected: - SessionTest() - : pss_(new rtc::PhysicalSocketServer), - ss_(new rtc::VirtualSocketServer(pss_.get())), - ss_scope_(ss_.get()) {} virtual void SetUp() { // Seed needed for each test to satisfy expectations. rtc::SetRandomTestMode(true); @@ -2282,10 +2276,6 @@ class SessionTest : public testing::Test { EXPECT_EQ(2ul, stats.proxy_to_transport.size()); EXPECT_EQ(2ul, stats.transport_stats.size()); } - - rtc::scoped_ptr pss_; - rtc::scoped_ptr ss_; - rtc::SocketServerScope ss_scope_; }; // For each of these, "X => Y = Z" means "if a client with protocol X