1 /*
2 * Copyright (c) 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 "common/include/task_scheduler.h"
17 #include <gtest/gtest.h>
18
19 using namespace testing;
20 using namespace testing::ext;
21 namespace OHOS {
22 namespace Rosen {
23 class TaskSchedulerText : public testing::Test {
24 public:
TaskSchedulerText()25 TaskSchedulerText() {}
~TaskSchedulerText()26 ~TaskSchedulerText() {}
27 };
28
29 namespace {
30 /**
31 * @tc.name: task_scheduler_test001
32 * @tc.desc: normal function
33 * @tc.type: FUNC
34 */
HWTEST_F(TaskSchedulerText, task_scheduler_test001, Function | SmallTest | Level2)35 HWTEST_F(TaskSchedulerText, task_scheduler_test001, Function | SmallTest | Level2)
36 {
37 GTEST_LOG_(INFO) << "TaskSchedulerText: task_scheduler_test001 start";
38 std::string threadName = "threadName";
39 std::string name = "name";
40 TaskScheduler* taskScheduler = new(std::nothrow) TaskScheduler(threadName);
41 int res = 0;
42 taskScheduler->RemoveTask(name);
43 ASSERT_EQ(res, 0);
44 delete taskScheduler;
45 GTEST_LOG_(INFO) << "TaskSchedulerText: task_scheduler_test001 end";
46 }
47
48 /**
49 * @tc.name: GetEventHandler
50 * @tc.desc: GetEventHandler function
51 * @tc.type: FUNC
52 */
HWTEST_F(TaskSchedulerText, GetEventHandler, Function | SmallTest | Level2)53 HWTEST_F(TaskSchedulerText, GetEventHandler, Function | SmallTest | Level2)
54 {
55 std::string threadName = "threadName";
56 std::shared_ptr<TaskScheduler> taskScheduler = std::make_shared<TaskScheduler>(threadName);
57 ASSERT_NE(taskScheduler, nullptr);
58 EXPECT_NE(taskScheduler->GetEventHandler(), nullptr);
59 }
60
61 /**
62 * @tc.name: PostTask
63 * @tc.desc: PostTask function
64 * @tc.type: FUNC
65 */
HWTEST_F(TaskSchedulerText, PostTask, Function | SmallTest | Level2)66 HWTEST_F(TaskSchedulerText, PostTask, Function | SmallTest | Level2)
67 {
68 std::string threadName = "threadName";
69 std::shared_ptr<TaskScheduler> taskScheduler = std::make_shared<TaskScheduler>(threadName);
70 ASSERT_NE(taskScheduler, nullptr);
71 int resultValue = 0;
72 auto taskFunc = [&resultValue]() {
73 GTEST_LOG_(INFO) << "START_TASK";
74 resultValue = 1;
75 };
76 taskScheduler->PostAsyncTask(taskFunc);
77 EXPECT_NE(taskScheduler->handler_, nullptr);
78 EXPECT_EQ(resultValue, 0);
79
80 std::string name = "ssmTask";
81 int64_t delayTime = 1;
82 taskScheduler->PostAsyncTask(taskFunc, name, delayTime);
83 EXPECT_EQ(resultValue, 0);
84 }
85
HWTEST_F(TaskSchedulerText, AddExportTask1, Function | SmallTest | Level2)86 HWTEST_F(TaskSchedulerText, AddExportTask1, Function | SmallTest | Level2)
87 {
88 std::string threadName = "threadName";
89 std::string funcName = "funcName";
90 std::shared_ptr<TaskScheduler> taskScheduler = std::make_shared<TaskScheduler>(threadName);
91 pid_t taskTid = 0;
92 auto taskFunc = [&taskTid]() {
93 GTEST_LOG_(INFO) << "START_TASK";
94 taskTid = gettid();
95 };
96 ASSERT_NE(taskScheduler, nullptr);
97 ASSERT_EQ(taskScheduler->exportFuncMap_.size(), 0);
98 ASSERT_NE(taskScheduler->ssmTid_, 0);
99 taskScheduler->AddExportTask(funcName, taskFunc);
100 ASSERT_EQ(taskTid, gettid());
101 ASSERT_EQ(taskScheduler->exportFuncMap_.size(), 0);
102 }
103
104
HWTEST_F(TaskSchedulerText, AddExportTask2, Function | SmallTest | Level2)105 HWTEST_F(TaskSchedulerText, AddExportTask2, Function | SmallTest | Level2)
106 {
107 std::string threadName = "threadName";
108 std::string funcName = "funcName";
109 std::shared_ptr<TaskScheduler> taskScheduler = std::make_shared<TaskScheduler>(threadName);
110 pid_t taskTid = 0;
111 auto taskFunc = [&taskTid]() {
112 GTEST_LOG_(INFO) << "START_TASK";
113 taskTid = gettid();
114 };
115 ASSERT_NE(taskScheduler, nullptr);
116 ASSERT_EQ(taskScheduler->exportFuncMap_.size(), 0);
117 ASSERT_NE(taskScheduler->ssmTid_, 0);
118 taskScheduler->ssmTid_ = gettid();
119 taskScheduler->AddExportTask(funcName, taskFunc);
120 ASSERT_EQ(taskTid, 0);
121 ASSERT_NE(taskScheduler->exportFuncMap_.size(), 0);
122 }
123
HWTEST_F(TaskSchedulerText, SetExportHandler, Function | SmallTest | Level2)124 HWTEST_F(TaskSchedulerText, SetExportHandler, Function | SmallTest | Level2)
125 {
126 std::string exportThreadName = "exportThread";
127 auto eventRunner = AppExecFwk::EventRunner::Create(exportThreadName);
128 auto eventHandler = std::make_shared<AppExecFwk::EventHandler>(eventRunner);
129 std::string threadName = "threadName";
130 std::shared_ptr<TaskScheduler> taskScheduler = std::make_shared<TaskScheduler>(threadName);
131 taskScheduler->SetExportHandler(eventHandler);
132 ASSERT_EQ(eventHandler.get(), taskScheduler->exportHandler_.lock().get());
133 }
134
HWTEST_F(TaskSchedulerText, ExecuteExportTask, Function | SmallTest | Level2)135 HWTEST_F(TaskSchedulerText, ExecuteExportTask, Function | SmallTest | Level2)
136 {
137 std::string threadName = "threadName";
138 std::shared_ptr<TaskScheduler> taskScheduler = std::make_shared<TaskScheduler>(threadName);
139 bool executed = false;
140 std::string funcName = "funcName";
141 auto taskFunc = [&executed]() {
142 GTEST_LOG_(INFO) << "START_TASK";
143 executed = true;
144 };
145 ASSERT_EQ(taskScheduler->exportFuncMap_.size(), 0);
146 taskScheduler->ExecuteExportTask();
147 ASSERT_EQ(taskScheduler->exportFuncMap_.size(), 0);
148 ASSERT_EQ(executed, false);
149 taskScheduler->exportFuncMap_["taskFunc"] = taskFunc;
150 ASSERT_EQ(taskScheduler->exportFuncMap_.size(), 1);
151 taskScheduler->ExecuteExportTask();
152
153 std::string exportThreadName = "exportThread";
154 auto eventRunner = AppExecFwk::EventRunner::Create(exportThreadName);
155 auto eventHandler = std::make_shared<AppExecFwk::EventHandler>(eventRunner);
156 taskScheduler->SetExportHandler(eventHandler);
157 taskScheduler->ExecuteExportTask();
158 ASSERT_EQ(taskScheduler->exportFuncMap_.size(), 0);
159 }
160 } // namespace
161 } // namespace Rosen
162 } // namespace OHOS
163