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 #include <chrono>
17 #include <gtest/gtest.h>
18 #include <string>
19 #include <thread>
20
21 #include "audio_errors.h"
22 #include "hpae_co_buffer_node.h"
23 #include "hpae_pcm_buffer.h"
24 #include "hpae_sink_input_node.h"
25 #include "hpae_sink_output_node.h"
26 #include "test_case_common.h"
27
28 using namespace OHOS;
29 using namespace AudioStandard;
30 using namespace HPAE;
31 using namespace testing::ext;
32 using namespace testing;
33
34 namespace OHOS {
35 namespace AudioStandard {
36 namespace HPAE {
37 namespace {
38 constexpr uint32_t TEST_FRAME_LEN = 960; // 20ms at 48kHz
39 constexpr uint32_t TEST_LATENCY_MS = 280; // 280ms latency for testing
GetTestNodeInfo()40 HpaeNodeInfo GetTestNodeInfo()
41 {
42 HpaeNodeInfo nodeInfo;
43 nodeInfo.samplingRate = SAMPLE_RATE_48000;
44 nodeInfo.format = SAMPLE_F32LE;
45 nodeInfo.channels = STEREO;
46 nodeInfo.frameLen = TEST_FRAME_LEN;
47 nodeInfo.channelLayout = CH_LAYOUT_STEREO;
48 return nodeInfo;
49 }
50
TestRendererRenderFrame(const char * data,uint64_t len)51 int32_t TestRendererRenderFrame(const char *data, uint64_t len)
52 {
53 for (int32_t i = 0; i < len / SAMPLE_F32LE; i++) {
54 EXPECT_EQ(*((float*)data + i), 0);
55 }
56 return 0;
57 }
58 }
59
60 class HpaeCoBufferNodeUnitTest : public testing::Test {
61 public:
62 void SetUp() override;
63 void TearDown() override;
64 };
65
SetUp(void)66 void HpaeCoBufferNodeUnitTest::SetUp(void)
67 {
68 }
69
TearDown(void)70 void HpaeCoBufferNodeUnitTest::TearDown(void)
71 {
72 }
73
74 /**
75 * @tc.name : Test Construct and be connected
76 * @tc.type : FUNC
77 * @tc.number: Construct_001
78 * @tc.desc : Test Construct when config in vaild.
79 */
80 HWTEST_F(HpaeCoBufferNodeUnitTest, Construct_001, TestSize.Level0)
81 {
82 std::shared_ptr<HpaeCoBufferNode> coBufferNode = std::make_shared<HpaeCoBufferNode>();
83 EXPECT_NE(coBufferNode, nullptr);
84 coBufferNode->SetOutputClusterConnected(true);
85 EXPECT_EQ(coBufferNode->IsOutputClusterConnected(), true);
86 coBufferNode->SetOutputClusterConnected(false);
87 EXPECT_EQ(coBufferNode->IsOutputClusterConnected(), false);
88 }
89
90 /**
91 * @tc.name : Test Connect
92 * @tc.type : FUNC
93 * @tc.number: Connect_001
94 * @tc.desc : Test Connect when config in vaild.
95 */
96 HWTEST_F(HpaeCoBufferNodeUnitTest, Connect_001, TestSize.Level0)
97 {
98 HpaeNodeInfo sinkInputNodeInfo = GetTestNodeInfo();
99 std::shared_ptr<HpaeSinkInputNode> sinkInputNode = std::make_shared<HpaeSinkInputNode>(sinkInputNodeInfo);
100 EXPECT_NE(sinkInputNode, nullptr);
101 std::shared_ptr<HpaeCoBufferNode> coBufferNode = std::make_shared<HpaeCoBufferNode>();
102 EXPECT_NE(coBufferNode, nullptr);
103 coBufferNode->Connect(sinkInputNode);
104 coBufferNode->Connect(sinkInputNode);
105 HpaeNodeInfo &coNodeInfo = coBufferNode->GetNodeInfo();
106 EXPECT_EQ(sinkInputNodeInfo.samplingRate, coNodeInfo.samplingRate);
107 EXPECT_EQ(sinkInputNodeInfo.format, coNodeInfo.format);
108 EXPECT_EQ(sinkInputNodeInfo.channels, coNodeInfo.channels);
109 EXPECT_EQ(sinkInputNodeInfo.frameLen, coNodeInfo.frameLen);
110 coBufferNode->DisConnect(sinkInputNode);
111 coBufferNode->DisConnect(sinkInputNode);
112 }
113
114 /**
115 * @tc.name : Test Process
116 * @tc.type : FUNC
117 * @tc.number: Process_001
118 * @tc.desc : Test Process when config in vaild.
119 */
120 HWTEST_F(HpaeCoBufferNodeUnitTest, Process_001, TestSize.Level0)
121 {
122 HpaeNodeInfo nodeInfo = GetTestNodeInfo();
123 std::shared_ptr<HpaeCoBufferNode> coBufferNode = std::make_shared<HpaeCoBufferNode>();
124 coBufferNode->SetLatency(TEST_LATENCY_MS);
125 coBufferNode->SetNodeInfo(nodeInfo);
126 std::shared_ptr<HpaeSinkOutputNode> sinkOutputNode = std::make_shared<HpaeSinkOutputNode>(nodeInfo);
127 sinkOutputNode->Connect(coBufferNode);
128 PcmBufferInfo pcmBufferInfo;
129 pcmBufferInfo.ch = STEREO;
130 pcmBufferInfo.frameLen = TEST_FRAME_LEN;
131 HpaePcmBuffer pcmBuffer(pcmBufferInfo);
132 coBufferNode->Enqueue(&pcmBuffer);
133 coBufferNode->Enqueue(&pcmBuffer);
134 std::string deviceClass = "file_io";
135 std::string deviceNetId = "LocalDevice";
136 EXPECT_EQ(sinkOutputNode->GetRenderSinkInstance(deviceClass, deviceNetId), 0);
137 sinkOutputNode->DoProcess();
138 TestRendererRenderFrame(sinkOutputNode->GetRenderFrameData(),
139 nodeInfo.frameLen * nodeInfo.channels * GetSizeFromFormat(nodeInfo.format));
140 }
141 }
142 }
143 }