GRPC Stream implementation of HeliumStream

blimp/net/test_common.cc

 // 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.
 
 #include "blimp/net/test_common.h"
 
 #include <string>
 #include <utility>
 
 #include "base/memory/ptr_util.h"
@@ skipping 52 matching lines @@
 
 MockConnectionErrorObserver::~MockConnectionErrorObserver() {}
 
 MockBlimpMessageProcessor::MockBlimpMessageProcessor() {}
 
 MockBlimpMessageProcessor::~MockBlimpMessageProcessor() {}
 
 void MockBlimpMessageProcessor::ProcessMessage(
     std::unique_ptr<BlimpMessage> message,
     const net::CompletionCallback& callback) {
+  if (!test_callback_.is_null()) {
+    test_callback_.Run(net::OK);
+  }
+  received_msg = *message;
   MockableProcessMessage(*message, callback);
 }
 
+void MockBlimpMessageProcessor::SetTestCompletionCallback(
+    const net::CompletionCallback& callback) {
+  test_callback_ = callback;
+}
+
 std::string EncodeHeader(size_t size) {
   std::unique_ptr<char[]> serialized(new char[kPacketHeaderSizeBytes]);
   uint32_t net_size = base::HostToNet32(size);
   memcpy(serialized.get(), &net_size, sizeof(net_size));
   return std::string(serialized.get(), kPacketHeaderSizeBytes);
 }
 
 bool BufferStartsWith(net::GrowableIOBuffer* buf,
                       size_t buf_size,
                       const std::string& str) {
   return (buf_size >= str.size() &&
           str == std::string(buf->data(), str.size()));
 }
 
 }  // namespace blimp