Gitiles
Code Review Sign In
gerrit.openfyde.cn / webrtc.googlesource.com / src / b296d0591cd71baacb4ecbd1a4698a6eb2b17bca / . / webrtc / test / mock_voice_engine.h
blob: fac088b0e1707d79cc3d3845f9ccc14faaf8a829 [file] [log] [blame]
Fredrik Solenberg0ccae132015-11-03 10:15:49 +0100[diff] [blame]1/*
2 * Copyright (c) 2015 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_AUDIO_MOCK_VOICE_ENGINE_H_
12#define WEBRTC_AUDIO_MOCK_VOICE_ENGINE_H_
13
kwibergb7f89d62016-02-17 10:04:18 -0800[diff] [blame]14#include <memory>
15
Fredrik Solenberg0ccae132015-11-03 10:15:49 +0100[diff] [blame]16#include "testing/gmock/include/gmock/gmock.h"
solenberg13725082015-11-25 08:16:52 -0800[diff] [blame]17#include "webrtc/test/mock_voe_channel_proxy.h"
Fredrik Solenberg0ccae132015-11-03 10:15:49 +0100[diff] [blame]18#include "webrtc/voice_engine/voice_engine_impl.h"
19
20namespace webrtc {
21namespace test {
22
23// NOTE: This class inherits from VoiceEngineImpl so that its clients will be
24// able to get the various interfaces as usual, via T::GetInterface().
solenberg3a941542015-11-16 07:34:50 -0800[diff] [blame]25class MockVoiceEngine : public VoiceEngineImpl {
Fredrik Solenberg0ccae132015-11-03 10:15:49 +0100[diff] [blame]26 public:
27 MockVoiceEngine() : VoiceEngineImpl(new Config(), true) {
28 // Increase ref count so this object isn't automatically deleted whenever
29 // interfaces are Release():d.
30 ++_ref_count;
solenberg13725082015-11-25 08:16:52 -0800[diff] [blame]31 // We add this default behavior to make the mock easier to use in tests. It
32 // will create a NiceMock of a voe::ChannelProxy.
33 ON_CALL(*this, ChannelProxyFactory(testing::_))
34 .WillByDefault(
35 testing::Invoke([](int channel_id) {
36 return new testing::NiceMock<MockVoEChannelProxy>();
37 }));
Fredrik Solenberg0ccae132015-11-03 10:15:49 +0100[diff] [blame]38 }
39 ~MockVoiceEngine() override {
40 // Decrease ref count before base class d-tor is called; otherwise it will
41 // trigger an assertion.
42 --_ref_count;
43 }
solenberg13725082015-11-25 08:16:52 -0800[diff] [blame]44 // Allows injecting a ChannelProxy factory.
45 MOCK_METHOD1(ChannelProxyFactory, voe::ChannelProxy*(int channel_id));
46
47 // VoiceEngineImpl
kwibergb7f89d62016-02-17 10:04:18 -0800[diff] [blame]48 std::unique_ptr<voe::ChannelProxy> GetChannelProxy(int channel_id) override {
49 return std::unique_ptr<voe::ChannelProxy>(ChannelProxyFactory(channel_id));
solenberg13725082015-11-25 08:16:52 -0800[diff] [blame]50 }
Fredrik Solenberg0ccae132015-11-03 10:15:49 +0100[diff] [blame]51
52 // VoEAudioProcessing
53 MOCK_METHOD2(SetNsStatus, int(bool enable, NsModes mode));
54 MOCK_METHOD2(GetNsStatus, int(bool& enabled, NsModes& mode));
55 MOCK_METHOD2(SetAgcStatus, int(bool enable, AgcModes mode));
56 MOCK_METHOD2(GetAgcStatus, int(bool& enabled, AgcModes& mode));
57 MOCK_METHOD1(SetAgcConfig, int(AgcConfig config));
58 MOCK_METHOD1(GetAgcConfig, int(AgcConfig& config));
59 MOCK_METHOD2(SetEcStatus, int(bool enable, EcModes mode));
60 MOCK_METHOD2(GetEcStatus, int(bool& enabled, EcModes& mode));
61 MOCK_METHOD1(EnableDriftCompensation, int(bool enable));
62 MOCK_METHOD0(DriftCompensationEnabled, bool());
63 MOCK_METHOD1(SetDelayOffsetMs, void(int offset));
64 MOCK_METHOD0(DelayOffsetMs, int());
65 MOCK_METHOD2(SetAecmMode, int(AecmModes mode, bool enableCNG));
66 MOCK_METHOD2(GetAecmMode, int(AecmModes& mode, bool& enabledCNG));
67 MOCK_METHOD1(EnableHighPassFilter, int(bool enable));
68 MOCK_METHOD0(IsHighPassFilterEnabled, bool());
69 MOCK_METHOD3(SetRxNsStatus, int(int channel, bool enable, NsModes mode));
70 MOCK_METHOD3(GetRxNsStatus, int(int channel, bool& enabled, NsModes& mode));
71 MOCK_METHOD3(SetRxAgcStatus, int(int channel, bool enable, AgcModes mode));
72 MOCK_METHOD3(GetRxAgcStatus, int(int channel, bool& enabled, AgcModes& mode));
73 MOCK_METHOD2(SetRxAgcConfig, int(int channel, AgcConfig config));
74 MOCK_METHOD2(GetRxAgcConfig, int(int channel, AgcConfig& config));
75 MOCK_METHOD2(RegisterRxVadObserver,
76 int(int channel, VoERxVadCallback& observer));
77 MOCK_METHOD1(DeRegisterRxVadObserver, int(int channel));
78 MOCK_METHOD1(VoiceActivityIndicator, int(int channel));
79 MOCK_METHOD1(SetEcMetricsStatus, int(bool enable));
80 MOCK_METHOD1(GetEcMetricsStatus, int(bool& enabled));
81 MOCK_METHOD4(GetEchoMetrics, int(int& ERL, int& ERLE, int& RERL, int& A_NLP));
82 MOCK_METHOD3(GetEcDelayMetrics,
83 int(int& delay_median,
84 int& delay_std,
85 float& fraction_poor_delays));
86 MOCK_METHOD1(StartDebugRecording, int(const char* fileNameUTF8));
87 MOCK_METHOD1(StartDebugRecording, int(FILE* file_handle));
88 MOCK_METHOD0(StopDebugRecording, int());
89 MOCK_METHOD1(SetTypingDetectionStatus, int(bool enable));
90 MOCK_METHOD1(GetTypingDetectionStatus, int(bool& enabled));
91 MOCK_METHOD1(TimeSinceLastTyping, int(int& seconds));
92 MOCK_METHOD5(SetTypingDetectionParameters,
93 int(int timeWindow,
94 int costPerTyping,
95 int reportingThreshold,
96 int penaltyDecay,
97 int typeEventDelay));
98 MOCK_METHOD1(EnableStereoChannelSwapping, void(bool enable));
99 MOCK_METHOD0(IsStereoChannelSwappingEnabled, bool());
100
101 // VoEBase
102 MOCK_METHOD1(RegisterVoiceEngineObserver, int(VoiceEngineObserver& observer));
103 MOCK_METHOD0(DeRegisterVoiceEngineObserver, int());
104 MOCK_METHOD2(Init,
105 int(AudioDeviceModule* external_adm,
106 AudioProcessing* audioproc));
107 MOCK_METHOD0(audio_processing, AudioProcessing*());
108 MOCK_METHOD0(Terminate, int());
109 MOCK_METHOD0(CreateChannel, int());
110 MOCK_METHOD1(CreateChannel, int(const Config& config));
111 MOCK_METHOD1(DeleteChannel, int(int channel));
112 MOCK_METHOD1(StartReceive, int(int channel));
113 MOCK_METHOD1(StopReceive, int(int channel));
114 MOCK_METHOD1(StartPlayout, int(int channel));
115 MOCK_METHOD1(StopPlayout, int(int channel));
116 MOCK_METHOD1(StartSend, int(int channel));
117 MOCK_METHOD1(StopSend, int(int channel));
118 MOCK_METHOD1(GetVersion, int(char version[1024]));
119 MOCK_METHOD0(LastError, int());
120 MOCK_METHOD0(audio_transport, AudioTransport*());
121 MOCK_METHOD2(AssociateSendChannel,
122 int(int channel, int accociate_send_channel));
123
124 // VoECodec
125 MOCK_METHOD0(NumOfCodecs, int());
126 MOCK_METHOD2(GetCodec, int(int index, CodecInst& codec));
127 MOCK_METHOD2(SetSendCodec, int(int channel, const CodecInst& codec));
128 MOCK_METHOD2(GetSendCodec, int(int channel, CodecInst& codec));
129 MOCK_METHOD2(SetBitRate, int(int channel, int bitrate_bps));
130 MOCK_METHOD2(GetRecCodec, int(int channel, CodecInst& codec));
131 MOCK_METHOD2(SetRecPayloadType, int(int channel, const CodecInst& codec));
132 MOCK_METHOD2(GetRecPayloadType, int(int channel, CodecInst& codec));
133 MOCK_METHOD3(SetSendCNPayloadType,
134 int(int channel, int type, PayloadFrequencies frequency));
135 MOCK_METHOD2(SetFECStatus, int(int channel, bool enable));
136 MOCK_METHOD2(GetFECStatus, int(int channel, bool& enabled));
137 MOCK_METHOD4(SetVADStatus,
138 int(int channel, bool enable, VadModes mode, bool disableDTX));
139 MOCK_METHOD4(
140 GetVADStatus,
141 int(int channel, bool& enabled, VadModes& mode, bool& disabledDTX));
142 MOCK_METHOD2(SetOpusMaxPlaybackRate, int(int channel, int frequency_hz));
143 MOCK_METHOD2(SetOpusDtx, int(int channel, bool enable_dtx));
144 MOCK_METHOD0(GetEventLog, RtcEventLog*());
145
Fredrik Solenberg0ccae132015-11-03 10:15:49 +0100[diff] [blame]146 // VoEExternalMedia
147 MOCK_METHOD3(RegisterExternalMediaProcessing,
148 int(int channel,
149 ProcessingTypes type,
150 VoEMediaProcess& processObject));
151 MOCK_METHOD2(DeRegisterExternalMediaProcessing,
152 int(int channel, ProcessingTypes type));
153 MOCK_METHOD3(GetAudioFrame,
154 int(int channel, int desired_sample_rate_hz, AudioFrame* frame));
155 MOCK_METHOD2(SetExternalMixing, int(int channel, bool enable));
156
157 // VoEFile
158 MOCK_METHOD7(StartPlayingFileLocally,
159 int(int channel,
160 const char fileNameUTF8[1024],
161 bool loop,
162 FileFormats format,
163 float volumeScaling,
164 int startPointMs,
165 int stopPointMs));
166 MOCK_METHOD6(StartPlayingFileLocally,
167 int(int channel,
168 InStream* stream,
169 FileFormats format,
170 float volumeScaling,
171 int startPointMs,
172 int stopPointMs));
173 MOCK_METHOD1(StopPlayingFileLocally, int(int channel));
174 MOCK_METHOD1(IsPlayingFileLocally, int(int channel));
175 MOCK_METHOD6(StartPlayingFileAsMicrophone,
176 int(int channel,
177 const char fileNameUTF8[1024],
178 bool loop,
179 bool mixWithMicrophone,
180 FileFormats format,
181 float volumeScaling));
182 MOCK_METHOD5(StartPlayingFileAsMicrophone,
183 int(int channel,
184 InStream* stream,
185 bool mixWithMicrophone,
186 FileFormats format,
187 float volumeScaling));
188 MOCK_METHOD1(StopPlayingFileAsMicrophone, int(int channel));
189 MOCK_METHOD1(IsPlayingFileAsMicrophone, int(int channel));
190 MOCK_METHOD4(StartRecordingPlayout,
191 int(int channel,
192 const char* fileNameUTF8,
193 CodecInst* compression,
194 int maxSizeBytes));
195 MOCK_METHOD1(StopRecordingPlayout, int(int channel));
196 MOCK_METHOD3(StartRecordingPlayout,
197 int(int channel, OutStream* stream, CodecInst* compression));
198 MOCK_METHOD3(StartRecordingMicrophone,
199 int(const char* fileNameUTF8,
200 CodecInst* compression,
201 int maxSizeBytes));
202 MOCK_METHOD2(StartRecordingMicrophone,
203 int(OutStream* stream, CodecInst* compression));
204 MOCK_METHOD0(StopRecordingMicrophone, int());
205
206 // VoEHardware
207 MOCK_METHOD1(GetNumOfRecordingDevices, int(int& devices));
208 MOCK_METHOD1(GetNumOfPlayoutDevices, int(int& devices));
209 MOCK_METHOD3(GetRecordingDeviceName,
210 int(int index, char strNameUTF8[128], char strGuidUTF8[128]));
211 MOCK_METHOD3(GetPlayoutDeviceName,
212 int(int index, char strNameUTF8[128], char strGuidUTF8[128]));
213 MOCK_METHOD2(SetRecordingDevice,
214 int(int index, StereoChannel recordingChannel));
215 MOCK_METHOD1(SetPlayoutDevice, int(int index));
216 MOCK_METHOD1(SetAudioDeviceLayer, int(AudioLayers audioLayer));
217 MOCK_METHOD1(GetAudioDeviceLayer, int(AudioLayers& audioLayer));
218 MOCK_METHOD1(SetRecordingSampleRate, int(unsigned int samples_per_sec));
219 MOCK_CONST_METHOD1(RecordingSampleRate, int(unsigned int* samples_per_sec));
220 MOCK_METHOD1(SetPlayoutSampleRate, int(unsigned int samples_per_sec));
221 MOCK_CONST_METHOD1(PlayoutSampleRate, int(unsigned int* samples_per_sec));
222 MOCK_CONST_METHOD0(BuiltInAECIsAvailable, bool());
223 MOCK_METHOD1(EnableBuiltInAEC, int(bool enable));
224 MOCK_CONST_METHOD0(BuiltInAGCIsAvailable, bool());
225 MOCK_METHOD1(EnableBuiltInAGC, int(bool enable));
226 MOCK_CONST_METHOD0(BuiltInNSIsAvailable, bool());
227 MOCK_METHOD1(EnableBuiltInNS, int(bool enable));
228
229 // VoENetEqStats
230 MOCK_METHOD2(GetNetworkStatistics,
231 int(int channel, NetworkStatistics& stats));
232 MOCK_CONST_METHOD2(GetDecodingCallStatistics,
233 int(int channel, AudioDecodingCallStats* stats));
234
235 // VoENetwork
236 MOCK_METHOD2(RegisterExternalTransport,
237 int(int channel, Transport& transport));
238 MOCK_METHOD1(DeRegisterExternalTransport, int(int channel));
239 MOCK_METHOD3(ReceivedRTPPacket,
240 int(int channel, const void* data, size_t length));
241 MOCK_METHOD4(ReceivedRTPPacket,
242 int(int channel,
243 const void* data,
244 size_t length,
245 const PacketTime& packet_time));
246 MOCK_METHOD3(ReceivedRTCPPacket,
247 int(int channel, const void* data, size_t length));
248
249 // VoERTP_RTCP
250 MOCK_METHOD2(SetLocalSSRC, int(int channel, unsigned int ssrc));
251 MOCK_METHOD2(GetLocalSSRC, int(int channel, unsigned int& ssrc));
252 MOCK_METHOD2(GetRemoteSSRC, int(int channel, unsigned int& ssrc));
253 MOCK_METHOD3(SetSendAudioLevelIndicationStatus,
254 int(int channel, bool enable, unsigned char id));
255 MOCK_METHOD3(SetReceiveAudioLevelIndicationStatus,
256 int(int channel, bool enable, unsigned char id));
257 MOCK_METHOD3(SetSendAbsoluteSenderTimeStatus,
258 int(int channel, bool enable, unsigned char id));
259 MOCK_METHOD3(SetReceiveAbsoluteSenderTimeStatus,
260 int(int channel, bool enable, unsigned char id));
261 MOCK_METHOD2(SetRTCPStatus, int(int channel, bool enable));
262 MOCK_METHOD2(GetRTCPStatus, int(int channel, bool& enabled));
263 MOCK_METHOD2(SetRTCP_CNAME, int(int channel, const char cName[256]));
264 MOCK_METHOD2(GetRTCP_CNAME, int(int channel, char cName[256]));
265 MOCK_METHOD2(GetRemoteRTCP_CNAME, int(int channel, char cName[256]));
266 MOCK_METHOD7(GetRemoteRTCPData,
267 int(int channel,
268 unsigned int& NTPHigh,
269 unsigned int& NTPLow,
270 unsigned int& timestamp,
271 unsigned int& playoutTimestamp,
272 unsigned int* jitter,
273 unsigned short* fractionLost));
274 MOCK_METHOD4(GetRTPStatistics,
275 int(int channel,
276 unsigned int& averageJitterMs,
277 unsigned int& maxJitterMs,
278 unsigned int& discardedPackets));
279 MOCK_METHOD2(GetRTCPStatistics, int(int channel, CallStatistics& stats));
280 MOCK_METHOD2(GetRemoteRTCPReportBlocks,
281 int(int channel, std::vector<ReportBlock>* receive_blocks));
282 MOCK_METHOD3(SetREDStatus, int(int channel, bool enable, int redPayloadtype));
283 MOCK_METHOD3(GetREDStatus,
284 int(int channel, bool& enable, int& redPayloadtype));
285 MOCK_METHOD3(SetNACKStatus, int(int channel, bool enable, int maxNoPackets));
286
287 // VoEVideoSync
288 MOCK_METHOD1(GetPlayoutBufferSize, int(int& buffer_ms));
289 MOCK_METHOD2(SetMinimumPlayoutDelay, int(int channel, int delay_ms));
290 MOCK_METHOD3(GetDelayEstimate,
291 int(int channel,
292 int* jitter_buffer_delay_ms,
293 int* playout_buffer_delay_ms));
294 MOCK_CONST_METHOD1(GetLeastRequiredDelayMs, int(int channel));
295 MOCK_METHOD2(SetInitTimestamp, int(int channel, unsigned int timestamp));
296 MOCK_METHOD2(SetInitSequenceNumber, int(int channel, short sequenceNumber));
297 MOCK_METHOD2(GetPlayoutTimestamp, int(int channel, unsigned int& timestamp));
298 MOCK_METHOD3(GetRtpRtcp,
299 int(int channel,
300 RtpRtcp** rtpRtcpModule,
301 RtpReceiver** rtp_receiver));
302
303 // VoEVolumeControl
304 MOCK_METHOD1(SetSpeakerVolume, int(unsigned int volume));
305 MOCK_METHOD1(GetSpeakerVolume, int(unsigned int& volume));
306 MOCK_METHOD1(SetMicVolume, int(unsigned int volume));
307 MOCK_METHOD1(GetMicVolume, int(unsigned int& volume));
308 MOCK_METHOD2(SetInputMute, int(int channel, bool enable));
309 MOCK_METHOD2(GetInputMute, int(int channel, bool& enabled));
310 MOCK_METHOD1(GetSpeechInputLevel, int(unsigned int& level));
311 MOCK_METHOD2(GetSpeechOutputLevel, int(int channel, unsigned int& level));
312 MOCK_METHOD1(GetSpeechInputLevelFullRange, int(unsigned int& level));
313 MOCK_METHOD2(GetSpeechOutputLevelFullRange,
314 int(int channel, unsigned& level));
315 MOCK_METHOD2(SetChannelOutputVolumeScaling, int(int channel, float scaling));
316 MOCK_METHOD2(GetChannelOutputVolumeScaling, int(int channel, float& scaling));
317 MOCK_METHOD3(SetOutputVolumePan, int(int channel, float left, float right));
318 MOCK_METHOD3(GetOutputVolumePan, int(int channel, float& left, float& right));
319};
320} // namespace test
321} // namespace webrtc
322
323#endif // WEBRTC_AUDIO_MOCK_VOICE_ENGINE_H_