diff --git a/test/network/cross_traffic_unittest.cc b/test/network/cross_traffic_unittest.cc index cfa80cd1ff..94a2dffd1e 100644 --- a/test/network/cross_traffic_unittest.cc +++ b/test/network/cross_traffic_unittest.cc @@ -126,8 +126,9 @@ TEST(TcpMessageRouteTest, DeliveredOnLossyNetwork) { auto ret = send; ret.loss_percent = 10; - auto* tcp_route = net.CreateTcpRoute({net.CreateEmulatedNode(send)}, - {net.CreateEmulatedNode(ret)}); + auto* tcp_route = + net.CreateTcpRoute(net.CreateRoute({net.CreateEmulatedNode(send)}), + net.CreateRoute({net.CreateEmulatedNode(ret)})); int deliver_count = 0; // 100 kB is more than what fits into a single packet. constexpr size_t kMessageSize = 100000; diff --git a/test/network/network_emulation_manager.cc b/test/network/network_emulation_manager.cc index b2ccddf24a..a2d8d90d67 100644 --- a/test/network/network_emulation_manager.cc +++ b/test/network/network_emulation_manager.cc @@ -229,10 +229,10 @@ FakeTcpCrossTraffic* NetworkEmulationManagerImpl::StartFakeTcpCrossTraffic( } TcpMessageRoute* NetworkEmulationManagerImpl::CreateTcpRoute( - std::vector send_link, - std::vector ret_link) { - auto tcp_route = std::make_unique( - clock_, task_queue_.Get(), CreateRoute(send_link), CreateRoute(ret_link)); + EmulatedRoute* send_route, + EmulatedRoute* ret_route) { + auto tcp_route = std::make_unique(clock_, task_queue_.Get(), + send_route, ret_route); auto* route_ptr = tcp_route.get(); task_queue_.PostTask([this, tcp_route = std::move(tcp_route)]() mutable { tcp_message_routes_.push_back(std::move(tcp_route)); diff --git a/test/network/network_emulation_manager.h b/test/network/network_emulation_manager.h index 8076e6cea2..da5b29553b 100644 --- a/test/network/network_emulation_manager.h +++ b/test/network/network_emulation_manager.h @@ -77,8 +77,8 @@ class NetworkEmulationManagerImpl : public NetworkEmulationManager { std::vector ret_link, FakeTcpConfig config); - TcpMessageRoute* CreateTcpRoute(std::vector send_link, - std::vector ret_link); + TcpMessageRoute* CreateTcpRoute(EmulatedRoute* send_route, + EmulatedRoute* ret_route); void StopCrossTraffic(FakeTcpCrossTraffic* traffic);