GRPC Stream implementation of HeliumStream

blimp/net/tcp_transport_unittest.cc

Index: blimp/net/tcp_transport_unittest.cc
diff --git a/blimp/net/tcp_transport_unittest.cc b/blimp/net/tcp_transport_unittest.cc
index 0a80efefd0ac88074d05a63c36304a917b7ed4e9..06c029c5fffac9a451aa54c7231ea35541ad576d 100644
--- a/blimp/net/tcp_transport_unittest.cc
+++ b/blimp/net/tcp_transport_unittest.cc
@@ -43,9 +43,9 @@ class TCPTransportTest : public testing::Test {
   }
 
   net::IPEndPoint GetLocalEndpoint() const {
-    net::IPEndPoint local_address;
-    engine_.GetLocalAddress(&local_address);
-    return local_address;
+    AssignmentOptions assignment_options;
+    engine_.GetAssignmentOptions(&assignment_options);
+    return assignment_options.engine_endpoint;
   }
 
   std::string payload_1_ = "foo";