OLD | NEW |
---|---|
(Empty) | |
1 // Copyright 2015 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 // Uniitest for data encryption functions. | |
Wez
2015/11/12 03:45:57
typo
Kevin M
2015/11/12 19:03:47
Done.
| |
6 | |
7 #include "blimp/net/blimp_message_demultiplexer.h" | |
8 | |
9 #include "base/logging.h" | |
10 #include "blimp/net/test_common.h" | |
11 #include "net/base/net_errors.h" | |
12 #include "net/base/test_completion_callback.h" | |
13 #include "testing/gmock/include/gmock/gmock.h" | |
14 #include "testing/gtest/include/gtest/gtest.h" | |
15 | |
16 using testing::_; | |
17 using testing::InvokeArgument; | |
18 using testing::Ref; | |
19 using testing::Return; | |
20 using testing::SaveArg; | |
21 | |
22 namespace blimp { | |
23 | |
24 class BlimpMessageDemultiplexerTest : public testing::Test { | |
25 public: | |
26 void SetUp() override { | |
27 EXPECT_CALL(receiver1_, ProcessMessage(Ref(input_msg_), _)) | |
28 .WillOnce(SaveArg<1>(&captured_cb_)); | |
29 EXPECT_CALL(receiver2_, ProcessMessage(Ref(compositor_msg_), _)) | |
30 .WillOnce(SaveArg<1>(&captured_cb_)); | |
31 | |
32 demux_.AddProcessor(BlimpMessage::INPUT, &receiver1_); | |
33 demux_.AddProcessor(BlimpMessage::COMPOSITOR, &receiver2_); | |
34 } | |
35 | |
36 protected: | |
37 BlimpMessage input_msg_; | |
38 BlimpMessage compositor_msg_; | |
39 MockBlimpMessageProcessor receiver1_; | |
40 MockBlimpMessageProcessor receiver2_; | |
41 net::CompletionCallback captured_cb_; | |
42 BlimpMessageDemultiplexer demux_; | |
43 }; | |
44 | |
45 TEST_F(BlimpMessageDemultiplexerTest, ProcessMessageOK) { | |
46 input_msg_.set_type(BlimpMessage::INPUT); | |
47 net::TestCompletionCallback cb; | |
48 demux_.ProcessMessage(input_msg_, cb.callback()); | |
49 captured_cb_.Run(net::OK); | |
50 EXPECT_EQ(net::OK, cb.WaitForResult()); | |
51 } | |
52 | |
53 TEST_F(BlimpMessageDemultiplexerTest, ProcessMessageFailed) { | |
54 compositor_msg_.set_type(BlimpMessage::COMPOSITOR); | |
55 net::TestCompletionCallback cb2; | |
56 demux_.ProcessMessage(compositor_msg_, cb2.callback()); | |
57 captured_cb_.Run(net::ERR_FAILED); | |
58 EXPECT_EQ(net::ERR_FAILED, cb2.WaitForResult()); | |
59 } | |
60 | |
61 | |
62 } // namespace blimp | |
OLD | NEW |