3a034e15b4
Done in preparation for some threading changes that would be quite messy if implemented with the class as-is. This results in some code duplication, but is preferable to one class having two completely different modes of operation. RTP data channels are in the process of being removed anyway, so the duplicated code won't last forever. Bug: webrtc:9883 Change-Id: Idfd41a669b56a4bb4819572e4a264a4ffaaba9c0 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/178940 Commit-Queue: Taylor <deadbeef@webrtc.org> Reviewed-by: Harald Alvestrand <hta@webrtc.org> Cr-Commit-Position: refs/heads/master@{#31691}
73 lines
2.8 KiB
C++
73 lines
2.8 KiB
C++
/*
|
|
* Copyright 2016 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#ifndef PC_TEST_MOCK_DATA_CHANNEL_H_
|
|
#define PC_TEST_MOCK_DATA_CHANNEL_H_
|
|
|
|
#include <string>
|
|
|
|
#include "pc/sctp_data_channel.h"
|
|
#include "test/gmock.h"
|
|
|
|
namespace webrtc {
|
|
|
|
class MockSctpDataChannel : public rtc::RefCountedObject<SctpDataChannel> {
|
|
public:
|
|
MockSctpDataChannel(int id, DataState state)
|
|
: MockSctpDataChannel(id,
|
|
"MockSctpDataChannel",
|
|
state,
|
|
"udp",
|
|
0,
|
|
0,
|
|
0,
|
|
0) {}
|
|
MockSctpDataChannel(
|
|
int id,
|
|
const std::string& label,
|
|
DataState state,
|
|
const std::string& protocol,
|
|
uint32_t messages_sent,
|
|
uint64_t bytes_sent,
|
|
uint32_t messages_received,
|
|
uint64_t bytes_received,
|
|
const InternalDataChannelInit& config = InternalDataChannelInit(),
|
|
rtc::Thread* signaling_thread = rtc::Thread::Current(),
|
|
rtc::Thread* network_thread = rtc::Thread::Current())
|
|
: rtc::RefCountedObject<SctpDataChannel>(config,
|
|
nullptr,
|
|
label,
|
|
signaling_thread,
|
|
network_thread) {
|
|
EXPECT_CALL(*this, id()).WillRepeatedly(::testing::Return(id));
|
|
EXPECT_CALL(*this, state()).WillRepeatedly(::testing::Return(state));
|
|
EXPECT_CALL(*this, protocol()).WillRepeatedly(::testing::Return(protocol));
|
|
EXPECT_CALL(*this, messages_sent())
|
|
.WillRepeatedly(::testing::Return(messages_sent));
|
|
EXPECT_CALL(*this, bytes_sent())
|
|
.WillRepeatedly(::testing::Return(bytes_sent));
|
|
EXPECT_CALL(*this, messages_received())
|
|
.WillRepeatedly(::testing::Return(messages_received));
|
|
EXPECT_CALL(*this, bytes_received())
|
|
.WillRepeatedly(::testing::Return(bytes_received));
|
|
}
|
|
MOCK_METHOD(int, id, (), (const, override));
|
|
MOCK_METHOD(DataState, state, (), (const, override));
|
|
MOCK_METHOD(std::string, protocol, (), (const, override));
|
|
MOCK_METHOD(uint32_t, messages_sent, (), (const, override));
|
|
MOCK_METHOD(uint64_t, bytes_sent, (), (const, override));
|
|
MOCK_METHOD(uint32_t, messages_received, (), (const, override));
|
|
MOCK_METHOD(uint64_t, bytes_received, (), (const, override));
|
|
};
|
|
|
|
} // namespace webrtc
|
|
|
|
#endif // PC_TEST_MOCK_DATA_CHANNEL_H_
|