OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include <memory> | 5 #include <memory> |
6 | 6 |
7 #include "base/location.h" | 7 #include "base/location.h" |
8 #include "base/memory/ptr_util.h" | 8 #include "base/memory/ptr_util.h" |
9 #include "base/message_loop/message_loop.h" | 9 #include "base/message_loop/message_loop.h" |
10 #include "base/run_loop.h" | 10 #include "base/run_loop.h" |
11 #include "base/single_thread_task_runner.h" | 11 #include "base/single_thread_task_runner.h" |
12 #include "base/threading/thread_task_runner_handle.h" | 12 #include "base/threading/thread_task_runner_handle.h" |
13 #include "blimp/common/blob_cache/id_util.h" | 13 #include "blimp/common/blob_cache/id_util.h" |
14 #include "blimp/common/blob_cache/in_memory_blob_cache.h" | 14 #include "blimp/common/blob_cache/in_memory_blob_cache.h" |
15 #include "blimp/common/blob_cache/test_util.h" | 15 #include "blimp/common/blob_cache/test_util.h" |
16 #include "blimp/net/blob_channel/blob_channel_receiver.h" | 16 #include "blimp/net/blob_channel/blob_channel_receiver.h" |
17 #include "blimp/net/blob_channel/blob_channel_sender.h" | 17 #include "blimp/net/blob_channel/blob_channel_sender.h" |
| 18 #include "blimp/net/blob_channel/blob_channel_sender_impl.h" |
18 #include "blimp/net/blob_channel/mock_blob_channel_receiver.h" | 19 #include "blimp/net/blob_channel/mock_blob_channel_receiver.h" |
19 #include "blimp/net/test_common.h" | 20 #include "blimp/net/test_common.h" |
20 #include "testing/gtest/include/gtest/gtest.h" | 21 #include "testing/gtest/include/gtest/gtest.h" |
21 | 22 |
22 namespace blimp { | 23 namespace blimp { |
23 namespace { | 24 namespace { |
24 | 25 |
25 using testing::_; | 26 using testing::_; |
26 using testing::SaveArg; | 27 using testing::SaveArg; |
27 | 28 |
28 const char kBlobId[] = "foo1"; | 29 const char kBlobId[] = "foo1"; |
29 const char kBlobPayload[] = "bar1"; | 30 const char kBlobPayload[] = "bar1"; |
30 | 31 |
31 // Routes sender delegate calls to a receiver delegate object. | 32 // Routes sender delegate calls to a receiver delegate object. |
32 // The caller is responsible for ensuring that the receiver delegate is deleted | 33 // The caller is responsible for ensuring that the receiver delegate is deleted |
33 // after |this| is deleted. | 34 // after |this| is deleted. |
34 class SenderDelegateProxy : public BlobChannelSender::Delegate { | 35 class SenderDelegateProxy : public BlobChannelSenderImpl::Delegate { |
35 public: | 36 public: |
36 explicit SenderDelegateProxy(BlobChannelReceiver* receiver) | 37 explicit SenderDelegateProxy(BlobChannelReceiver* receiver) |
37 : receiver_(receiver) {} | 38 : receiver_(receiver) {} |
38 ~SenderDelegateProxy() override {} | 39 ~SenderDelegateProxy() override {} |
39 | 40 |
40 private: | 41 private: |
41 // BlobChannelSender implementation. | 42 // BlobChannelSender implementation. |
42 void DeliverBlob(const BlobId& id, BlobDataPtr data) override { | 43 void DeliverBlob(const BlobId& id, BlobDataPtr data) override { |
43 base::ThreadTaskRunnerHandle::Get()->PostTask( | 44 base::ThreadTaskRunnerHandle::Get()->PostTask( |
44 FROM_HERE, base::Bind(&BlobChannelReceiver::OnBlobReceived, | 45 FROM_HERE, base::Bind(&BlobChannelReceiver::OnBlobReceived, |
(...skipping 16 matching lines...) Expand all Loading... |
61 receiver_delegate_ = receiver_delegate.get(); | 62 receiver_delegate_ = receiver_delegate.get(); |
62 | 63 |
63 EXPECT_CALL(*receiver_delegate, SetReceiver(_)) | 64 EXPECT_CALL(*receiver_delegate, SetReceiver(_)) |
64 .WillOnce(SaveArg<0>(&stored_receiver)); | 65 .WillOnce(SaveArg<0>(&stored_receiver)); |
65 | 66 |
66 receiver_ = BlobChannelReceiver::Create( | 67 receiver_ = BlobChannelReceiver::Create( |
67 base::WrapUnique(new InMemoryBlobCache), std::move(receiver_delegate)); | 68 base::WrapUnique(new InMemoryBlobCache), std::move(receiver_delegate)); |
68 | 69 |
69 EXPECT_EQ(receiver_.get(), stored_receiver); | 70 EXPECT_EQ(receiver_.get(), stored_receiver); |
70 | 71 |
71 sender_.reset(new BlobChannelSender( | 72 sender_.reset(new BlobChannelSenderImpl( |
72 base::WrapUnique(new InMemoryBlobCache), | 73 base::WrapUnique(new InMemoryBlobCache), |
73 base::WrapUnique(new SenderDelegateProxy(receiver_.get())))); | 74 base::WrapUnique(new SenderDelegateProxy(receiver_.get())))); |
74 } | 75 } |
75 | 76 |
76 ~BlobChannelIntegrationTest() override {} | 77 ~BlobChannelIntegrationTest() override {} |
77 | 78 |
78 protected: | 79 protected: |
79 MockBlobChannelReceiverDelegate* receiver_delegate_; | 80 MockBlobChannelReceiverDelegate* receiver_delegate_; |
80 std::unique_ptr<BlobChannelReceiver> receiver_; | 81 std::unique_ptr<BlobChannelReceiver> receiver_; |
81 std::unique_ptr<BlobChannelSender> sender_; | 82 std::unique_ptr<BlobChannelSender> sender_; |
(...skipping 12 matching lines...) Expand all Loading... |
94 | 95 |
95 EXPECT_EQ(nullptr, receiver_->Get(blob_id).get()); | 96 EXPECT_EQ(nullptr, receiver_->Get(blob_id).get()); |
96 sender_->DeliverBlob(blob_id); | 97 sender_->DeliverBlob(blob_id); |
97 | 98 |
98 base::RunLoop().RunUntilIdle(); | 99 base::RunLoop().RunUntilIdle(); |
99 EXPECT_EQ(kBlobPayload, receiver_->Get(blob_id)->data); | 100 EXPECT_EQ(kBlobPayload, receiver_->Get(blob_id)->data); |
100 } | 101 } |
101 | 102 |
102 } // namespace | 103 } // namespace |
103 } // namespace blimp | 104 } // namespace blimp |
OLD | NEW |