GRPC Stream implementation of HeliumStream

blimp/net/test_common.h

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef BLIMP_NET_TEST_COMMON_H_
 #define BLIMP_NET_TEST_COMMON_H_
 
 #include <stddef.h>
 #include <stdint.h>
 
@@ skipping 205 matching lines @@
   ~MockBlimpMessageProcessor() override;
 
   // Adapts calls from ProcessMessage to MockableProcessMessage by
   // unboxing the |message| std::unique_ptr for GMock compatibility.
   void ProcessMessage(std::unique_ptr<BlimpMessage> message,
                       const net::CompletionCallback& callback) override;
 
   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
 
 #endif  // BLIMP_NET_TEST_COMMON_H_