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 <condition_variable>
17 #include <mutex>
18 #include "gtest/gtest.h"
19 #include "audio_zone.h"
20 #include "audio_zone_client_manager.h"
21 #include "audio_policy_server_handler.h"
22 #include "audio_zone_service.h"
23 #include "audio_interrupt_service.h"
24 #include "i_audio_zone_event_dispatcher.h"
25
26 using namespace testing::ext;
27
28 namespace OHOS {
29 namespace AudioStandard {
30 class AudioZoneDeviceUnitTest : public testing::Test {
31 public:
SetUp()32 void SetUp() override
33 {
34 std::shared_ptr<AudioPolicyServerHandler> handler = std::make_shared<AudioPolicyServerHandler>();
35 std::shared_ptr<AudioInterruptService> interruptService = std::make_shared<AudioInterruptService>();
36 AudioZoneService::GetInstance().Init(handler, interruptService);
37 AudioZoneContext context;
38 zoneId1_ = AudioZoneService::GetInstance().CreateAudioZone("TestZone1", context, 0);
39 zoneId2_ = AudioZoneService::GetInstance().CreateAudioZone("TestZone2", context, 0);
40 }
41
TearDown()42 void TearDown() override
43 {
44 zoneId1_ = 0;
45 zoneId2_ = 0;
46 AudioZoneService::GetInstance().DeInit();
47 }
48
49 int32_t zoneId1_ = 0;
50 int32_t zoneId2_ = 0;
51 };
52
CreateDevice(DeviceType type,DeviceRole role,const std::string & macAddress,const std::string & networkId)53 static std::shared_ptr<AudioDeviceDescriptor> CreateDevice(DeviceType type, DeviceRole role,
54 const std::string &macAddress, const std::string &networkId)
55 {
56 std::shared_ptr<AudioDeviceDescriptor> desc = std::make_shared<AudioDeviceDescriptor>(type, role);
57 EXPECT_NE(desc, nullptr);
58 desc->macAddress_ = macAddress;
59 desc->networkId_ = networkId;
60 return desc;
61 }
62
63 /**
64 * @tc.name : Test AudioZone.
65 * @tc.number: AudioZoneDevice_001
66 * @tc.desc : Test bind device to audio zone.
67 */
68 HWTEST_F(AudioZoneDeviceUnitTest, AudioZoneDevice_001, TestSize.Level1)
69 {
70 auto device1 = CreateDevice(DEVICE_TYPE_SPEAKER, OUTPUT_DEVICE, "", "LocalDevice");
71 auto device2 = CreateDevice(DEVICE_TYPE_MIC, INPUT_DEVICE, "", "LocalDevice");
72 std::vector<std::shared_ptr<AudioDeviceDescriptor>> devices;
73 devices.push_back(device1);
74 devices.push_back(device2);
75 EXPECT_EQ(AudioZoneService::GetInstance().BindDeviceToAudioZone(zoneId1_, devices), 0);
76
77 AudioConnectedDevice::GetInstance().AddConnectedDevice(device1);
78 AudioConnectedDevice::GetInstance().AddConnectedDevice(device2);
79 AudioZoneService::GetInstance().UpdateDeviceFromGlobalForAllZone(device1);
80 AudioZoneService::GetInstance().UpdateDeviceFromGlobalForAllZone(device2);
81 auto fechOutputDevice = AudioZoneService::GetInstance().FetchOutputDevices(zoneId1_,
82 STREAM_USAGE_MUSIC, 0, ROUTER_TYPE_DEFAULT);
83 auto fechInputDevice = AudioZoneService::GetInstance().FetchInputDevice(zoneId1_,
84 SOURCE_TYPE_MIC, 0);
85 EXPECT_EQ(fechOutputDevice.size(), 1);
86 EXPECT_EQ(fechOutputDevice[0]->IsSameDeviceDesc(*device1), true);
87 EXPECT_NE(fechInputDevice, nullptr);
88 EXPECT_EQ(fechInputDevice->IsSameDeviceDesc(*device2), true);
89 }
90 } // namespace AudioStandard
91 } // namespace OHOS