1 /* 2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. 3 * 4 * Use of this source code is governed by a BSD-style license 5 * that can be found in the LICENSE file in the root of the source 6 * tree. An additional intellectual property rights grant can be found 7 * in the file PATENTS. All contributing project authors may 8 * be found in the AUTHORS file in the root of the source tree. 9 */ 10 11 #ifndef WEBRTC_MODULES_RTP_RTCP_MOCKS_MOCK_RTP_RTCP_H_ 12 #define WEBRTC_MODULES_RTP_RTCP_MOCKS_MOCK_RTP_RTCP_H_ 13 14 #include <set> 15 #include <utility> 16 #include <vector> 17 18 #include "webrtc/base/optional.h" 19 #include "webrtc/modules/include/module.h" 20 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp.h" 21 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" 22 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/transport_feedback.h" 23 #include "webrtc/test/gmock.h" 24 25 namespace webrtc { 26 27 class MockRtpData : public RtpData { 28 public: 29 MOCK_METHOD3(OnReceivedPayloadData, 30 int32_t(const uint8_t* payload_data, 31 size_t payload_size, 32 const WebRtcRTPHeader* rtp_header)); 33 34 MOCK_METHOD2(OnRecoveredPacket, 35 bool(const uint8_t* packet, size_t packet_length)); 36 }; 37 38 class MockRtpRtcp : public RtpRtcp { 39 public: 40 MOCK_METHOD1(RegisterDefaultModule, int32_t(RtpRtcp* module)); 41 MOCK_METHOD0(DeRegisterDefaultModule, int32_t()); 42 MOCK_METHOD0(DefaultModuleRegistered, bool()); 43 MOCK_METHOD0(NumberChildModules, uint32_t()); 44 MOCK_METHOD1(RegisterSyncModule, int32_t(RtpRtcp* module)); 45 MOCK_METHOD0(DeRegisterSyncModule, int32_t()); 46 MOCK_METHOD2(IncomingRtcpPacket, 47 int32_t(const uint8_t* incoming_packet, size_t packet_length)); 48 MOCK_METHOD1(SetRemoteSSRC, void(uint32_t ssrc)); 49 MOCK_METHOD4(IncomingAudioNTP, 50 int32_t(uint32_t audio_received_ntp_secs, 51 uint32_t audio_received_ntp_frac, 52 uint32_t audio_rtcp_arrival_time_secs, 53 uint32_t audio_rtcp_arrival_time_frac)); 54 MOCK_METHOD0(InitSender, int32_t()); 55 MOCK_METHOD1(RegisterSendTransport, int32_t(Transport* outgoing_transport)); 56 MOCK_METHOD1(SetMaxRtpPacketSize, void(size_t size)); 57 MOCK_METHOD1(SetTransportOverhead, void(int transport_overhead_per_packet)); 58 MOCK_CONST_METHOD0(MaxPayloadSize, size_t()); 59 MOCK_CONST_METHOD0(MaxRtpPacketSize, size_t()); 60 MOCK_METHOD1(RegisterSendPayload, int32_t(const CodecInst& voice_codec)); 61 MOCK_METHOD1(RegisterSendPayload, int32_t(const VideoCodec& video_codec)); 62 MOCK_METHOD2(RegisterVideoSendPayload, 63 void(int payload_type, const char* payload_name)); 64 MOCK_METHOD1(DeRegisterSendPayload, int32_t(int8_t payload_type)); 65 MOCK_METHOD2(RegisterSendRtpHeaderExtension, 66 int32_t(RTPExtensionType type, uint8_t id)); 67 MOCK_METHOD1(DeregisterSendRtpHeaderExtension, 68 int32_t(RTPExtensionType type)); 69 MOCK_CONST_METHOD0(StartTimestamp, uint32_t()); 70 MOCK_METHOD1(SetStartTimestamp, void(uint32_t timestamp)); 71 MOCK_CONST_METHOD0(SequenceNumber, uint16_t()); 72 MOCK_METHOD1(SetSequenceNumber, void(uint16_t seq)); 73 MOCK_METHOD1(SetRtpState, void(const RtpState& rtp_state)); 74 MOCK_METHOD1(SetRtxState, void(const RtpState& rtp_state)); 75 MOCK_CONST_METHOD0(GetRtpState, RtpState()); 76 MOCK_CONST_METHOD0(GetRtxState, RtpState()); 77 MOCK_CONST_METHOD0(SSRC, uint32_t()); 78 MOCK_METHOD1(SetSSRC, void(uint32_t ssrc)); 79 MOCK_METHOD1(SetRID, int32_t(const char *rid)); 80 MOCK_METHOD1(SetMID, int32_t(const char *mid)); 81 MOCK_CONST_METHOD1(CSRCs, int32_t(uint32_t csrcs[kRtpCsrcSize])); 82 MOCK_METHOD1(SetCsrcs, void(const std::vector<uint32_t>& csrcs)); 83 MOCK_METHOD1(SetCSRCStatus, int32_t(bool include)); 84 MOCK_METHOD1(SetRtxSendStatus, void(int modes)); 85 MOCK_CONST_METHOD0(RtxSendStatus, int()); 86 MOCK_METHOD1(SetRtxSsrc, void(uint32_t)); 87 MOCK_METHOD2(SetRtxSendPayloadType, void(int, int)); 88 MOCK_CONST_METHOD0(FlexfecSsrc, rtc::Optional<uint32_t>()); 89 MOCK_CONST_METHOD0(RtxSendPayloadType, std::pair<int, int>()); 90 MOCK_METHOD1(SetSendingStatus, int32_t(bool sending)); 91 MOCK_CONST_METHOD0(Sending, bool()); 92 MOCK_METHOD1(SetSendingMediaStatus, void(bool sending)); 93 MOCK_CONST_METHOD0(SendingMedia, bool()); 94 MOCK_CONST_METHOD4(BitrateSent, 95 void(uint32_t* total_rate, 96 uint32_t* video_rate, 97 uint32_t* fec_rate, 98 uint32_t* nack_rate)); 99 MOCK_METHOD1(RegisterVideoBitrateObserver, void(BitrateStatisticsObserver*)); 100 MOCK_CONST_METHOD0(GetVideoBitrateObserver, BitrateStatisticsObserver*(void)); 101 MOCK_CONST_METHOD1(EstimatedReceiveBandwidth, 102 int(uint32_t* available_bandwidth)); 103 MOCK_METHOD9(SendOutgoingData, 104 bool(FrameType frame_type, 105 int8_t payload_type, 106 uint32_t timestamp, 107 int64_t capture_time_ms, 108 const uint8_t* payload_data, 109 size_t payload_size, 110 const RTPFragmentationHeader* fragmentation, 111 const RTPVideoHeader* rtp_video_header, 112 uint32_t* frame_id_out)); 113 MOCK_METHOD5(TimeToSendPacket, 114 bool(uint32_t ssrc, 115 uint16_t sequence_number, 116 int64_t capture_time_ms, 117 bool retransmission, 118 int probe_cluster_id)); 119 MOCK_METHOD2(TimeToSendPadding, size_t(size_t bytes, int probe_cluster_id)); 120 MOCK_METHOD2(RegisterRtcpObservers, 121 void(RtcpIntraFrameObserver* intra_frame_callback, 122 RtcpBandwidthObserver* bandwidth_callback)); 123 MOCK_CONST_METHOD0(RTCP, RtcpMode()); 124 MOCK_METHOD1(SetRTCPStatus, void(RtcpMode method)); 125 MOCK_METHOD1(SetCNAME, int32_t(const char cname[RTCP_CNAME_SIZE])); 126 MOCK_CONST_METHOD2(RemoteCNAME, 127 int32_t(uint32_t remote_ssrc, 128 char cname[RTCP_CNAME_SIZE])); 129 MOCK_CONST_METHOD5(RemoteNTP, 130 int32_t(uint32_t* received_ntp_secs, 131 uint32_t* received_ntp_frac, 132 uint32_t* rtcp_arrival_time_secs, 133 uint32_t* rtcp_arrival_time_frac, 134 uint32_t* rtcp_timestamp)); 135 MOCK_METHOD2(AddMixedCNAME, 136 int32_t(uint32_t ssrc, const char cname[RTCP_CNAME_SIZE])); 137 MOCK_METHOD1(RemoveMixedCNAME, int32_t(uint32_t ssrc)); 138 MOCK_CONST_METHOD5(RTT, 139 int32_t(uint32_t remote_ssrc, 140 int64_t* rtt, 141 int64_t* avg_rtt, 142 int64_t* min_rtt, 143 int64_t* max_rtt)); 144 MOCK_CONST_METHOD5(GetReportBlockInfo, 145 int32_t(const uint32_t remote_ssrc, 146 uint32_t* ntp_high, 147 uint32_t* ntp_low, 148 uint32_t* packets_received, 149 uint64_t* octets_received)); 150 MOCK_METHOD1(SendRTCP, int32_t(RTCPPacketType packet_type)); 151 MOCK_METHOD1(SendCompoundRTCP, 152 int32_t(const std::set<RTCPPacketType>& packet_types)); 153 MOCK_METHOD1(SendRTCPReferencePictureSelection, int32_t(uint64_t picture_id)); 154 MOCK_METHOD1(SendRTCPSliceLossIndication, int32_t(uint8_t picture_id)); 155 MOCK_CONST_METHOD2(DataCountersRTP, 156 int32_t(size_t* bytes_sent, uint32_t* packets_sent)); 157 MOCK_CONST_METHOD2(GetSendStreamDataCounters, 158 void(StreamDataCounters*, StreamDataCounters*)); 159 MOCK_CONST_METHOD3(GetRtpPacketLossStats, 160 void(bool, uint32_t, struct RtpPacketLossStats*)); 161 MOCK_METHOD1(RemoteRTCPStat, int32_t(RTCPSenderInfo* sender_info)); 162 MOCK_CONST_METHOD1(RemoteRTCPStat, 163 int32_t(std::vector<RTCPReportBlock>* receive_blocks)); 164 MOCK_METHOD4(SetRTCPApplicationSpecificData, 165 int32_t(uint8_t sub_type, 166 uint32_t name, 167 const uint8_t* data, 168 uint16_t length)); 169 MOCK_METHOD1(SetRTCPVoIPMetrics, int32_t(const RTCPVoIPMetric* voip_metric)); 170 MOCK_METHOD1(SetRtcpXrRrtrStatus, void(bool enable)); 171 MOCK_CONST_METHOD0(RtcpXrRrtrStatus, bool()); 172 MOCK_CONST_METHOD0(REMB, bool()); 173 MOCK_METHOD1(SetREMBStatus, void(bool enable)); 174 MOCK_METHOD2(SetREMBData, 175 void(uint32_t bitrate, const std::vector<uint32_t>& ssrcs)); 176 MOCK_CONST_METHOD0(TMMBR, bool()); 177 MOCK_METHOD1(SetTMMBRStatus, void(bool enable)); 178 MOCK_METHOD1(OnBandwidthEstimateUpdate, void(uint16_t bandwidth_kbit)); 179 MOCK_CONST_METHOD0(SelectiveRetransmissions, int()); 180 MOCK_METHOD1(SetSelectiveRetransmissions, int(uint8_t settings)); 181 MOCK_METHOD2(SendNACK, int32_t(const uint16_t* nack_list, uint16_t size)); 182 MOCK_METHOD1(SendNack, void(const std::vector<uint16_t>& sequence_numbers)); 183 MOCK_METHOD2(SetStorePacketsStatus, 184 void(bool enable, uint16_t number_to_store)); 185 MOCK_CONST_METHOD0(StorePackets, bool()); 186 MOCK_METHOD1(RegisterRtcpStatisticsCallback, void(RtcpStatisticsCallback*)); 187 MOCK_METHOD0(GetRtcpStatisticsCallback, RtcpStatisticsCallback*()); 188 MOCK_METHOD1(SendFeedbackPacket, bool(const rtcp::TransportFeedback& packet)); 189 MOCK_METHOD1(SetAudioPacketSize, int32_t(uint16_t packet_size_samples)); 190 MOCK_METHOD3(SendTelephoneEventOutband, 191 int32_t(uint8_t key, uint16_t time_ms, uint8_t level)); 192 MOCK_METHOD1(SetSendREDPayloadType, int32_t(int8_t payload_type)); 193 MOCK_CONST_METHOD1(SendREDPayloadType, int32_t(int8_t* payload_type)); 194 MOCK_METHOD2(SetRTPAudioLevelIndicationStatus, 195 int32_t(bool enable, uint8_t id)); 196 MOCK_METHOD1(SetAudioLevel, int32_t(uint8_t level_dbov)); 197 MOCK_METHOD1(SetTargetSendBitrate, void(uint32_t bitrate_bps)); 198 MOCK_METHOD2(SetUlpfecConfig, 199 void(int red_payload_type, int fec_payload_type)); 200 MOCK_METHOD2(SetFecParameters, 201 bool(const FecProtectionParams& delta_params, 202 const FecProtectionParams& key_params)); 203 MOCK_METHOD1(SetKeyFrameRequestMethod, int32_t(KeyFrameRequestMethod method)); 204 MOCK_METHOD0(RequestKeyFrame, int32_t()); 205 MOCK_METHOD0(TimeUntilNextProcess, int64_t()); 206 MOCK_METHOD0(Process, void()); 207 MOCK_METHOD1(RegisterSendFrameCountObserver, void(FrameCountObserver*)); 208 MOCK_CONST_METHOD0(GetSendFrameCountObserver, FrameCountObserver*(void)); 209 MOCK_METHOD1(RegisterSendChannelRtpStatisticsCallback, 210 void(StreamDataCountersCallback*)); 211 MOCK_CONST_METHOD0(GetSendChannelRtpStatisticsCallback, 212 StreamDataCountersCallback*(void)); 213 MOCK_METHOD1(SetVideoBitrateAllocation, void(const BitrateAllocation&)); 214 // Members. 215 unsigned int remote_ssrc_; 216 }; 217 218 } // namespace webrtc 219 220 #endif // WEBRTC_MODULES_RTP_RTCP_MOCKS_MOCK_RTP_RTCP_H_ 221