1 /*
2 * Copyright (c) 2022-2023 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 "thermal_listener_test.h"
17
18 #include "mock_thermal_mgr_client.h"
19 #include "thermal_log.h"
20 #include "thermal_mgr_client.h"
21
22 #define private public
23 #define protected public
24 #include "thermal_service.h"
25 #include "thermal_srv_config_parser.h"
26 #include "v1_1/ithermal_interface.h"
27 #include "v1_1/thermal_types.h"
28 #undef private
29 #undef protected
30
31 using namespace testing::ext;
32 using namespace OHOS::PowerMgr;
33 using namespace OHOS;
34 using namespace std;
35 using namespace OHOS::HDI::Thermal::V1_1;
36
37 namespace {
38 static ThermalLevel g_thermalLevel = ThermalLevel::COOL;
39 const std::string SYSTEM_THERMAL_SERVICE_CONFIG_PATH = "/system/etc/thermal_config/thermal_service_config.xml";
40 sptr<ThermalService> g_service = nullptr;
41 }
42
OnThermalLevelResult(const ThermalLevel & level)43 void ThermalListenerTest::ThermalLevelTestEvent::OnThermalLevelResult(const ThermalLevel& level)
44 {
45 THERMAL_HILOGD(LABEL_TEST, "OnThermalLevelResult Enter");
46 g_thermalLevel = level;
47 }
48
SetUpTestCase()49 void ThermalListenerTest::SetUpTestCase()
50 {
51 g_service = ThermalService::GetInstance();
52 g_service->InitSystemTestModules();
53 g_service->OnStart();
54 g_service->InitStateMachine();
55 }
56
TearDownTestCase()57 void ThermalListenerTest::TearDownTestCase()
58 {
59 g_service->OnStop();
60 }
61
TearDown()62 void ThermalListenerTest::TearDown()
63 {
64 g_service->SetScene("");
65 g_thermalLevel = ThermalLevel::COOL;
66 }
67
68 namespace {
69 /**
70 * @tc.name: ThermalListenerTest001
71 * @tc.desc: test class ThermalMgrListener function(thermal level event is not nullptr)
72 * @tc.type: FUNC
73 * @tc.require: issueI63SZ4
74 */
75 HWTEST_F(ThermalListenerTest, ThermalListenerTest001, TestSize.Level0)
76 {
77 THERMAL_HILOGD(LABEL_TEST, "ThermalListenerTest001 start");
78 std::shared_ptr<ThermalMgrListener> thermalListener = std::make_shared<ThermalMgrListener>();
79 ASSERT_NE(thermalListener, nullptr);
80 const std::shared_ptr<ThermalMgrListener::ThermalLevelEvent> thermalEvent =
81 std::make_shared<ThermalListenerTest::ThermalLevelTestEvent>();
82 ASSERT_NE(thermalEvent, nullptr);
83 int32_t ret = thermalListener->SubscribeLevelEvent(thermalEvent);
84 EXPECT_EQ(ret, ERR_OK);
85
86 sptr<IThermalLevelCallback> callback = new ThermalMgrListener::ThermalLevelCallback(thermalListener);
87 g_service->SubscribeThermalLevelCallback(callback);
88
89 HdfThermalCallbackInfo event;
90 ThermalZoneInfo info1;
91 info1.type = "battery";
92 info1.temp = 46100;
93 event.info.push_back(info1);
94 g_service->HandleThermalCallbackEvent(event);
95 EXPECT_TRUE(g_thermalLevel == ThermalLevel::HOT);
96
97 thermalListener->UnSubscribeLevelEvent();
98 g_service->UnSubscribeThermalLevelCallback(callback);
99
100 thermalListener->GetThermalLevel();
101 ThermalLevel level;
102 g_service->GetThermalLevel(level);
103 EXPECT_EQ(g_thermalLevel, level);
104 THERMAL_HILOGD(LABEL_TEST, "ThermalListenerTest001 end");
105 }
106
107 /**
108 * @tc.name: ThermalListenerTest002
109 * @tc.desc: test class ThermalMgrListener function(thermal level event is nullptr)
110 * @tc.type: FUNC
111 * @tc.require: issueI63SZ4
112 */
113 HWTEST_F(ThermalListenerTest, ThermalListenerTest002, TestSize.Level0)
114 {
115 THERMAL_HILOGD(LABEL_TEST, "ThermalListenerTest002 start");
116 std::shared_ptr<ThermalMgrListener> thermalListener = std::make_shared<ThermalMgrListener>();
117 ASSERT_NE(thermalListener, nullptr);
118
119 int32_t ret = thermalListener->SubscribeLevelEvent(nullptr);
120 EXPECT_EQ(true, ret == ERR_INVALID_VALUE);
121 THERMAL_HILOGD(LABEL_TEST, "ThermalListenerTest002 end");
122 }
123 } // namespace
124