Blimp: Add BlobChannel Helium messages and delegate impls.

blimp/net/blob_channel/helium_blob_channel_unittest.cc

+// Copyright 2016 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 <memory>
+
+#include "base/memory/ptr_util.h"
+#include "blimp/common/blob_cache/id_util.h"
+#include "blimp/common/blob_cache/test_util.h"
+#include "blimp/net/blob_channel/helium_blob_receiver_delegate.h"
+#include "blimp/net/blob_channel/helium_blob_sender_delegate.h"
+#include "blimp/net/blob_channel/mock_blob_channel_receiver.h"
+#include "blimp/net/test_common.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace blimp {
+namespace {
+
+using testing::_;
+using testing::SaveArg;
+
+const char kBlobId[] = "foo1";
+const char kBlobPayload[] = "bar1";
+
+class HeliumBlobChannelTest : public testing::Test {
+ public:
+  HeliumBlobChannelTest() : receiver_delegate_(new HeliumBlobReceiverDelegate) {
+    receiver_delegate_->SetReceiver(&mock_receiver_);
+    sender_delegate_.set_outgoing_message_processor(
+        base::WrapUnique(receiver_delegate_));
+  }
+
+  ~HeliumBlobChannelTest() override {}
+
+ protected:
+  const std::string blob_id_ = CalculateBlobId(kBlobId);
+
+  MockBlobChannelReceiver mock_receiver_;
+  HeliumBlobReceiverDelegate* receiver_delegate_;
+  HeliumBlobSenderDelegate sender_delegate_;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(HeliumBlobChannelTest);
+};
+
+// Verifies the content of BlimpMessages generated by the Sender.
+TEST_F(HeliumBlobChannelTest, TransferBlobContents) {
+  BlimpMessage captured_msg;
+  MockBlimpMessageProcessor* mock_processor = new MockBlimpMessageProcessor;
+  EXPECT_CALL(*mock_processor, MockableProcessMessage(_, _))
+      .WillOnce(SaveArg<0>(&captured_msg));
+  sender_delegate_.set_outgoing_message_processor(
+      base::WrapUnique(mock_processor));
+
+  sender_delegate_.DeliverBlob(blob_id_, CreateBlobDataPtr(kBlobPayload));
+
+  EXPECT_EQ(BlobChannelMessage::TypeCase::kTransferBlob,
+            captured_msg.blob_channel().type_case());
+  EXPECT_EQ(blob_id_, captured_msg.blob_channel().transfer_blob().blob_id());
+  EXPECT_EQ(kBlobPayload,
+            captured_msg.blob_channel().transfer_blob().payload());
+}
+
+// Verifies that the Receiver understands messages sent by the Sender.
+TEST_F(HeliumBlobChannelTest, TransferBlobCompatibility) {
+  EXPECT_CALL(mock_receiver_,
+              OnBlobReceived(blob_id_, BlobDataPtrEqualsString(kBlobPayload)));
+  sender_delegate_.DeliverBlob(blob_id_, CreateBlobDataPtr(kBlobPayload));
+}
+
+}  // namespace
+}  // namespace blimp