GRPC Stream implementation of HeliumStream

blimp/net/tcp_engine_transport.h

Index: blimp/net/tcp_engine_transport.h
diff --git a/blimp/net/tcp_engine_transport.h b/blimp/net/tcp_engine_transport.h
index b7e85e06b83ec98e60d4439e80dd710370d0c846..4ef00f4b83533c9efc86349e535eb5579d930268 100644
--- a/blimp/net/tcp_engine_transport.h
+++ b/blimp/net/tcp_engine_transport.h
@@ -9,6 +9,7 @@
 
 #include "base/callback.h"
 #include "base/macros.h"
+#include "blimp/common/public/session/assignment_options.h"
 #include "blimp/net/blimp_engine_transport.h"
 #include "blimp/net/blimp_net_export.h"
 #include "net/base/ip_endpoint.h"
@@ -28,15 +29,15 @@ class MessagePort;
 class BLIMP_NET_EXPORT TCPEngineTransport : public BlimpEngineTransport {
  public:
   // Caller retains the ownership of |net_log|.
-  TCPEngineTransport(const net::IPEndPoint& address,
-                     net::NetLog* net_log);
+  TCPEngineTransport(const net::IPEndPoint& address, net::NetLog* net_log);
   ~TCPEngineTransport() override;
 
   // BlimpTransport implementation.
   void Connect(const net::CompletionCallback& callback) override;
   std::unique_ptr<BlimpConnection> MakeConnection() override;
   const char* GetName() const override;
-  void GetLocalAddress(net::IPEndPoint* address) const override;
+  void GetAssignmentOptions(
+      AssignmentOptions* assignment_options) const override;
 
   // Internal use only except tests.
   std::unique_ptr<MessagePort> TakeMessagePort();