| Index: blimp/net/blob_channel/helium_blob_channel_unittest.cc
|
| diff --git a/blimp/net/blob_channel/helium_blob_channel_unittest.cc b/blimp/net/blob_channel/helium_blob_channel_unittest.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..38a81e8387c6fcedf337cf83094fa41e63c77f7c
|
| --- /dev/null
|
| +++ b/blimp/net/blob_channel/helium_blob_channel_unittest.cc
|
| @@ -0,0 +1,73 @@
|
| +// 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
|
|
|