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 <future>
17 #include <string>
18
19 #include <unistd.h>
20
21 #include <gtest/gtest.h>
22
23 #include "b_error/b_error.h"
24 #include "b_radar/b_radar.h"
25 #include "module_ipc/service.h"
26 #include "module_ipc/svc_session_manager.h"
27 #include "module_sched/sched_scheduler.h"
28
29 namespace OHOS::FileManagement::Backup {
30 using namespace std;
31 using namespace testing;
32
33 namespace {
34 const string BUNDLE_NAME = "com.example.app2backup";
35 constexpr int32_t CLIENT_TOKEN_ID = 100;
36 constexpr int32_t SERVICE_ID = 5203;
37 constexpr int32_t WAIT_TIME = 3;
38 } // namespace
39
40 class SchedSchedulerTest : public testing::Test {
41 public:
42 static void SetUpTestCase(void);
43 static void TearDownTestCase();
SetUp()44 void SetUp() override {};
TearDown()45 void TearDown() override {};
46 void Init(IServiceReverseType::Scenario scenario);
47
48 static inline sptr<SchedScheduler> schedPtr_ = nullptr;
49 static inline sptr<SvcSessionManager> sessionManagerPtr_ = nullptr;
50 static inline sptr<Service> servicePtr_ = nullptr;
51 };
52
SetUpTestCase()53 void SchedSchedulerTest::SetUpTestCase()
54 {
55 servicePtr_ = sptr<Service>(new Service(SERVICE_ID));
56 sessionManagerPtr_ = sptr<SvcSessionManager>(new SvcSessionManager(wptr(servicePtr_)));
57 schedPtr_ = sptr(new SchedScheduler(wptr(servicePtr_), wptr(sessionManagerPtr_)));
58 }
59
TearDownTestCase()60 void SchedSchedulerTest::TearDownTestCase()
61 {
62 schedPtr_ = nullptr;
63 servicePtr_ = nullptr;
64 sessionManagerPtr_ = nullptr;
65 }
66
Init(IServiceReverseType::Scenario scenario)67 void SchedSchedulerTest::Init(IServiceReverseType::Scenario scenario)
68 {
69 vector<string> bundleNames;
70 map<string, BackupExtInfo> backupExtNameMap;
71 bundleNames.emplace_back(BUNDLE_NAME);
72 auto setBackupExtNameMap = [](const string &bundleName) {
73 BackupExtInfo info {};
74 info.backupExtName = BUNDLE_NAME;
75 info.receExtManageJson = true;
76 info.receExtAppDone = true;
77 return make_pair(bundleName, info);
78 };
79 transform(bundleNames.begin(), bundleNames.end(), inserter(backupExtNameMap, backupExtNameMap.end()),
80 setBackupExtNameMap);
81
82 EXPECT_TRUE(sessionManagerPtr_ != nullptr);
83 sessionManagerPtr_->Active({
84 .clientToken = CLIENT_TOKEN_ID,
85 .scenario = scenario,
86 .backupExtNameMap = move(backupExtNameMap),
87 });
88 }
89
90 /**
91 * @tc.number: SUB_Service_Sched_0100
92 * @tc.name: SUB_Service_Sched_0100
93 * @tc.desc: 测试 Sched接口
94 * @tc.size: MEDIUM
95 * @tc.type: FUNC
96 * @tc.level Level 1
97 * @tc.require: I6F3GV
98 */
99 HWTEST_F(SchedSchedulerTest, SUB_Service_Sched_0100, testing::ext::TestSize.Level1)
100 {
101 GTEST_LOG_(INFO) << "SchedSchedulerTest-begin SUB_Service_Sched_0100";
102 try {
103 Init(IServiceReverseType::Scenario::BACKUP);
104 EXPECT_TRUE(schedPtr_ != nullptr);
105 schedPtr_->Sched();
106 GTEST_LOG_(INFO) << "SchedSchedulerTest-Sched Branches";
107 sessionManagerPtr_->SetServiceSchedAction(BUNDLE_NAME, BConstants::ServiceSchedAction::FINISH);
108 schedPtr_->Sched();
109 sessionManagerPtr_->SetServiceSchedAction(BUNDLE_NAME, BConstants::ServiceSchedAction::WAIT);
110 schedPtr_->Sched();
111 GTEST_LOG_(INFO) << "SchedSchedulerTest-ExecutingQueueTasks time callback";
112 schedPtr_->RemoveExtConn(BUNDLE_NAME);
113 schedPtr_->Sched("test");
114 // SchedScheduler在析构时释放OHOS::ThreadPool若此时没有完成的任务，ThreadPool在析构调用stop时会出现异常，目前暂时sleep处理，后续更改ThreadPool方案
115 // 原因是sleep等待ThreadPool任务完成后，任务不再额外持有SchedScheduler的引用计数
116 // 于是本函数执行结束后SchedScheduler引用计数清空，得以在另外的线程上析构SchedScheduler对象及其中的线程池。
117 sleep(WAIT_TIME);
118 } catch (...) {
119 EXPECT_TRUE(false);
120 GTEST_LOG_(INFO) << "SchedSchedulerTest-an exception occurred by Sched.";
121 }
122 GTEST_LOG_(INFO) << "SchedSchedulerTest-end SUB_Service_Sched_0100";
123 }
124
125 /**
126 * @tc.number: SUB_Service_Sched_0200
127 * @tc.name: SUB_Service_Sched_0200
128 * @tc.desc: 测试 Sched接口,sessionPtr_为空场景
129 * @tc.size: MEDIUM
130 * @tc.type: FUNC
131 * @tc.level Level 1
132 * @tc.require: I6F3GV
133 */
134 HWTEST_F(SchedSchedulerTest, SUB_Service_Sched_0200, testing::ext::TestSize.Level1)
135 {
136 GTEST_LOG_(INFO) << "SchedSchedulerTest-begin SUB_Service_Sched_0200";
137 try {
138 sptr<SchedScheduler> schedPtrNull = sptr(new SchedScheduler(wptr(servicePtr_), nullptr));
139 schedPtrNull->Sched();
140 schedPtrNull->ExecutingQueueTasks("test");
141 EXPECT_TRUE(true);
142 } catch (...) {
143 EXPECT_TRUE(false);
144 GTEST_LOG_(INFO) << "SchedSchedulerTest-an exception occurred by Sched.";
145 }
146 GTEST_LOG_(INFO) << "SchedSchedulerTest-end SUB_Service_Sched_0200";
147 }
148
149 /**
150 * @tc.number: SUB_Service_ExecutingQueueTasks_0100
151 * @tc.name: SUB_Service_ExecutingQueueTasks_0100
152 * @tc.desc: 测试 ExecutingQueueTasks 接口
153 * @tc.size: MEDIUM
154 * @tc.type: FUNC
155 * @tc.level Level 1
156 * @tc.require: I6F3GV
157 */
158 HWTEST_F(SchedSchedulerTest, SUB_Service_ExecutingQueueTasks_0100, testing::ext::TestSize.Level1)
159 {
160 GTEST_LOG_(INFO) << "SchedSchedulerTest-begin SUB_Service_ExecutingQueueTasks_0100";
161 try {
162 EXPECT_TRUE(sessionManagerPtr_ != nullptr);
163 EXPECT_TRUE(schedPtr_ != nullptr);
164 sessionManagerPtr_->SetServiceSchedAction(BUNDLE_NAME, BConstants::ServiceSchedAction::START);
165 schedPtr_->ExecutingQueueTasks(BUNDLE_NAME);
166 sessionManagerPtr_->SetServiceSchedAction(BUNDLE_NAME, BConstants::ServiceSchedAction::RUNNING);
167 schedPtr_->ExecutingQueueTasks(BUNDLE_NAME);
168 sessionManagerPtr_->SetServiceSchedAction(BUNDLE_NAME, BConstants::ServiceSchedAction::CLEAN);
169 schedPtr_->ExecutingQueueTasks(BUNDLE_NAME);
170 } catch (...) {
171 EXPECT_TRUE(false);
172 GTEST_LOG_(INFO) << "SchedSchedulerTest-an exception occurred by ExecutingQueueTasks.";
173 }
174 GTEST_LOG_(INFO) << "SchedSchedulerTest-end SUB_Service_ExecutingQueueTasks_0100";
175 }
176
177 /**
178 * @tc.number: SUB_Service_RemoveExtConn_0100
179 * @tc.name: SUB_Service_RemoveExtConn_0100
180 * @tc.desc: 测试 RemoveExtConn 接口
181 * @tc.size: MEDIUM
182 * @tc.type: FUNC
183 * @tc.level Level 1
184 * @tc.require: I6F3GV
185 */
186 HWTEST_F(SchedSchedulerTest, SUB_Service_RemoveExtConn_0100, testing::ext::TestSize.Level1)
187 {
188 GTEST_LOG_(INFO) << "SchedSchedulerTest-begin SUB_Service_RemoveExtConn_0100";
189 try {
190 EXPECT_TRUE(schedPtr_ != nullptr);
191 schedPtr_->RemoveExtConn("test");
192 } catch (...) {
193 EXPECT_TRUE(false);
194 GTEST_LOG_(INFO) << "SchedSchedulerTest-an exception occurred by RemoveExtConn.";
195 }
196 GTEST_LOG_(INFO) << "SchedSchedulerTest-end SUB_Service_RemoveExtConn_0100";
197 }
198
199 /**
200 * @tc.number: SUB_Service_TryUnloadServiceTimer_0100
201 * @tc.name: SUB_Service_TryUnloadServiceTimer_0100
202 * @tc.desc: 测试 TryUnloadServiceTimer 接口
203 * @tc.size: MEDIUM
204 * @tc.type: FUNC
205 * @tc.level Level 1
206 * @tc.require: I6F3GV
207 */
208 HWTEST_F(SchedSchedulerTest, SUB_Service_TryUnloadServiceTimer_0100, testing::ext::TestSize.Level1)
209 {
210 GTEST_LOG_(INFO) << "SchedSchedulerTest-begin SUB_Service_TryUnloadServiceTimer_0100";
211 try {
212 EXPECT_TRUE(schedPtr_ != nullptr);
213 schedPtr_->TryUnloadServiceTimer(true);
214 schedPtr_->TryUnloadServiceTimer(false);
215 } catch (...) {
216 EXPECT_TRUE(false);
217 GTEST_LOG_(INFO) << "SchedSchedulerTest-an exception occurred by TryUnloadServiceTimer.";
218 }
219 GTEST_LOG_(INFO) << "SchedSchedulerTest-end SUB_Service_TryUnloadServiceTimer_0100";
220 }
221
222 /**
223 * @tc.number: SUB_Service_TryUnloadService_0100
224 * @tc.name: SUB_Service_TryUnloadServicer_0100
225 * @tc.desc: 测试 TryUnloadService 接口
226 * @tc.size: MEDIUM
227 * @tc.type: FUNC
228 * @tc.level Level 1
229 * @tc.require: I6F3GV
230 */
231 HWTEST_F(SchedSchedulerTest, SUB_Service_TryUnloadService_0100, testing::ext::TestSize.Level1)
232 {
233 GTEST_LOG_(INFO) << "SchedSchedulerTest-begin SUB_Service_TryUnloadService_0100";
234 try {
235 EXPECT_TRUE(schedPtr_ != nullptr);
236 schedPtr_->TryUnloadService();
237 } catch (...) {
238 EXPECT_TRUE(false);
239 GTEST_LOG_(INFO) << "SchedSchedulerTest-an exception occurred by TryUnloadService.";
240 }
241 GTEST_LOG_(INFO) << "SchedSchedulerTest-end SUB_Service_TryUnloadService_0100";
242 }
243 } // namespace OHOS::FileManagement::Backup