Index: webrtc/modules/audio_processing/include/mock_audio_processing.h |
diff --git a/webrtc/modules/audio_processing/include/mock_audio_processing.h b/webrtc/modules/audio_processing/include/mock_audio_processing.h |
index 998153deaa2ac7dc2070fe0ca8afb0e2351d9dd3..8acb83ba23a31d07621b4e1a61241d7903a88e43 100644 |
--- a/webrtc/modules/audio_processing/include/mock_audio_processing.h |
+++ b/webrtc/modules/audio_processing/include/mock_audio_processing.h |
@@ -14,257 +14,174 @@ |
#include <memory> |
#include "webrtc/modules/audio_processing/include/audio_processing.h" |
+#include "webrtc/test/gmock.h" |
namespace webrtc { |
+namespace test { |
+ |
class MockEchoCancellation : public EchoCancellation { |
public: |
- MOCK_METHOD1(Enable, |
- int(bool enable)); |
- MOCK_CONST_METHOD0(is_enabled, |
- bool()); |
- MOCK_METHOD1(enable_drift_compensation, |
- int(bool enable)); |
- MOCK_CONST_METHOD0(is_drift_compensation_enabled, |
- bool()); |
- MOCK_METHOD1(set_stream_drift_samples, |
- void(int drift)); |
- MOCK_CONST_METHOD0(stream_drift_samples, |
- int()); |
- MOCK_METHOD1(set_suppression_level, |
- int(SuppressionLevel level)); |
- MOCK_CONST_METHOD0(suppression_level, |
- SuppressionLevel()); |
- MOCK_CONST_METHOD0(stream_has_echo, |
- bool()); |
- MOCK_METHOD1(enable_metrics, |
- int(bool enable)); |
- MOCK_CONST_METHOD0(are_metrics_enabled, |
- bool()); |
- MOCK_METHOD1(GetMetrics, |
- int(Metrics* metrics)); |
- MOCK_METHOD1(enable_delay_logging, |
- int(bool enable)); |
- MOCK_CONST_METHOD0(is_delay_logging_enabled, |
- bool()); |
- MOCK_METHOD2(GetDelayMetrics, |
- int(int* median, int* std)); |
- MOCK_METHOD3(GetDelayMetrics, |
- int(int* median, int* std, float* fraction_poor_delays)); |
- MOCK_CONST_METHOD0(aec_core, |
- struct AecCore*()); |
+ virtual ~MockEchoCancellation() {} |
+ MOCK_METHOD1(Enable, int(bool enable)); |
+ MOCK_CONST_METHOD0(is_enabled, bool()); |
+ MOCK_METHOD1(enable_drift_compensation, int(bool enable)); |
+ MOCK_CONST_METHOD0(is_drift_compensation_enabled, bool()); |
+ MOCK_METHOD1(set_stream_drift_samples, void(int drift)); |
+ MOCK_CONST_METHOD0(stream_drift_samples, int()); |
+ MOCK_METHOD1(set_suppression_level, int(SuppressionLevel level)); |
+ MOCK_CONST_METHOD0(suppression_level, SuppressionLevel()); |
+ MOCK_CONST_METHOD0(stream_has_echo, bool()); |
+ MOCK_METHOD1(enable_metrics, int(bool enable)); |
+ MOCK_CONST_METHOD0(are_metrics_enabled, bool()); |
+ MOCK_METHOD1(GetMetrics, int(Metrics* metrics)); |
+ MOCK_METHOD1(enable_delay_logging, int(bool enable)); |
+ MOCK_CONST_METHOD0(is_delay_logging_enabled, bool()); |
+ MOCK_METHOD2(GetDelayMetrics, int(int* median, int* std)); |
+ MOCK_METHOD3(GetDelayMetrics, int(int* median, int* std, |
+ float* fraction_poor_delays)); |
+ MOCK_CONST_METHOD0(aec_core, struct AecCore*()); |
}; |
class MockEchoControlMobile : public EchoControlMobile { |
public: |
- MOCK_METHOD1(Enable, |
- int(bool enable)); |
- MOCK_CONST_METHOD0(is_enabled, |
- bool()); |
- MOCK_METHOD1(set_routing_mode, |
- int(RoutingMode mode)); |
- MOCK_CONST_METHOD0(routing_mode, |
- RoutingMode()); |
- MOCK_METHOD1(enable_comfort_noise, |
- int(bool enable)); |
- MOCK_CONST_METHOD0(is_comfort_noise_enabled, |
- bool()); |
- MOCK_METHOD2(SetEchoPath, |
- int(const void* echo_path, size_t size_bytes)); |
- MOCK_CONST_METHOD2(GetEchoPath, |
- int(void* echo_path, size_t size_bytes)); |
+ virtual ~MockEchoControlMobile() {} |
+ MOCK_METHOD1(Enable, int(bool enable)); |
+ MOCK_CONST_METHOD0(is_enabled, bool()); |
+ MOCK_METHOD1(set_routing_mode, int(RoutingMode mode)); |
+ MOCK_CONST_METHOD0(routing_mode, RoutingMode()); |
+ MOCK_METHOD1(enable_comfort_noise, int(bool enable)); |
+ MOCK_CONST_METHOD0(is_comfort_noise_enabled, bool()); |
+ MOCK_METHOD2(SetEchoPath, int(const void* echo_path, size_t size_bytes)); |
+ MOCK_CONST_METHOD2(GetEchoPath, int(void* echo_path, size_t size_bytes)); |
}; |
class MockGainControl : public GainControl { |
public: |
- MOCK_METHOD1(Enable, |
- int(bool enable)); |
- MOCK_CONST_METHOD0(is_enabled, |
- bool()); |
- MOCK_METHOD1(set_stream_analog_level, |
- int(int level)); |
- MOCK_METHOD0(stream_analog_level, |
- int()); |
- MOCK_METHOD1(set_mode, |
- int(Mode mode)); |
- MOCK_CONST_METHOD0(mode, |
- Mode()); |
- MOCK_METHOD1(set_target_level_dbfs, |
- int(int level)); |
- MOCK_CONST_METHOD0(target_level_dbfs, |
- int()); |
- MOCK_METHOD1(set_compression_gain_db, |
- int(int gain)); |
- MOCK_CONST_METHOD0(compression_gain_db, |
- int()); |
- MOCK_METHOD1(enable_limiter, |
- int(bool enable)); |
- MOCK_CONST_METHOD0(is_limiter_enabled, |
- bool()); |
- MOCK_METHOD2(set_analog_level_limits, |
- int(int minimum, int maximum)); |
- MOCK_CONST_METHOD0(analog_level_minimum, |
- int()); |
- MOCK_CONST_METHOD0(analog_level_maximum, |
- int()); |
- MOCK_CONST_METHOD0(stream_is_saturated, |
- bool()); |
+ virtual ~MockGainControl() {} |
+ MOCK_METHOD1(Enable, int(bool enable)); |
+ MOCK_CONST_METHOD0(is_enabled, bool()); |
+ MOCK_METHOD1(set_stream_analog_level, int(int level)); |
+ MOCK_METHOD0(stream_analog_level, int()); |
+ MOCK_METHOD1(set_mode, int(Mode mode)); |
+ MOCK_CONST_METHOD0(mode, Mode()); |
+ MOCK_METHOD1(set_target_level_dbfs, int(int level)); |
+ MOCK_CONST_METHOD0(target_level_dbfs, int()); |
+ MOCK_METHOD1(set_compression_gain_db, int(int gain)); |
+ MOCK_CONST_METHOD0(compression_gain_db, int()); |
+ MOCK_METHOD1(enable_limiter, int(bool enable)); |
+ MOCK_CONST_METHOD0(is_limiter_enabled, bool()); |
+ MOCK_METHOD2(set_analog_level_limits, int(int minimum, int maximum)); |
+ MOCK_CONST_METHOD0(analog_level_minimum, int()); |
+ MOCK_CONST_METHOD0(analog_level_maximum, int()); |
+ MOCK_CONST_METHOD0(stream_is_saturated, bool()); |
}; |
class MockHighPassFilter : public HighPassFilter { |
public: |
- MOCK_METHOD1(Enable, |
- int(bool enable)); |
- MOCK_CONST_METHOD0(is_enabled, |
- bool()); |
+ virtual ~MockHighPassFilter() {} |
+ MOCK_METHOD1(Enable, int(bool enable)); |
+ MOCK_CONST_METHOD0(is_enabled, bool()); |
}; |
class MockLevelEstimator : public LevelEstimator { |
public: |
- MOCK_METHOD1(Enable, |
- int(bool enable)); |
- MOCK_CONST_METHOD0(is_enabled, |
- bool()); |
- MOCK_METHOD0(RMS, |
- int()); |
+ virtual ~MockLevelEstimator() {} |
+ MOCK_METHOD1(Enable, int(bool enable)); |
+ MOCK_CONST_METHOD0(is_enabled, bool()); |
+ MOCK_METHOD0(RMS, int()); |
}; |
class MockNoiseSuppression : public NoiseSuppression { |
public: |
- MOCK_METHOD1(Enable, |
- int(bool enable)); |
- MOCK_CONST_METHOD0(is_enabled, |
- bool()); |
- MOCK_METHOD1(set_level, |
- int(Level level)); |
- MOCK_CONST_METHOD0(level, |
- Level()); |
- MOCK_CONST_METHOD0(speech_probability, |
- float()); |
+ virtual ~MockNoiseSuppression() {} |
+ MOCK_METHOD1(Enable, int(bool enable)); |
+ MOCK_CONST_METHOD0(is_enabled, bool()); |
+ MOCK_METHOD1(set_level, int(Level level)); |
+ MOCK_CONST_METHOD0(level, Level()); |
+ MOCK_CONST_METHOD0(speech_probability, float()); |
MOCK_METHOD0(NoiseEstimate, std::vector<float>()); |
}; |
class MockVoiceDetection : public VoiceDetection { |
public: |
- MOCK_METHOD1(Enable, |
- int(bool enable)); |
- MOCK_CONST_METHOD0(is_enabled, |
- bool()); |
- MOCK_CONST_METHOD0(stream_has_voice, |
- bool()); |
- MOCK_METHOD1(set_stream_has_voice, |
- int(bool has_voice)); |
- MOCK_METHOD1(set_likelihood, |
- int(Likelihood likelihood)); |
- MOCK_CONST_METHOD0(likelihood, |
- Likelihood()); |
- MOCK_METHOD1(set_frame_size_ms, |
- int(int size)); |
- MOCK_CONST_METHOD0(frame_size_ms, |
- int()); |
+ virtual ~MockVoiceDetection() {} |
+ MOCK_METHOD1(Enable, int(bool enable)); |
+ MOCK_CONST_METHOD0(is_enabled, bool()); |
+ MOCK_CONST_METHOD0(stream_has_voice, bool()); |
+ MOCK_METHOD1(set_stream_has_voice, int(bool has_voice)); |
+ MOCK_METHOD1(set_likelihood, int(Likelihood likelihood)); |
+ MOCK_CONST_METHOD0(likelihood, Likelihood()); |
+ MOCK_METHOD1(set_frame_size_ms, int(int size)); |
+ MOCK_CONST_METHOD0(frame_size_ms, int()); |
}; |
class MockAudioProcessing : public AudioProcessing { |
public: |
MockAudioProcessing() |
- : echo_cancellation_(new MockEchoCancellation), |
- echo_control_mobile_(new MockEchoControlMobile), |
- gain_control_(new MockGainControl), |
- high_pass_filter_(new MockHighPassFilter), |
- level_estimator_(new MockLevelEstimator), |
- noise_suppression_(new MockNoiseSuppression), |
- voice_detection_(new MockVoiceDetection) { |
+ : echo_cancellation_(new testing::NiceMock<MockEchoCancellation>()), |
+ echo_control_mobile_(new testing::NiceMock<MockEchoControlMobile>()), |
+ gain_control_(new testing::NiceMock<MockGainControl>()), |
+ high_pass_filter_(new testing::NiceMock<MockHighPassFilter>()), |
+ level_estimator_(new testing::NiceMock<MockLevelEstimator>()), |
+ noise_suppression_(new testing::NiceMock<MockNoiseSuppression>()), |
+ voice_detection_(new testing::NiceMock<MockVoiceDetection>()) { |
} |
- virtual ~MockAudioProcessing() { |
- } |
+ virtual ~MockAudioProcessing() {} |
- MOCK_METHOD0(Initialize, |
- int()); |
- MOCK_METHOD6(Initialize, |
- int(int capture_input_sample_rate_hz, |
- int capture_output_sample_rate_hz, |
- int render_sample_rate_hz, |
- ChannelLayout capture_input_layout, |
- ChannelLayout capture_output_layout, |
- ChannelLayout render_input_layout)); |
- MOCK_METHOD1(Initialize, |
- int(const ProcessingConfig& processing_config)); |
+ MOCK_METHOD0(Initialize, int()); |
+ MOCK_METHOD6(Initialize, int(int capture_input_sample_rate_hz, |
+ int capture_output_sample_rate_hz, |
+ int render_sample_rate_hz, |
+ ChannelLayout capture_input_layout, |
+ ChannelLayout capture_output_layout, |
+ ChannelLayout render_input_layout)); |
+ MOCK_METHOD1(Initialize, int(const ProcessingConfig& processing_config)); |
MOCK_METHOD1(ApplyConfig, void(const Config& config)); |
MOCK_METHOD1(SetExtraOptions, void(const webrtc::Config& config)); |
- MOCK_METHOD1(set_sample_rate_hz, |
- int(int rate)); |
- MOCK_CONST_METHOD0(input_sample_rate_hz, |
- int()); |
- MOCK_CONST_METHOD0(sample_rate_hz, |
- int()); |
- MOCK_CONST_METHOD0(proc_sample_rate_hz, |
- int()); |
- MOCK_CONST_METHOD0(proc_split_sample_rate_hz, |
- int()); |
- MOCK_CONST_METHOD0(num_input_channels, |
- size_t()); |
- MOCK_CONST_METHOD0(num_output_channels, |
- size_t()); |
- MOCK_CONST_METHOD0(num_reverse_channels, |
- size_t()); |
- MOCK_METHOD1(set_output_will_be_muted, |
- void(bool muted)); |
- MOCK_CONST_METHOD0(output_will_be_muted, |
- bool()); |
- MOCK_METHOD1(ProcessStream, |
- int(AudioFrame* frame)); |
- MOCK_METHOD7(ProcessStream, |
- int(const float* const* src, |
- size_t samples_per_channel, |
- int input_sample_rate_hz, |
- ChannelLayout input_layout, |
- int output_sample_rate_hz, |
- ChannelLayout output_layout, |
- float* const* dest)); |
- MOCK_METHOD4(ProcessStream, |
- int(const float* const* src, |
- const StreamConfig& input_config, |
- const StreamConfig& output_config, |
- float* const* dest)); |
- MOCK_METHOD1(AnalyzeReverseStream, |
- int(AudioFrame* frame)); |
+ MOCK_CONST_METHOD0(proc_sample_rate_hz, int()); |
+ MOCK_CONST_METHOD0(proc_split_sample_rate_hz, int()); |
+ MOCK_CONST_METHOD0(num_input_channels, size_t()); |
+ MOCK_CONST_METHOD0(num_proc_channels, size_t()); |
+ MOCK_CONST_METHOD0(num_output_channels, size_t()); |
+ MOCK_CONST_METHOD0(num_reverse_channels, size_t()); |
+ MOCK_METHOD1(set_output_will_be_muted, void(bool muted)); |
+ MOCK_METHOD1(ProcessStream, int(AudioFrame* frame)); |
+ MOCK_METHOD7(ProcessStream, int(const float* const* src, |
+ size_t samples_per_channel, |
+ int input_sample_rate_hz, |
+ ChannelLayout input_layout, |
+ int output_sample_rate_hz, |
+ ChannelLayout output_layout, |
+ float* const* dest)); |
+ MOCK_METHOD4(ProcessStream, int(const float* const* src, |
+ const StreamConfig& input_config, |
+ const StreamConfig& output_config, |
+ float* const* dest)); |
MOCK_METHOD1(ProcessReverseStream, int(AudioFrame* frame)); |
- MOCK_METHOD4(AnalyzeReverseStream, |
- int(const float* const* data, |
- size_t samples_per_channel, |
- int sample_rate_hz, |
- ChannelLayout layout)); |
- MOCK_METHOD4(ProcessReverseStream, |
- int(const float* const* src, |
- const StreamConfig& input_config, |
- const StreamConfig& output_config, |
- float* const* dest)); |
- MOCK_METHOD1(set_stream_delay_ms, |
- int(int delay)); |
- MOCK_CONST_METHOD0(stream_delay_ms, |
- int()); |
- MOCK_CONST_METHOD0(was_stream_delay_set, |
- bool()); |
- MOCK_METHOD1(set_stream_key_pressed, |
- void(bool key_pressed)); |
- MOCK_CONST_METHOD0(stream_key_pressed, |
- bool()); |
- MOCK_METHOD1(set_delay_offset_ms, |
- void(int offset)); |
- MOCK_CONST_METHOD0(delay_offset_ms, |
- int()); |
- MOCK_METHOD2(StartDebugRecording, |
- int(const char filename[kMaxFilenameSize], |
- int64_t max_log_size_bytes)); |
- MOCK_METHOD2(StartDebugRecording, |
- int(FILE* handle, int64_t max_log_size_bytes)); |
- MOCK_METHOD1(StartDebugRecording, |
- int (FILE* handle)); |
- MOCK_METHOD1(StartDebugRecording, |
+ MOCK_METHOD4(AnalyzeReverseStream, int(const float* const* data, |
+ size_t samples_per_channel, |
+ int sample_rate_hz, |
+ ChannelLayout layout)); |
+ MOCK_METHOD4(ProcessReverseStream, int(const float* const* src, |
+ const StreamConfig& input_config, |
+ const StreamConfig& output_config, |
+ float* const* dest)); |
+ MOCK_METHOD1(set_stream_delay_ms, int(int delay)); |
+ MOCK_CONST_METHOD0(stream_delay_ms, int()); |
+ MOCK_CONST_METHOD0(was_stream_delay_set, bool()); |
+ MOCK_METHOD1(set_stream_key_pressed, void(bool key_pressed)); |
+ MOCK_METHOD1(set_delay_offset_ms, void(int offset)); |
+ MOCK_CONST_METHOD0(delay_offset_ms, int()); |
+ MOCK_METHOD2(StartDebugRecording, int(const char filename[kMaxFilenameSize], |
+ int64_t max_log_size_bytes)); |
+ MOCK_METHOD2(StartDebugRecording, int(FILE* handle, |
+ int64_t max_log_size_bytes)); |
+ MOCK_METHOD1(StartDebugRecording, int (FILE* handle)); |
+ MOCK_METHOD1(StartDebugRecordingForPlatformFile, |
int(rtc::PlatformFile handle)); |
- MOCK_METHOD0(StopDebugRecording, |
- int()); |
+ MOCK_METHOD0(StopDebugRecording, int()); |
MOCK_METHOD0(UpdateHistogramsOnCallEnd, void()); |
virtual MockEchoCancellation* echo_cancellation() const { |
return echo_cancellation_.get(); |
@@ -298,6 +215,7 @@ class MockAudioProcessing : public AudioProcessing { |
std::unique_ptr<MockVoiceDetection> voice_detection_; |
}; |
+} // namespace test |
} // namespace webrtc |
#endif // WEBRTC_MODULES_AUDIO_PROCESSING_INCLUDE_MOCK_AUDIO_PROCESSING_H_ |