1 /*
2 * Copyright (C) 2022 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 "schedule_node_helper_test.h"
17
18 #include "iam_ptr.h"
19
20 #include "mock_schedule_node_callback.h"
21 #include "mock_resource_node.h"
22 #include "resource_node_pool.h"
23 #include "schedule_node_helper.h"
24
25 namespace OHOS {
26 namespace UserIam {
27 namespace UserAuth {
28 using namespace testing;
29 using namespace testing::ext;
SetUpTestCase()30 void ScheduleNodeHelperTest::SetUpTestCase()
31 {
32 }
33
TearDownTestCase()34 void ScheduleNodeHelperTest::TearDownTestCase()
35 {
36 }
37
SetUp()38 void ScheduleNodeHelperTest::SetUp()
39 {
40 }
41
TearDown()42 void ScheduleNodeHelperTest::TearDown()
43 {
44 }
45
46 HWTEST_F(ScheduleNodeHelperTest, ScheduleNodeHelperTest_001, TestSize.Level0)
47 {
48 std::vector<HdiScheduleInfo> scheduleInfoList;
49 auto callback = Common::MakeShared<MockScheduleNodeCallback>();
50 EXPECT_NE(callback, nullptr);
51 std::vector<std::shared_ptr<ScheduleNode>> scheduleNodeList;
52 ScheduleNodeHelper::NodeOptionalPara para = {};
53 EXPECT_TRUE(ScheduleNodeHelper::BuildFromHdi(scheduleInfoList, callback, scheduleNodeList, para));
54 }
55
56 HWTEST_F(ScheduleNodeHelperTest, ScheduleNodeHelperTest_002, TestSize.Level0)
57 {
58 std::vector<HdiScheduleInfo> scheduleInfoList;
59 HdiScheduleInfo scheduleInfo = {};
60 scheduleInfo.authType = HdiAuthType::FACE;
61 scheduleInfo.executorMatcher = 10;
62 scheduleInfo.executorIndexes.push_back(60);
63
64 scheduleInfo.executorIndexes.push_back(90);
65
66 scheduleInfo.scheduleId = 20;
67 scheduleInfo.scheduleMode = HdiScheduleMode::AUTH;
68 scheduleInfo.templateIds.push_back(30);
69 scheduleInfoList.push_back(scheduleInfo);
70
71 std::vector<std::vector<ExecutorRole>> executorRoles = {
72 {COLLECTOR, COLLECTOR},
73 {VERIFIER, VERIFIER},
74 {ALL_IN_ONE, ALL_IN_ONE},
75 {COLLECTOR, VERIFIER},
76 {COLLECTOR, ALL_IN_ONE},
77 {VERIFIER, COLLECTOR},
78 {VERIFIER, ALL_IN_ONE},
79 {ALL_IN_ONE, COLLECTOR},
80 {ALL_IN_ONE, VERIFIER},
81 };
82
83 for (const auto &executorRole : executorRoles) {
84 auto resourceNode1 = MockResourceNode::CreateWithExecuteIndex(60, FACE, executorRole[0]);
85 EXPECT_NE(resourceNode1, nullptr);
86 auto resourceNode2 = MockResourceNode::CreateWithExecuteIndex(90, FACE, executorRole[1]);
87 EXPECT_NE(resourceNode2, nullptr);
88 EXPECT_TRUE(ResourceNodePool::Instance().Insert(resourceNode1));
89 EXPECT_TRUE(ResourceNodePool::Instance().Insert(resourceNode2));
90
91 auto callback = Common::MakeShared<MockScheduleNodeCallback>();
92 EXPECT_NE(callback, nullptr);
93 std::vector<std::shared_ptr<ScheduleNode>> scheduleNodeList;
94 ScheduleNodeHelper::NodeOptionalPara para = {};
95 ScheduleNodeHelper::BuildFromHdi(scheduleInfoList, callback, scheduleNodeList, para);
96 ResourceNodePool::Instance().DeleteAll();
97 }
98 }
99
100 HWTEST_F(ScheduleNodeHelperTest, ScheduleNodeHelperTestScheduleInfoToExecutors_001, TestSize.Level0)
101 {
102 HdiScheduleInfo scheduleInfo = {};
103 scheduleInfo.authType = HdiAuthType::FACE;
104 scheduleInfo.executorMatcher = 10;
105 scheduleInfo.executorIndexes.push_back(60);
106 scheduleInfo.executorMessages.push_back({6});
107 scheduleInfo.scheduleId = 20;
108 scheduleInfo.scheduleMode = HdiScheduleMode::AUTH;
109 scheduleInfo.templateIds.push_back(30);
110 std::vector<ExecutorRole> executorRole = {SCHEDULER, COLLECTOR, VERIFIER, ALL_IN_ONE};
111 auto resourceNode1 = MockResourceNode::CreateWithExecuteIndex(60, FACE, executorRole[0]);
112 EXPECT_NE(resourceNode1, nullptr);
113 EXPECT_TRUE(ResourceNodePool::Instance().Insert(resourceNode1));
114 std::shared_ptr<ResourceNode> collector;
115 std::shared_ptr<ResourceNode> verifier;
116 std::vector<uint8_t> collectorMessage;
117 std::vector<uint8_t> verifierMessage;
118 auto callback = Common::MakeShared<MockScheduleNodeCallback>();
119 EXPECT_NE(callback, nullptr);
120 EXPECT_FALSE(ScheduleNodeHelper::ScheduleInfoToExecutors(scheduleInfo, collector, verifier, collectorMessage,
121 verifierMessage));
122 ResourceNodePool::Instance().DeleteAll();
123 }
124
125 HWTEST_F(ScheduleNodeHelperTest, ScheduleNodeHelperTestScheduleInfoToExecutors_002, TestSize.Level0)
126 {
127 HdiScheduleInfo scheduleInfo = {};
128 scheduleInfo.authType = HdiAuthType::FACE;
129 scheduleInfo.executorMatcher = 10;
130 scheduleInfo.executorIndexes.push_back(60);
131 scheduleInfo.executorMessages.push_back({6});
132 scheduleInfo.scheduleId = 20;
133 scheduleInfo.scheduleMode = HdiScheduleMode::AUTH;
134 scheduleInfo.templateIds.push_back(30);
135 std::vector<ExecutorRole> executorRole = {SCHEDULER, COLLECTOR, VERIFIER, ALL_IN_ONE};
136 auto resourceNode1 = MockResourceNode::CreateWithExecuteIndex(60, FACE, executorRole[1]);
137 EXPECT_NE(resourceNode1, nullptr);
138 EXPECT_TRUE(ResourceNodePool::Instance().Insert(resourceNode1));
139 std::shared_ptr<ResourceNode> collector;
140 std::shared_ptr<ResourceNode> verifier;
141 std::vector<uint8_t> collectorMessage;
142 std::vector<uint8_t> verifierMessage;
143 auto callback = Common::MakeShared<MockScheduleNodeCallback>();
144 EXPECT_NE(callback, nullptr);
145 EXPECT_FALSE(ScheduleNodeHelper::ScheduleInfoToExecutors(scheduleInfo, collector, verifier, collectorMessage,
146 verifierMessage));
147 ResourceNodePool::Instance().DeleteAll();
148 }
149
150 HWTEST_F(ScheduleNodeHelperTest, ScheduleNodeHelperTestScheduleInfoToExecutors_003, TestSize.Level0)
151 {
152 HdiScheduleInfo scheduleInfo = {};
153 scheduleInfo.authType = HdiAuthType::FACE;
154 scheduleInfo.executorMatcher = 10;
155 scheduleInfo.executorIndexes.push_back(60);
156 scheduleInfo.executorMessages.push_back({6});
157 scheduleInfo.scheduleId = 20;
158 scheduleInfo.scheduleMode = HdiScheduleMode::AUTH;
159 scheduleInfo.templateIds.push_back(30);
160 std::vector<ExecutorRole> executorRole = {SCHEDULER, COLLECTOR, VERIFIER, ALL_IN_ONE};
161 auto resourceNode1 = MockResourceNode::CreateWithExecuteIndex(60, FACE, executorRole[2]);
162 EXPECT_NE(resourceNode1, nullptr);
163 EXPECT_TRUE(ResourceNodePool::Instance().Insert(resourceNode1));
164 std::shared_ptr<ResourceNode> collector;
165 std::shared_ptr<ResourceNode> verifier;
166 std::vector<uint8_t> collectorMessage;
167 std::vector<uint8_t> verifierMessage;
168 auto callback = Common::MakeShared<MockScheduleNodeCallback>();
169 EXPECT_NE(callback, nullptr);
170 EXPECT_FALSE(ScheduleNodeHelper::ScheduleInfoToExecutors(scheduleInfo, collector, verifier, collectorMessage,
171 verifierMessage));
172 ResourceNodePool::Instance().DeleteAll();
173 }
174 } // namespace UserAuth
175 } // namespace UserIam
176 } // namespace OHOS
177