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 <gtest/gtest.h>
17 #include "hpae_node_common.h"
18
19 using namespace testing::ext;
20 using namespace testing;
21
22 namespace OHOS {
23 namespace AudioStandard {
24 namespace HPAE {
25
26 std::string DEFAULT_NODE_STRING_KEY = "1_48000_2_1";
27 std::string DEFAULT_SCENE_MUSIC = "SCENE_MUSIC";
28 std::string DEFAULT_SCENE_EXTRA = "SCENE_EXTRA";
29 constexpr size_t DEFAULT_LEN = 100;
30 constexpr size_t DEFAULT_US_TIME = 100000;
31 constexpr uint64_t DEFAULT_BUFFER_SIZE = 100;
32 constexpr uint64_t DEFAULT_CONVERTER_US_TIME = 520;
33 constexpr size_t DEFAULT_FRAME_COUNT = 48;
34
35 class HpaeNodeCommonTest : public testing::Test {
36 public:
37 void SetUp() override;
38 void TearDown() override;
39 };
40
SetUp()41 void HpaeNodeCommonTest::SetUp()
42 {}
43
TearDown()44 void HpaeNodeCommonTest::TearDown()
45 {}
46
GetTestNodeInfo()47 static HpaeNodeInfo GetTestNodeInfo()
48 {
49 HpaeNodeInfo nodeInfo;
50 nodeInfo.channels = STEREO;
51 nodeInfo.samplingRate = SAMPLE_RATE_48000;
52 nodeInfo.channelLayout = CH_LAYOUT_MONO;
53 nodeInfo.format = SAMPLE_S16LE;
54 nodeInfo.frameLen = DEFAULT_LEN;
55 nodeInfo.sourceBufferType = HPAE_SOURCE_BUFFER_TYPE_MIC;
56 return nodeInfo;
57 }
58
59 HWTEST_F(HpaeNodeCommonTest, TestHpaeNodeCommon_001, TestSize.Level0)
60 {
61 HpaeNodeInfo nodeInfo1 = GetTestNodeInfo();
62 HpaeNodeInfo nodeInfo2 = GetTestNodeInfo();
63 EXPECT_EQ(CheckHpaeNodeInfoIsSame(nodeInfo1, nodeInfo2), true);
64 nodeInfo2.channels = MONO;
65 EXPECT_EQ(CheckHpaeNodeInfoIsSame(nodeInfo1, nodeInfo2), false);
66 }
67
68 HWTEST_F(HpaeNodeCommonTest, TestHpaeNodeCommon_002, TestSize.Level0)
69 {
70 EXPECT_EQ(TransStreamTypeToSceneType(STREAM_MUSIC), HPAE_SCENE_MUSIC);
71 EXPECT_EQ(TransStreamTypeToSceneType(STREAM_RECORDING), HPAE_SCENE_EFFECT_NONE);
72
73 EXPECT_EQ(TransSourceTypeToSceneType(SOURCE_TYPE_MIC), HPAE_SCENE_RECORD);
74 EXPECT_EQ(TransSourceTypeToSceneType(SOURCE_TYPE_INVALID), HPAE_SCENE_EFFECT_NONE);
75
76 HpaeNodeInfo nodeInfo = GetTestNodeInfo();
77 EXPECT_EQ(TransNodeInfoToStringKey(nodeInfo), DEFAULT_NODE_STRING_KEY);
78
79 EXPECT_EQ(TransProcessType2EnhanceScene(HPAE_SCENE_RECORD), SCENE_RECORD);
80 EXPECT_EQ(TransProcessType2EnhanceScene(HPAE_SCENE_MUSIC), SCENE_NONE);
81
82 EXPECT_EQ(TransProcessorTypeToSceneType(HPAE_SCENE_MUSIC), DEFAULT_SCENE_MUSIC);
83 EXPECT_EQ(TransProcessorTypeToSceneType(HPAE_SCENE_RECORD), DEFAULT_SCENE_EXTRA);
84 }
85
86 HWTEST_F(HpaeNodeCommonTest, TestHpaeNodeCommon_003, TestSize.Level0)
87 {
88 EXPECT_EQ(CheckSceneTypeNeedEc(HPAE_SCENE_VOIP_UP), true);
89 EXPECT_EQ(CheckSceneTypeNeedMicRef(HPAE_SCENE_VOIP_UP), true);
90 }
91
92 HWTEST_F(HpaeNodeCommonTest, TestHpaeNodeCommon_004, TestSize.Level0)
93 {
94 HpaeNodeInfo nodeInfo = GetTestNodeInfo();
95 EXPECT_EQ(ConvertDatalenToUs(DEFAULT_BUFFER_SIZE, nodeInfo), DEFAULT_CONVERTER_US_TIME);
96 EXPECT_EQ(ConvertUsToFrameCount(DEFAULT_US_TIME, nodeInfo), DEFAULT_FRAME_COUNT);
97 }
98 } // namespace HPAE
99 } // namespace AudioStandard
100 } // namespace OHOS