From 888c3c49b41102c62a7c9413e98990bbc2adc10a Mon Sep 17 00:00:00 2001
From: zhangke <zhangke173@huawei.com>
Date: Thu, 28 Aug 2025 22:48:19 +0800
Subject: [PATCH] =?UTF-8?q?UT=E8=93=9D=E9=BB=84=E5=B7=AE=E5=BC=82=E8=A1=A5?=
 =?UTF-8?q?=E9=BD=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: zhangke <zhangke173@huawei.com>
---
 .../audio_background_manager_unit_test.cpp    | 108 ++++++++++++++++++
 1 file changed, 108 insertions(+)

diff --git a/services/audio_policy/test/unittest/audio_background_manager_unit_test/src/audio_background_manager_unit_test.cpp b/services/audio_policy/test/unittest/audio_background_manager_unit_test/src/audio_background_manager_unit_test.cpp
index ec9170cd4e..fbee9dec53 100644
--- a/services/audio_policy/test/unittest/audio_background_manager_unit_test/src/audio_background_manager_unit_test.cpp
+++ b/services/audio_policy/test/unittest/audio_background_manager_unit_test/src/audio_background_manager_unit_test.cpp
@@ -1184,5 +1184,113 @@ HWTEST(AudioBackgroundManagerUnitTest, AudioBackgroundManager_046, TestSize.Leve
     EXPECT_EQ(ret, true);
     EXPECT_EQ(audioBackgroundManagerTest_->appStatesMap_.empty(), false);
 }
+
+/**
+ * @tc.name  : Test IsAllowedPlayback API
+ * @tc.type  : FUNC
+ * @tc.number: AudioBackgroundManager_047
+ * @tc.desc  : Test IsAllowedPlayback
+ */
+HWTEST(AudioBackgroundManagerUnitTest, AudioBackgroundManager_047, TestSize.Level1)
+{
+    AudioBackgroundManager* audioBackgroundManagerTest_ = nullptr;
+    audioBackgroundManagerTest_ = &AudioBackgroundManager::GetInstance();
+    ASSERT_TRUE(audioBackgroundManagerTest_ != nullptr);
+
+    int32_t pid = 0;
+    int32_t uid = 0;
+    AppState appState;
+    appState.isBack = true;
+    appState.hasBackTask = false;
+    appState.isBinder = true;
+    appState.hasSession = false;
+    appState.isSystem = true;
+    audioBackgroundManagerTest_->appStatesMap_.clear();
+    audioBackgroundManagerTest_->InsertIntoAppStatesMap(pid, uid, appState);
+
+    bool ret = audioBackgroundManagerTest_->IsAllowedPlayback(1, pid);
+    EXPECT_EQ(ret, true);
+}
+
+/**
+ * @tc.name  : Test IsAllowedPlayback API
+ * @tc.type  : FUNC
+ * @tc.number: AudioBackgroundManager_048
+ * @tc.desc  : Test IsAllowedPlayback
+ */
+HWTEST(AudioBackgroundManagerUnitTest, AudioBackgroundManager_048, TestSize.Level1)
+{
+    AudioBackgroundManager* audioBackgroundManagerTest_ = nullptr;
+    audioBackgroundManagerTest_ = &AudioBackgroundManager::GetInstance();
+    ASSERT_TRUE(audioBackgroundManagerTest_ != nullptr);
+
+    int32_t pid = 0;
+    int32_t uid = 0;
+    AppState appState;
+    appState.isBack = false;
+    appState.hasBackTask = false;
+    appState.isBinder = true;
+    appState.hasSession = false;
+    appState.isSystem = true;
+    audioBackgroundManagerTest_->appStatesMap_.clear();
+    audioBackgroundManagerTest_->InsertIntoAppStatesMap(pid, uid, appState);
+
+    bool ret = audioBackgroundManagerTest_->IsAllowedPlayback(1, pid);
+    EXPECT_EQ(ret, true);
+}
+
+/**
+ * @tc.name  : Test HandleFreezeStateChange API
+ * @tc.type  : FUNC
+ * @tc.number: AudioBackgroundManager_049
+ * @tc.desc  : Test HandleFreezeStateChange
+ */
+HWTEST(AudioBackgroundManagerUnitTest, AudioBackgroundManager_049, TestSize.Level1)
+{
+    AudioBackgroundManager* audioBackgroundManagerTest_ = nullptr;
+    audioBackgroundManagerTest_ = &AudioBackgroundManager::GetInstance();
+    ASSERT_TRUE(audioBackgroundManagerTest_ != nullptr);
+
+    int32_t pid = 0;
+    int32_t uid = 0;
+    AppState appState;
+    appState.hasBackTask = true;
+    appState.isSystem = true;
+    bool isFreeze = true;
+
+    audioBackgroundManagerTest_->appStatesMap_.clear();
+    audioBackgroundManagerTest_->InsertIntoAppStatesMap(pid, uid, appState);
+    EXPECT_EQ(audioBackgroundManagerTest_->appStatesMap_.empty(), false);
+
+    audioBackgroundManagerTest_->HandleFreezeStateChange(pid, isFreeze);
+    EXPECT_EQ(audioBackgroundManagerTest_->appStatesMap_[pid].hasBackTask, true);
+}
+
+/**
+ * @tc.name  : Test HandleFreezeStateChange API
+ * @tc.type  : FUNC
+ * @tc.number: AudioBackgroundManager_050
+ * @tc.desc  : Test HandleFreezeStateChange
+ */
+HWTEST(AudioBackgroundManagerUnitTest, AudioBackgroundManager_050, TestSize.Level1)
+{
+    AudioBackgroundManager* audioBackgroundManagerTest_ = nullptr;
+    audioBackgroundManagerTest_ = &AudioBackgroundManager::GetInstance();
+    ASSERT_TRUE(audioBackgroundManagerTest_ != nullptr);
+
+    int32_t pid = 0;
+    int32_t uid = 0;
+    AppState appState;
+    appState.hasBackTask = true;
+    appState.isSystem = true;
+    bool isFreeze = false;
+
+    audioBackgroundManagerTest_->appStatesMap_.clear();
+    audioBackgroundManagerTest_->InsertIntoAppStatesMap(pid, uid, appState);
+    EXPECT_EQ(audioBackgroundManagerTest_->appStatesMap_.empty(), false);
+
+    audioBackgroundManagerTest_->HandleFreezeStateChange(pid, isFreeze);
+    EXPECT_EQ(audioBackgroundManagerTest_->appStatesMap_[pid].hasBackTask, true);
+}
 } // namespace AudioStandard
 } // namespace OHOS
-- 
Gitee