GRPC Stream implementation of HeliumStream

blimp/net/engine_connection_manager_unittest.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "blimp/net/engine_connection_manager.h"
 
 #include <stddef.h>
 
 #include <string>
 #include <utility>
 
 #include "base/callback_helpers.h"
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
+#include "blimp/common/assignment_options.h"
 #include "blimp/net/blimp_transport.h"
 #include "blimp/net/common.h"
 #include "blimp/net/connection_error_observer.h"
 #include "blimp/net/tcp_client_transport.h"
 #include "blimp/net/tcp_connection.h"
 #include "blimp/net/test_common.h"
 #include "net/base/completion_callback.h"
 #include "net/base/io_buffer.h"
 #include "net/base/ip_address.h"
 #include "net/base/net_errors.h"
@@ skipping 18 matching lines @@
  protected:
   testing::StrictMock<MockConnectionHandler> connection_handler_;
   std::unique_ptr<EngineConnectionManager> manager_;
   base::MessageLoopForIO message_loop_;
 };
 
 TEST_F(EngineConnectionManagerTest, ConnectionSucceeds) {
   EXPECT_CALL(connection_handler_, HandleConnectionPtr(_)).Times(1);
 
   net::IPAddress ip_addr(net::IPAddress::IPv4Localhost());
-  net::IPEndPoint engine_endpoint(ip_addr, 0);
-
-  manager_->ConnectTransport(&engine_endpoint,
+  AssignmentOptions assignment_options;
+  assignment_options.engine_endpoint = net::IPEndPoint(ip_addr, 0);
+  manager_->ConnectTransport(&assignment_options,
                              EngineConnectionManager::EngineTransportType::TCP);
 
   net::TestCompletionCallback connect_callback;
-  TCPClientTransport client(engine_endpoint, nullptr);
+  TCPClientTransport client(assignment_options.engine_endpoint, nullptr);
   client.Connect(connect_callback.callback());
   EXPECT_EQ(net::OK, connect_callback.WaitForResult());
 }
 
 }  // namespace blimp