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 } // namespace UserAuth
100 } // namespace UserIam
101 } // namespace OHOS
102