| // Copyright 2019 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. |
| |
| #ifndef OSP_PUBLIC_TESTING_MESSAGE_DEMUXER_TEST_SUPPORT_H_ |
| #define OSP_PUBLIC_TESTING_MESSAGE_DEMUXER_TEST_SUPPORT_H_ |
| |
| #include "gmock/gmock.h" |
| #include "osp/public/message_demuxer.h" |
| #include "platform/api/time.h" |
| |
| namespace openscreen { |
| namespace osp { |
| |
| class MockMessageCallback final : public MessageDemuxer::MessageCallback { |
| public: |
| ~MockMessageCallback() override = default; |
| |
| MOCK_METHOD6(OnStreamMessage, |
| ErrorOr<size_t>(uint64_t endpoint_id, |
| uint64_t connection_id, |
| msgs::Type message_type, |
| const uint8_t* buffer, |
| size_t buffer_size, |
| Clock::time_point now)); |
| }; |
| |
| } // namespace osp |
| } // namespace openscreen |
| |
| #endif // OSP_PUBLIC_TESTING_MESSAGE_DEMUXER_TEST_SUPPORT_H_ |