| OLD | NEW |
| (Empty) |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include <memory> | |
| 6 | |
| 7 #include "base/memory/ptr_util.h" | |
| 8 #include "blimp/common/blob_cache/id_util.h" | |
| 9 #include "blimp/common/blob_cache/test_util.h" | |
| 10 #include "blimp/net/blob_channel/helium_blob_receiver_delegate.h" | |
| 11 #include "blimp/net/blob_channel/helium_blob_sender_delegate.h" | |
| 12 #include "blimp/net/blob_channel/mock_blob_channel_receiver.h" | |
| 13 #include "blimp/net/test_common.h" | |
| 14 #include "testing/gmock/include/gmock/gmock.h" | |
| 15 #include "testing/gtest/include/gtest/gtest.h" | |
| 16 | |
| 17 namespace blimp { | |
| 18 namespace { | |
| 19 | |
| 20 using testing::_; | |
| 21 using testing::SaveArg; | |
| 22 | |
| 23 const char kBlobId[] = "foo1"; | |
| 24 const char kBlobPayload[] = "bar1"; | |
| 25 | |
| 26 class HeliumBlobChannelTest : public testing::Test { | |
| 27 public: | |
| 28 HeliumBlobChannelTest() : receiver_delegate_(new HeliumBlobReceiverDelegate) { | |
| 29 receiver_delegate_->SetReceiver(&mock_receiver_); | |
| 30 sender_delegate_.set_outgoing_message_processor( | |
| 31 base::WrapUnique(receiver_delegate_)); | |
| 32 } | |
| 33 | |
| 34 ~HeliumBlobChannelTest() override {} | |
| 35 | |
| 36 protected: | |
| 37 const std::string blob_id_ = CalculateBlobId(kBlobId); | |
| 38 | |
| 39 MockBlobChannelReceiver mock_receiver_; | |
| 40 HeliumBlobReceiverDelegate* receiver_delegate_; | |
| 41 HeliumBlobSenderDelegate sender_delegate_; | |
| 42 | |
| 43 private: | |
| 44 DISALLOW_COPY_AND_ASSIGN(HeliumBlobChannelTest); | |
| 45 }; | |
| 46 | |
| 47 // Verifies the content of BlimpMessages generated by the Sender. | |
| 48 TEST_F(HeliumBlobChannelTest, TransferBlobContents) { | |
| 49 BlimpMessage captured_msg; | |
| 50 MockBlimpMessageProcessor* mock_processor = new MockBlimpMessageProcessor; | |
| 51 EXPECT_CALL(*mock_processor, MockableProcessMessage(_, _)) | |
| 52 .WillOnce(SaveArg<0>(&captured_msg)); | |
| 53 sender_delegate_.set_outgoing_message_processor( | |
| 54 base::WrapUnique(mock_processor)); | |
| 55 | |
| 56 sender_delegate_.DeliverBlob(blob_id_, CreateBlobDataPtr(kBlobPayload)); | |
| 57 | |
| 58 EXPECT_EQ(BlobChannelMessage::TypeCase::kTransferBlob, | |
| 59 captured_msg.blob_channel().type_case()); | |
| 60 EXPECT_EQ(blob_id_, captured_msg.blob_channel().transfer_blob().blob_id()); | |
| 61 EXPECT_EQ(kBlobPayload, | |
| 62 captured_msg.blob_channel().transfer_blob().payload()); | |
| 63 } | |
| 64 | |
| 65 // Verifies that the Receiver understands messages sent by the Sender. | |
| 66 TEST_F(HeliumBlobChannelTest, TransferBlobCompatibility) { | |
| 67 EXPECT_CALL(mock_receiver_, | |
| 68 OnBlobReceived(blob_id_, BlobDataPtrEqualsString(kBlobPayload))); | |
| 69 sender_delegate_.DeliverBlob(blob_id_, CreateBlobDataPtr(kBlobPayload)); | |
| 70 } | |
| 71 | |
| 72 } // namespace | |
| 73 } // namespace blimp | |
| OLD | NEW |