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 |