GRPC Stream implementation of HeliumStream

blimp/net/test_common.h

Index: blimp/net/test_common.h
diff --git a/blimp/net/test_common.h b/blimp/net/test_common.h
index 757caef94858e2191f90ab570d8c797c04932416..174c76f1999eae4a59612032a094f2d50708d385 100644
--- a/blimp/net/test_common.h
+++ b/blimp/net/test_common.h
@@ -223,6 +223,12 @@ class MockBlimpMessageProcessor : public BlimpMessageProcessor {
   MOCK_METHOD2(MockableProcessMessage,
                void(const BlimpMessage& message,
                     const net::CompletionCallback& callback));
+
+  void SetTestCompletionCallback(const net::CompletionCallback& callback);
+  BlimpMessage received_msg;
+
+ private:
+  net::CompletionCallback test_callback_;
 };
 
 }  // namespace blimp