GRPC Stream implementation of HeliumStream

blimp/engine/session/blimp_engine_session.h

Index: blimp/engine/session/blimp_engine_session.h
diff --git a/blimp/engine/session/blimp_engine_session.h b/blimp/engine/session/blimp_engine_session.h
index f815a5a4102bbea8adf17761084503447995292e..4d3964f88ac2043e69c2a1ea48b7a17015befed5 100644
--- a/blimp/engine/session/blimp_engine_session.h
+++ b/blimp/engine/session/blimp_engine_session.h
@@ -11,6 +11,7 @@
 #include <vector>
 
 #include "base/macros.h"
+#include "blimp/common/assignment_options.h"
 #include "blimp/common/proto/blimp_message.pb.h"
 #include "blimp/engine/feature/engine_render_widget_feature.h"
 #include "blimp/engine/feature/engine_settings_feature.h"
@@ -75,7 +76,7 @@ class BlimpEngineSession : public BlimpMessageProcessor,
                            public content::WebContentsDelegate,
                            public ui::InputMethodObserver {
  public:
-  using GetPortCallback = base::Callback<void(uint16_t)>;
+  using GetAssignmentOptionsCallback = base::Callback<void(AssignmentOptions)>;
 
   BlimpEngineSession(std::unique_ptr<BlimpBrowserContext> browser_context,
                      net::NetLog* net_log,
@@ -97,7 +98,7 @@ class BlimpEngineSession : public BlimpMessageProcessor,
   BlobChannelService* GetBlobChannelService();
 
   // Gets Engine's listening port. Invokes callback with the allocated port.
-  void GetEnginePortForTesting(const GetPortCallback& callback);
+  void GetAssignmentOptions(const GetAssignmentOptionsCallback& callback);
 
   // BlimpMessageProcessor implementation.
   // This object handles incoming TAB_CONTROL and NAVIGATION messages directly.