1 // Copyright 2019 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 #ifndef CAST_COMMON_CHANNEL_TESTING_MOCK_CAST_MESSAGE_HANDLER_H_
6 #define CAST_COMMON_CHANNEL_TESTING_MOCK_CAST_MESSAGE_HANDLER_H_
7 
8 #include "cast/common/channel/cast_message_handler.h"
9 #include "cast/common/channel/proto/cast_channel.pb.h"
10 #include "gmock/gmock.h"
11 
12 namespace openscreen {
13 namespace cast {
14 
15 class MockCastMessageHandler final : public CastMessageHandler {
16  public:
17   MOCK_METHOD(void,
18               OnMessage,
19               (VirtualConnectionRouter * router,
20                CastSocket* socket,
21                ::cast::channel::CastMessage message),
22               (override));
23 };
24 
25 }  // namespace cast
26 }  // namespace openscreen
27 
28 #endif  // CAST_COMMON_CHANNEL_TESTING_MOCK_CAST_MESSAGE_HANDLER_H_
29