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 #include <gtest/gtest.h>
16 #include <cmath>
17 #include <memory>
18 #include <cstdio>
19 #include "hpae_sink_input_node.h"
20 #include "hpae_render_effect_node.h"
21 #include "hpae_sink_output_node.h"
22 #include "hpae_source_input_node.h"
23 #include <fstream>
24 #include <streambuf>
25 #include <string>
26 #include "test_case_common.h"
27 #include "audio_errors.h"
28 #include "audio_effect_chain_manager.h"
29
30 using namespace OHOS;
31 using namespace AudioStandard;
32 using namespace HPAE;
33 using namespace testing::ext;
34 using namespace testing;
35
36 namespace OHOS {
37 namespace AudioStandard {
38 namespace HPAE {
39 static constexpr uint32_t TEST_ID = 1266;
40 static constexpr uint32_t TEST_FRAMELEN1 = 960;
41 static constexpr uint32_t NODEINFO_EFFECTSCENEVALID = 100;
42 std::vector<EffectChain> DEFAULT_EFFECT_CHAINS = {
43 {"EFFECTCHAIN_SPK_MUSIC", {"apply1", "apply2", "apply3"}, ""},
44 {"EFFECTCHAIN_BT_MUSIC", {}, ""}
45 };
46
47 EffectChainManagerParam DEFAULT_EFFECT_CHAIN_MANAGER_PARAM{
48 3,
49 "SCENE_DEFAULT",
50 {},
51 {{"SCENE_MOVIE_&_EFFECT_DEFAULT_&_DEVICE_TYPE_SPEAKER", "EFFECTCHAIN_SPK_MUSIC"},
52 {"SCENE_MOVIE_&_EFFECT_DEFAULT_&_DEVICE_TYPE_BLUETOOTH_A2DP", "EFFECTCHAIN_BT_MUSIC"}},
53 {{"effect1", "property1"}, {"effect4", "property5"}, {"effect1", "property4"}}
54 };
55
56 std::vector<std::shared_ptr<AudioEffectLibEntry>> DEFAULT_EFFECT_LIBRARY_LIST = {};
57 class HpaeRenderEffectNodeTest : public testing::Test {
58 public:
59 void SetUp();
60 void TearDown();
61 };
62
SetUp()63 void HpaeRenderEffectNodeTest::SetUp()
64 {}
65
TearDown()66 void HpaeRenderEffectNodeTest::TearDown()
67 {}
68
69 HWTEST_F(HpaeRenderEffectNodeTest, testCreate_001, TestSize.Level0)
70 {
71 HpaeNodeInfo nodeInfo;
72 nodeInfo.nodeId = TEST_ID;
73 nodeInfo.frameLen = TEST_FRAMELEN1;
74 nodeInfo.samplingRate = SAMPLE_RATE_48000;
75 nodeInfo.channels = STEREO;
76 nodeInfo.format = SAMPLE_F32LE;
77 std::shared_ptr<HpaeRenderEffectNode> hpaeRenderEffectNode = std::make_shared<HpaeRenderEffectNode>(nodeInfo);
78 nodeInfo.effectInfo.effectScene = (AudioEffectScene)0xff;
79 EXPECT_EQ(hpaeRenderEffectNode->AudioRendererCreate(nodeInfo), 0);
80 EXPECT_NE(hpaeRenderEffectNode->ReleaseAudioEffectChain(nodeInfo), 0);
81 }
82
83 HWTEST_F(HpaeRenderEffectNodeTest, testCreate_002, TestSize.Level0)
84 {
85 constexpr uint32_t idOffset = 5;
86 HpaeNodeInfo nodeInfo;
87 nodeInfo.nodeId = TEST_ID;
88 nodeInfo.frameLen = TEST_FRAMELEN1;
89 nodeInfo.samplingRate = SAMPLE_RATE_48000;
90 nodeInfo.channels = STEREO;
91 nodeInfo.format = SAMPLE_F32LE;
92 std::shared_ptr<HpaeRenderEffectNode> hpaeRenderEffectNode = std::make_shared<HpaeRenderEffectNode>(nodeInfo);
93 EXPECT_EQ(hpaeRenderEffectNode->AudioRendererCreate(nodeInfo), 0);
94 HpaeNodeInfo nodeInfo2 = nodeInfo;
95 nodeInfo2.nodeId += idOffset;
96 EXPECT_NE(hpaeRenderEffectNode->ReleaseAudioEffectChain(nodeInfo2), 0);
97 EXPECT_EQ(hpaeRenderEffectNode->ReleaseAudioEffectChain(nodeInfo), 0);
98 }
99
100 HWTEST_F(HpaeRenderEffectNodeTest, testSignalProcess_001, TestSize.Level0)
101 {
102 HpaeNodeInfo nodeInfo;
103 nodeInfo.nodeId = TEST_ID;
104 nodeInfo.frameLen = TEST_FRAMELEN1;
105 nodeInfo.samplingRate = SAMPLE_RATE_48000;
106 nodeInfo.channels = STEREO;
107 nodeInfo.format = SAMPLE_F32LE;
108 HpaeNodeInfo dstNodeInfo;
109 std::shared_ptr<HpaeRenderEffectNode> hpaeRenderEffectNode = std::make_shared<HpaeRenderEffectNode>(nodeInfo);
110
111 std::vector<HpaePcmBuffer *> inputs;
112 EXPECT_EQ(hpaeRenderEffectNode->SignalProcess(inputs), nullptr);
113 PcmBufferInfo pcmBufferInfo(MONO, TEST_FRAMELEN1, SAMPLE_RATE_44100);
114 HpaePcmBuffer hpaePcmBuffer(pcmBufferInfo);
115 inputs.emplace_back(&hpaePcmBuffer);
116 EXPECT_NE(hpaeRenderEffectNode->SignalProcess(inputs), nullptr);
117 }
118
119 HWTEST_F(HpaeRenderEffectNodeTest, testSignalProcess_002, TestSize.Level0)
120 {
121 HpaeNodeInfo nodeInfo;
122 nodeInfo.nodeId = TEST_ID;
123 nodeInfo.frameLen = TEST_FRAMELEN1;
124 nodeInfo.samplingRate = SAMPLE_RATE_48000;
125 nodeInfo.channels = STEREO;
126 nodeInfo.format = SAMPLE_F32LE;
127 std::shared_ptr<HpaeRenderEffectNode> hpaeRenderEffectNode = std::make_shared<HpaeRenderEffectNode>(nodeInfo);
128
129 std::vector<HpaePcmBuffer *> inputs;
130 PcmBufferInfo pcmBufferInfo(MONO, TEST_FRAMELEN1, SAMPLE_RATE_44100);
131 HpaePcmBuffer hpaePcmBuffer(pcmBufferInfo);
132 hpaePcmBuffer.SetBufferSilence(true);
133 inputs.emplace_back(&hpaePcmBuffer);
134 EXPECT_NE(hpaeRenderEffectNode->SignalProcess(inputs), nullptr);
135 hpaeRenderEffectNode->ReconfigOutputBuffer();
136 }
137
138 HWTEST_F(HpaeRenderEffectNodeTest, testSignalProcess_003, TestSize.Level0)
139 {
140 AudioEffectChainManager::GetInstance()->InitAudioEffectChainManager(DEFAULT_EFFECT_CHAINS,
141 DEFAULT_EFFECT_CHAIN_MANAGER_PARAM, DEFAULT_EFFECT_LIBRARY_LIST);
142 std::string sceneStr = "SCENE_MUSIC";
143 AudioEffectChainManager::GetInstance()->CreateAudioEffectChainDynamic(sceneStr);
144 HpaeNodeInfo nodeInfo;
145 nodeInfo.nodeId = TEST_ID;
146 nodeInfo.frameLen = TEST_FRAMELEN1;
147 nodeInfo.samplingRate = SAMPLE_RATE_16000;
148 nodeInfo.channels = CHANNEL_6;
149 nodeInfo.format = SAMPLE_F32LE;
150 nodeInfo.effectInfo.effectScene = SCENE_MUSIC;
151 std::shared_ptr<HpaeRenderEffectNode> hpaeRenderEffectNode = std::make_shared<HpaeRenderEffectNode>(nodeInfo);
152
153 std::vector<HpaePcmBuffer *> inputs;
154 PcmBufferInfo pcmBufferInfo(MONO, TEST_FRAMELEN1, SAMPLE_RATE_44100);
155 HpaePcmBuffer hpaePcmBuffer(pcmBufferInfo);
156 hpaePcmBuffer.SetBufferSilence(true);
157 inputs.emplace_back(&hpaePcmBuffer);
158 EXPECT_NE(hpaeRenderEffectNode->SignalProcess(inputs), nullptr);
159 hpaeRenderEffectNode->ReconfigOutputBuffer();
160 AudioEffectChainManager::GetInstance()->ResetInfo();
161 }
162
163 HWTEST_F(HpaeRenderEffectNodeTest, testModifyAudioEffectChainInfo_001, TestSize.Level0)
164 {
165 HpaeNodeInfo nodeInfo;
166 nodeInfo.nodeId = TEST_ID;
167 nodeInfo.frameLen = TEST_FRAMELEN1;
168 nodeInfo.samplingRate = SAMPLE_RATE_48000;
169 nodeInfo.channels = STEREO;
170 nodeInfo.format = SAMPLE_F32LE;
171 std::shared_ptr<HpaeRenderEffectNode> hpaeRenderEffectNode = std::make_shared<HpaeRenderEffectNode>(nodeInfo);
172 EXPECT_NE(hpaeRenderEffectNode, nullptr);
173 ModifyAudioEffectChainInfoReason testReason = static_cast<ModifyAudioEffectChainInfoReason>(2);
174 hpaeRenderEffectNode->ModifyAudioEffectChainInfo(nodeInfo, testReason);
175 nodeInfo.effectInfo.effectScene = (AudioEffectScene)0xff;
176 hpaeRenderEffectNode->ModifyAudioEffectChainInfo(nodeInfo, testReason);
177 }
178
179 HWTEST_F(HpaeRenderEffectNodeTest, testUpdateAudioEffectChainInfo_001, TestSize.Level0)
180 {
181 HpaeNodeInfo nodeInfo;
182 nodeInfo.nodeId = TEST_ID;
183 nodeInfo.frameLen = TEST_FRAMELEN1;
184 nodeInfo.samplingRate = SAMPLE_RATE_48000;
185 nodeInfo.channels = STEREO;
186 nodeInfo.format = SAMPLE_F32LE;
187 std::shared_ptr<HpaeRenderEffectNode> hpaeRenderEffectNode = std::make_shared<HpaeRenderEffectNode>(nodeInfo);
188 EXPECT_NE(hpaeRenderEffectNode, nullptr);
189 hpaeRenderEffectNode->UpdateAudioEffectChainInfo(nodeInfo);
190 nodeInfo.effectInfo.effectScene = (AudioEffectScene)0xff;
191 hpaeRenderEffectNode->UpdateAudioEffectChainInfo(nodeInfo);
192 }
193
194 HWTEST_F(HpaeRenderEffectNodeTest, testHpaeRenderEffectNode_001, TestSize.Level0)
195 {
196 HpaeNodeInfo nodeInfo;
197 nodeInfo.nodeId = TEST_ID;
198 nodeInfo.frameLen = TEST_FRAMELEN1;
199 nodeInfo.samplingRate = SAMPLE_RATE_48000;
200 nodeInfo.channels = STEREO;
201 nodeInfo.format = SAMPLE_F32LE;
202 //1, default nodeInfo.sceneType
203 nodeInfo.sceneType = HPAE_SCENE_DEFAULT;
204 std::shared_ptr<HpaeRenderEffectNode> hpaeRenderEffectNode_0 = std::make_shared<HpaeRenderEffectNode>(nodeInfo);
205 EXPECT_NE(hpaeRenderEffectNode_0, nullptr);
206 }
207
208 HWTEST_F(HpaeRenderEffectNodeTest, testHpaeRenderEffectNode_002, TestSize.Level0)
209 {
210 HpaeNodeInfo nodeInfo;
211 nodeInfo.nodeId = TEST_ID;
212 nodeInfo.frameLen = TEST_FRAMELEN1;
213 nodeInfo.samplingRate = SAMPLE_RATE_48000;
214 nodeInfo.channels = STEREO;
215 nodeInfo.format = SAMPLE_F32LE;
216 //2, non default nodeInfo.sceneType
217 nodeInfo.sceneType = HPAE_SCENE_MUSIC;
218 nodeInfo.effectInfo.effectScene = SCENE_COLLABORATIVE;
219 std::shared_ptr<HpaeRenderEffectNode> hpaeRenderEffectNode_1 = std::make_shared<HpaeRenderEffectNode>(nodeInfo);
220 EXPECT_NE(hpaeRenderEffectNode_1, nullptr);
221 }
222
223 HWTEST_F(HpaeRenderEffectNodeTest, testHpaeRenderEffectNode_003, TestSize.Level0)
224 {
225 HpaeNodeInfo nodeInfo;
226 nodeInfo.nodeId = TEST_ID;
227 nodeInfo.frameLen = TEST_FRAMELEN1;
228 nodeInfo.samplingRate = SAMPLE_RATE_48000;
229 nodeInfo.channels = STEREO;
230 nodeInfo.format = SAMPLE_F32LE;
231 //3, else branch 00
232 nodeInfo.sceneType = HPAE_SCENE_MUSIC;
233 nodeInfo.effectInfo.effectScene = SCENE_SPEECH;
234 std::shared_ptr<HpaeRenderEffectNode> hpaeRenderEffectNode_2 = std::make_shared<HpaeRenderEffectNode>(nodeInfo);
235 EXPECT_NE(hpaeRenderEffectNode_2, nullptr);
236 }
237
238 HWTEST_F(HpaeRenderEffectNodeTest, testHpaeRenderEffectNode_004, TestSize.Level0)
239 {
240 HpaeNodeInfo nodeInfo;
241 nodeInfo.nodeId = TEST_ID;
242 nodeInfo.frameLen = TEST_FRAMELEN1;
243 nodeInfo.samplingRate = SAMPLE_RATE_48000;
244 nodeInfo.channels = STEREO;
245 nodeInfo.format = SAMPLE_F32LE;
246 //4, else branch 01 NODEINFO_EFFECTSCENEVALID
247 nodeInfo.sceneType = HPAE_SCENE_MUSIC;
248 nodeInfo.effectInfo.effectScene = static_cast<AudioEffectScene>(NODEINFO_EFFECTSCENEVALID);
249 std::shared_ptr<HpaeRenderEffectNode> hpaeRenderEffectNode_3 = std::make_shared<HpaeRenderEffectNode>(nodeInfo);
250 EXPECT_NE(hpaeRenderEffectNode_3, nullptr);
251 }
252 } // namespace HPAE
253 } // namespace AudioStandard
254 } // namespace OHOS