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