GRPC Stream implementation of HeliumStream

blimp/engine/browser_tests/blimp_browser_test.h

Index: blimp/engine/browser_tests/blimp_browser_test.h
diff --git a/blimp/engine/browser_tests/blimp_browser_test.h b/blimp/engine/browser_tests/blimp_browser_test.h
index 4f54e84644326aa7af52e6cde42ff7c6270a44c5..7c3628fbee00ac0d7a7ee692c070da6999f57d59 100644
--- a/blimp/engine/browser_tests/blimp_browser_test.h
+++ b/blimp/engine/browser_tests/blimp_browser_test.h
@@ -9,6 +9,7 @@
 
 #include "base/synchronization/waitable_event.h"
 #include "base/threading/thread_restrictions.h"
+#include "blimp/common/public/session/assignment_options.h"
 #include "content/public/test/browser_test_base.h"
 
 namespace base {
@@ -65,11 +66,12 @@ class BlimpBrowserTest : public content::BrowserTestBase {
   void TearDownOnMainThread() override;
 
  private:
-  // Receives the port number from an asynchronously connected socket.
-  // Calls SignalCompletion() when set.
-  void OnGetEnginePortCompletion(uint16_t port);
+  // Receives the engine assignment after it completes setup to be passed to
+  // the client as is (independent of the specific transport).
+  void OnGetEngineSetupCompletion(AssignmentOptions options);
 
-  uint16_t engine_port_;
+  // Engine assignment options filled after the engine is setup.
+  AssignmentOptions assignment_options_;
 
   // Used to signal the completion of asynchronous processing to
   // RunUntilCompletion().