2 files changed, 12 insertions, 6 deletions
diff --git a/remoting/test/fake_port_allocator.cc b/remoting/test/fake_port_allocator.cc
index 65d9047..49db553 100644
--- a/remoting/test/fake_port_allocator.cc
+++ b/remoting/test/fake_port_allocator.cc
@@ -46,8 +46,10 @@ FakePortAllocatorSession::~FakePortAllocatorSession() {}
 
 FakePortAllocator::FakePortAllocator(
     rtc::NetworkManager* network_manager,
-    rtc::PacketSocketFactory* socket_factory)
-    : BasicPortAllocator(network_manager, socket_factory) {
+    rtc::PacketSocketFactory* socket_factory,
+    scoped_refptr<protocol::TransportContext> transport_context)
+    : BasicPortAllocator(network_manager, socket_factory),
+      transport_context_(transport_context) {
   set_flags(cricket::PORTALLOCATOR_DISABLE_TCP |
             cricket::PORTALLOCATOR_ENABLE_SHARED_UFRAG |
             cricket::PORTALLOCATOR_ENABLE_IPV6 |
@@ -78,8 +80,8 @@ FakePortAllocatorFactory::~FakePortAllocatorFactory() {}
 scoped_ptr<cricket::PortAllocator>
 FakePortAllocatorFactory::CreatePortAllocator(
     scoped_refptr<protocol::TransportContext> transport_context) {
-  return make_scoped_ptr(
-      new FakePortAllocator(network_manager_.get(), socket_factory_.get()));
+  return make_scoped_ptr(new FakePortAllocator(
+      network_manager_.get(), socket_factory_.get(), transport_context));
 }
 
 }  // namespace remoting
diff --git a/remoting/test/fake_port_allocator.h b/remoting/test/fake_port_allocator.h
index ed2e095..cc5e6eb 100644
--- a/remoting/test/fake_port_allocator.h
+++ b/remoting/test/fake_port_allocator.h
@@ -20,8 +20,10 @@ class FakePacketSocketFactory;
 
 class FakePortAllocator : public cricket::BasicPortAllocator {
  public:
-  FakePortAllocator(rtc::NetworkManager* network_manager,
-                    rtc::PacketSocketFactory* socket_factory);
+  FakePortAllocator(
+      rtc::NetworkManager* network_manager,
+      rtc::PacketSocketFactory* socket_factory,
+      scoped_refptr<protocol::TransportContext> transport_context_);
   ~FakePortAllocator() override;
 
   // cricket::BasicPortAllocator overrides.
@@ -32,6 +34,8 @@ class FakePortAllocator : public cricket::BasicPortAllocator {
       const std::string& ice_password) override;
 
  private:
+  scoped_refptr<protocol::TransportContext> transport_context_;
+
   DISALLOW_COPY_AND_ASSIGN(FakePortAllocator);
 };