1 /* 2 * Copyright (C) 2025 Huawei Device Co., Ltd. 3 * Licensed under the Apache License, Version 2.0 (the "License"); 4 * you may not use this file except in compliance with the License. 5 * You may obtain a copy of the License at 6 * 7 * http://www.apache.org/licenses/LICENSE-2.0 8 * 9 * Unless required by applicable law or agreed to in writing, software 10 * distributed under the License is distributed on an "AS IS" BASIS, 11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 * See the License for the specific language governing permissions and 13 * limitations under the License. 14 */ 15 16 #ifndef MOCK_AUDIO_CAPTURE_H 17 #define MOCK_AUDIO_CAPTURE_H 18 19 #include "gmock/gmock.h" 20 #include "audio_capturer.h" 21 22 namespace OHOS { 23 namespace Media { 24 using namespace AudioStandard; 25 26 class MockAudioCapturer : public AudioStandard::AudioCapturer { 27 public: 28 MockAudioCapturer() = default; 29 virtual ~MockAudioCapturer() = default; 30 31 MOCK_METHOD(int32_t, SetParams, (const AudioCapturerParams params), (override)); 32 MOCK_METHOD(int32_t, UpdatePlaybackCaptureConfig, (const AudioPlaybackCaptureConfig &config), (override)); 33 MOCK_METHOD(int32_t, SetCapturerCallback, (const std::shared_ptr<AudioCapturerCallback> &callback), (override)); 34 MOCK_METHOD(void, SetAudioCapturerErrorCallback, 35 (std::shared_ptr<AudioCapturerErrorCallback> errorCallback), (override)); 36 MOCK_METHOD(void, SetFastStatusChangeCallback, 37 (const std::shared_ptr<AudioCapturerFastStatusChangeCallback> &callback), (override)); 38 MOCK_METHOD(int32_t, GetParams, (AudioCapturerParams ¶ms), (const, override)); 39 MOCK_METHOD(int32_t, GetCapturerInfo, (AudioCapturerInfo &capturerInfo), (const, override)); 40 MOCK_METHOD(int32_t, GetStreamInfo, (AudioStreamInfo &streamInfo), (const, override)); 41 MOCK_METHOD(bool, Start, (), (override)); 42 MOCK_METHOD(int32_t, Read, (uint8_t &buffer, size_t userSize, bool isBlockingRead), (override)); 43 MOCK_METHOD(CapturerState, GetStatus, (), (const, override)); 44 MOCK_METHOD(bool, GetTimeStampInfo, (Timestamp ×tamp, Timestamp::Timestampbase base), (const, override)); 45 MOCK_METHOD(bool, GetAudioTime, (Timestamp ×tamp, Timestamp::Timestampbase base), (const, override)); 46 MOCK_METHOD(bool, GetFirstPkgTimeStampInfo, (int64_t &firstTs), (const, override)); 47 MOCK_METHOD(bool, Pause, (), (const, override)); 48 MOCK_METHOD(bool, Stop, (), (const, override)); 49 MOCK_METHOD(bool, Flush, (), (const, override)); 50 MOCK_METHOD(bool, Release, (), (override)); 51 MOCK_METHOD(int32_t, GetBufferSize, (size_t &bufferSize), (const, override)); 52 MOCK_METHOD(int32_t, GetAudioStreamId, (uint32_t &sessionID), (const, override)); 53 MOCK_METHOD(int32_t, GetFrameCount, (uint32_t &frameCount), (const, override)); 54 MOCK_METHOD(int32_t, SetCapturerPositionCallback, 55 (int64_t markPosition, const std::shared_ptr<CapturerPositionCallback> &callback), (override)); 56 MOCK_METHOD(void, UnsetCapturerPositionCallback, (), (override)); 57 MOCK_METHOD(int32_t, SetCapturerPeriodPositionCallback, 58 (int64_t frameNumber, const std::shared_ptr<CapturerPeriodPositionCallback> &callback), (override)); 59 MOCK_METHOD(void, UnsetCapturerPeriodPositionCallback, (), (override)); 60 MOCK_METHOD(int32_t, RegisterAudioPolicyServerDiedCb, 61 (const int32_t clientPid, const std::shared_ptr<AudioCapturerPolicyServiceDiedCallback> &callback), 62 (override)); 63 MOCK_METHOD(int32_t, SetBufferDuration, (uint64_t bufferDuration), (const, override)); 64 MOCK_METHOD(int32_t, SetCaptureMode, (AudioCaptureMode captureMode), (override)); 65 MOCK_METHOD(AudioCaptureMode, GetCaptureMode, (), (const, override)); 66 MOCK_METHOD(int32_t, SetCapturerReadCallback, 67 (const std::shared_ptr<AudioCapturerReadCallback> &callback), (override)); 68 MOCK_METHOD(int32_t, GetBufferDesc, (BufferDesc &bufDesc), (override)); 69 MOCK_METHOD(int32_t, Enqueue, (const BufferDesc &bufDesc), (override)); 70 MOCK_METHOD(int32_t, Clear, (), (const, override)); 71 MOCK_METHOD(int32_t, GetBufQueueState, (BufferQueueState &bufState), (const, override)); 72 MOCK_METHOD(void, SetValid, (bool valid), (override)); 73 MOCK_METHOD(int64_t, GetFramesRead, (), (const, override)); 74 MOCK_METHOD(int32_t, SetAudioCapturerDeviceChangeCallback, 75 (const std::shared_ptr<AudioCapturerDeviceChangeCallback> &callback), (override)); 76 MOCK_METHOD(int32_t, RemoveAudioCapturerDeviceChangeCallback, 77 (const std::shared_ptr<AudioCapturerDeviceChangeCallback> &callback), (override)); 78 MOCK_METHOD(int32_t, SetAudioCapturerInfoChangeCallback, 79 (const std::shared_ptr<AudioCapturerInfoChangeCallback> &callback), (override)); 80 MOCK_METHOD(int32_t, RemoveAudioCapturerInfoChangeCallback, 81 (const std::shared_ptr<AudioCapturerInfoChangeCallback> &callback), (override)); 82 MOCK_METHOD(int32_t, RegisterAudioCapturerEventListener, (), (override)); 83 MOCK_METHOD(int32_t, UnregisterAudioCapturerEventListener, (), (override)); 84 MOCK_METHOD(int32_t, GetCurrentInputDevices, (AudioDeviceDescriptor &deviceInfo), (const, override)); 85 MOCK_METHOD(int32_t, GetCurrentCapturerChangeInfo, (AudioCapturerChangeInfo &changeInfo), (const, override)); 86 MOCK_METHOD(std::vector<sptr<MicrophoneDescriptor>>, GetCurrentMicrophones, (), (const, override)); 87 MOCK_METHOD(int32_t, GetAudioTimestampInfo, (Timestamp ×tamp, 88 Timestamp::Timestampbase base), (const, override)); 89 MOCK_METHOD(uint32_t, GetOverflowCount, (), (const, override)); 90 MOCK_METHOD(int32_t, SetInputDevice, (DeviceType deviceType), (const, override)); 91 MOCK_METHOD(FastStatus, GetFastStatus, (), (override)); 92 MOCK_METHOD(int32_t, SetAudioSourceConcurrency, (const std::vector<SourceType> &targetSources), (override)); 93 MOCK_METHOD(int32_t, SetInterruptStrategy, (InterruptStrategy strategy), (override)); 94 MOCK_METHOD(void, SetInterruptEventCallbackType, (InterruptEventCallbackType callbackType), (override)); 95 }; 96 } // namespace Media 97 } // namespace OHOS 98 #endif // MOCK_AUDIO_CAPTURE_H