1 /*
2 * Copyright (c) 2024 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 #include "audio_server_sink_plugin_unit_test.h"
17
18 using namespace testing;
19 using namespace testing::ext;
20 using namespace OHOS::AudioStandard;
21 using namespace OHOS::Media::Plugins;
22
23 namespace OHOS {
24 namespace Media {
25 class MockAudioRenderer : public AudioRenderer {
26 public:
27 MOCK_METHOD(void, SetAudioPrivacyType, (AudioPrivacyType), (override));
28 MOCK_METHOD(AudioPrivacyType, GetAudioPrivacyType, (), (override));
29 MOCK_METHOD(int32_t, SetParams, (const AudioRendererParams), (override));
30 MOCK_METHOD(int32_t, SetRendererCallback, (const std::shared_ptr<AudioRendererCallback> &), (override));
31 MOCK_METHOD(int32_t, GetParams, (AudioRendererParams &), (const, override));
32 MOCK_METHOD(int32_t, GetRendererInfo, (AudioRendererInfo &), (const, override));
33 MOCK_METHOD(int32_t, GetStreamInfo, (AudioStreamInfo &), (const, override));
34 MOCK_METHOD(bool, Start, (StateChangeCmdType), (override));
35 MOCK_METHOD(int32_t, Write, (uint8_t *, size_t), (override));
36 MOCK_METHOD(int32_t, Write, (uint8_t *, size_t, uint8_t *, size_t), (override));
37 MOCK_METHOD(RendererState, GetStatus, (), (const, override));
38 MOCK_METHOD(bool, GetAudioTime, (Timestamp &, Timestamp::Timestampbase), (const, override));
39 MOCK_METHOD(bool, GetAudioPosition, (Timestamp &, Timestamp::Timestampbase), (override));
40 MOCK_METHOD(int32_t, GetLatency, (uint64_t &), (const, override));
41 MOCK_METHOD(bool, Drain, (), (const, override));
42 MOCK_METHOD(bool, Flush, (), (const, override));
43 MOCK_METHOD(bool, PauseTransitent, (StateChangeCmdType), (override));
44 MOCK_METHOD(bool, Pause, (StateChangeCmdType), (override));
45 MOCK_METHOD(bool, Stop, (), (override));
46 MOCK_METHOD(bool, Release, (), (override));
47 MOCK_METHOD(int32_t, GetBufferSize, (size_t &), (const, override));
48 MOCK_METHOD(int32_t, GetAudioStreamId, (uint32_t &), (const, override));
49 MOCK_METHOD(int32_t, GetFrameCount, (uint32_t &), (const, override));
50 MOCK_METHOD(int32_t, SetAudioRendererDesc, (AudioRendererDesc), (override));
51 MOCK_METHOD(int32_t, SetStreamType, (AudioStreamType), (override));
52 MOCK_METHOD(int32_t, SetVolume, (float), (const, override));
53 MOCK_METHOD(float, GetVolume, (), (const, override));
54 MOCK_METHOD(int32_t, SetRenderRate, (AudioRendererRate), (const, override));
55 MOCK_METHOD(AudioRendererRate, GetRenderRate, (), (const, override));
56 MOCK_METHOD(int32_t, SetRendererSamplingRate, (uint32_t), (const, override));
57 MOCK_METHOD(uint32_t, GetRendererSamplingRate, (), (const, override));
58 MOCK_METHOD(
59 int32_t, SetRendererPositionCallback, (int64_t, const std::shared_ptr<RendererPositionCallback> &), (override));
60 MOCK_METHOD(void, UnsetRendererPositionCallback, (), (override));
61 MOCK_METHOD(int32_t, SetRendererPeriodPositionCallback,
62 (int64_t, const std::shared_ptr<RendererPeriodPositionCallback> &), (override));
63 MOCK_METHOD(void, UnsetRendererPeriodPositionCallback, (), (override));
64 MOCK_METHOD(int32_t, SetBufferDuration, (uint64_t), (const, override));
65 MOCK_METHOD(int32_t, SetRenderMode, (AudioRenderMode), (override));
66 MOCK_METHOD(AudioRenderMode, GetRenderMode, (), (const, override));
67 MOCK_METHOD(int32_t, SetRendererWriteCallback, (const std::shared_ptr<AudioRendererWriteCallback> &), (override));
68 MOCK_METHOD(int32_t, SetRendererFirstFrameWritingCallback,
69 (const std::shared_ptr<AudioRendererFirstFrameWritingCallback> &), (override));
70 MOCK_METHOD(int32_t, GetBufferDesc, (BufferDesc &), (override));
71 MOCK_METHOD(int32_t, Enqueue, (const BufferDesc &), (override));
72 MOCK_METHOD(int32_t, Clear, (), (const, override));
73 MOCK_METHOD(int32_t, GetBufQueueState, (BufferQueueState &), (const, override));
74 MOCK_METHOD(void, SetInterruptMode, (InterruptMode), (override));
75 MOCK_METHOD(int32_t, SetParallelPlayFlag, (bool), (override));
76 MOCK_METHOD(int32_t, SetLowPowerVolume, (float), (const, override));
77 MOCK_METHOD(float, GetLowPowerVolume, (), (const, override));
78 MOCK_METHOD(int32_t, SetOffloadAllowed, (bool), (override));
79 MOCK_METHOD(int32_t, SetOffloadMode, (int32_t, bool), (const, override));
80 MOCK_METHOD(int32_t, UnsetOffloadMode, (), (const, override));
81 MOCK_METHOD(float, GetSingleStreamVolume, (), (const, override));
82 MOCK_METHOD(float, GetMinStreamVolume, (), (const, override));
83 MOCK_METHOD(float, GetMaxStreamVolume, (), (const, override));
84 MOCK_METHOD(uint32_t, GetUnderflowCount, (), (const, override));
85 MOCK_METHOD(int32_t, GetCurrentOutputDevices, (AudioDeviceDescriptor &), (const, override));
86 MOCK_METHOD(AudioEffectMode, GetAudioEffectMode, (), (const, override));
87 MOCK_METHOD(int64_t, GetFramesWritten, (), (const, override));
88 MOCK_METHOD(int32_t, SetAudioEffectMode, (AudioEffectMode), (const, override));
89 MOCK_METHOD(void, SetAudioRendererErrorCallback, (std::shared_ptr<AudioRendererErrorCallback>), (override));
90 MOCK_METHOD(int32_t, RegisterOutputDeviceChangeWithInfoCallback,
91 (const std::shared_ptr<AudioRendererOutputDeviceChangeCallback> &), (override));
92 MOCK_METHOD(int32_t, UnregisterOutputDeviceChangeWithInfoCallback, (), (override));
93 MOCK_METHOD(int32_t, UnregisterOutputDeviceChangeWithInfoCallback,
94 (const std::shared_ptr<AudioRendererOutputDeviceChangeCallback> &), (override));
95 MOCK_METHOD(int32_t, RegisterAudioPolicyServerDiedCb,
96 (const int32_t, const std::shared_ptr<AudioRendererPolicyServiceDiedCallback> &), (override));
97 MOCK_METHOD(int32_t, UnregisterAudioPolicyServerDiedCb, (const int32_t), (override));
98 MOCK_METHOD(int32_t, SetChannelBlendMode, (ChannelBlendMode), (override));
99 MOCK_METHOD(int32_t, SetVolumeWithRamp, (float, int32_t), (override));
100 MOCK_METHOD(void, SetPreferredFrameSize, (int32_t), (override));
101 MOCK_METHOD(int32_t, SetSpeed, (float), (override));
102 MOCK_METHOD(float, GetSpeed, (), (override));
103 MOCK_METHOD(bool, IsFastRenderer, (), (override));
104 MOCK_METHOD(void, SetSilentModeAndMixWithOthers, (bool), (override));
105 MOCK_METHOD(bool, GetSilentModeAndMixWithOthers, (), (override));
106 MOCK_METHOD(void, EnableVoiceModemCommunicationStartStream, (bool), (override));
107 MOCK_METHOD(bool, IsNoStreamRenderer, (), (const, override));
108 MOCK_METHOD(int32_t, SetDefaultOutputDevice, (DeviceType), (override));
109 MOCK_METHOD(bool, Mute, (StateChangeCmdType), (const, override));
110 MOCK_METHOD(bool, Unmute, (StateChangeCmdType), (const, override));
111 MOCK_METHOD(int32_t, GetAudioTimestampInfo, (Timestamp &, Timestamp::Timestampbase), (const, override));
~MockAudioRenderer()112 ~MockAudioRenderer() override {}
113 };
114
115 class TestEventReceiver : public Pipeline::EventReceiver {
116 public:
TestEventReceiver()117 explicit TestEventReceiver()
118 {
119 std::cout << "event receiver constructor" << std::endl;
120 }
121
OnEvent(const Event & event)122 void OnEvent(const Event &event)
123 {
124 std::cout << event.srcFilter << std::endl;
125 }
126
127 private:
128 };
129
SetUpTestCase(void)130 void AudioServerSinkPluginUnitTest::SetUpTestCase(void) {}
131
TearDownTestCase(void)132 void AudioServerSinkPluginUnitTest::TearDownTestCase(void) {}
133
SetUp(void)134 void AudioServerSinkPluginUnitTest::SetUp(void)
135 {
136 plugin_ = std::make_shared<AudioServerSinkPlugin>("TestAudioServerSinkPlugin");
137 }
138
TearDown(void)139 void AudioServerSinkPluginUnitTest::TearDown(void)
140 {
141 plugin_ = nullptr;
142 }
143
144 HWTEST_F(AudioServerSinkPluginUnitTest, audio_server_sink_plugin_init_deinit, TestSize.Level1)
145 {
146 ASSERT_TRUE(plugin_ != nullptr);
147 auto initStatus = plugin_->Init();
148 ASSERT_TRUE(initStatus == Status::OK);
149
150 initStatus = plugin_->Init();
151 ASSERT_TRUE(initStatus == Status::OK);
152
153 plugin_->audioRenderer_ = nullptr;
154 initStatus = plugin_->Init();
155 ASSERT_TRUE(initStatus == Status::OK);
156
157 auto audioRenderer = std::make_unique<NiceMock<MockAudioRenderer>>();
158
159 EXPECT_CALL(*audioRenderer, GetStatus())
160 .WillOnce(Return(RendererState::RENDERER_RELEASED));
161
162 plugin_->audioRenderer_ = std::move(audioRenderer);
163
164 auto freeStatus = plugin_->Deinit();
165 ASSERT_TRUE(freeStatus == Status::OK);
166
167 freeStatus = plugin_->Deinit();
168 ASSERT_TRUE(freeStatus == Status::OK);
169
170 plugin_->audioRenderer_ = nullptr;
171 freeStatus = plugin_->Deinit();
172 ASSERT_TRUE(freeStatus == Status::OK);
173 }
174
175 HWTEST_F(AudioServerSinkPluginUnitTest, audio_server_sink_plugin_start, TestSize.Level1)
176 {
177 auto res = plugin_->Start();
178 ASSERT_EQ(res, Status::ERROR_WRONG_STATE);
179
180 auto audioRenderer = std::make_unique<NiceMock<MockAudioRenderer>>();
181
182 EXPECT_CALL(*audioRenderer, Start(_)).WillOnce(Return(true));
183 EXPECT_CALL(*audioRenderer, Stop()).WillOnce(Return(false)).WillOnce(Return(true));
184
185 plugin_->audioRenderer_ = std::move(audioRenderer);
186
187 res = plugin_->Start();
188 ASSERT_EQ(res, Status::OK);
189
190 auto stopRenderRes = plugin_->StopRender();
191 ASSERT_EQ(res, Status::OK);
192
193 stopRenderRes = plugin_->StopRender();
194 ASSERT_EQ(res, Status::OK);
195 }
196 } // namespace Media
197 } // namespace OHOS