Chromium Code Reviews| Index: blimp/net/blimp_message_dispatcher_unittest.cc |
| diff --git a/blimp/net/blimp_message_dispatcher_unittest.cc b/blimp/net/blimp_message_dispatcher_unittest.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..e00500eeb1b9d25c1f0209c62b1aeb86d3250cea |
| --- /dev/null |
| +++ b/blimp/net/blimp_message_dispatcher_unittest.cc |
| @@ -0,0 +1,48 @@ |
| +// Copyright 2015 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. |
| +// |
| +// Uniitest for data encryption functions. |
| + |
| +#include "blimp/net/blimp_message_dispatcher.h" |
| + |
| +#include "base/logging.h" |
| +#include "testing/gmock/include/gmock/gmock.h" |
| +#include "testing/gtest/include/gtest/gtest.h" |
| + |
| +using testing::Ref; |
| +using testing::Return; |
| + |
| +namespace blimp { |
| + |
| +class MockReceiver : public BlimpMessageReceiver { |
| + public: |
| + MockReceiver() {} |
| + ~MockReceiver() override {} |
| + |
| + MOCK_METHOD1(OnMessage, bool(const BlimpMessage& message)); |
| +}; |
| + |
| +TEST(BlimpMessageDispatcherTest, AllInteractions) { |
| + BlimpMessage input_msg; |
| + BlimpMessage compositor_msg; |
| + |
| + input_msg.set_type(BlimpMessage::INPUT); |
| + compositor_msg.set_type(BlimpMessage::COMPOSITOR); |
| + |
| + BlimpMessageDispatcher dispatcher; |
| + MockReceiver receiver1; |
| + MockReceiver receiver2; |
| + |
| + dispatcher.AddReceiver(BlimpMessage::INPUT, &receiver1); |
| + dispatcher.AddReceiver(BlimpMessage::COMPOSITOR, &receiver2); |
| + |
| + EXPECT_CALL(receiver1, OnMessage(Ref(input_msg))).WillOnce(Return(true)); |
| + EXPECT_TRUE(dispatcher.OnMessage(input_msg)); |
| + |
| + EXPECT_CALL(receiver2, OnMessage(Ref(compositor_msg))) |
| + .WillOnce(Return(false)); |
|
Wez
2015/10/09 23:15:34
Note that with gMock it's not safe, in general, to
Kevin M
2015/10/10 00:42:34
Done.
|
| + EXPECT_FALSE(dispatcher.OnMessage(compositor_msg)); |
| +} |
| + |
| +} // namespace blimp |