1 /*
2 * Copyright (c) 2021-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 "gtest/gtest.h"
17
18 #include "iam_logger.h"
19 #include "iam_ptr.h"
20
21 #include "driver_manager.h"
22 #include "iam_executor_idriver_manager.h"
23 #include "mock_iauth_driver_hdi.h"
24 #include "mock_iauth_executor_hdi.h"
25
26 #define LOG_TAG "USER_AUTH_EXECUTOR"
27
28 using namespace testing;
29 using namespace testing::ext;
30
31 namespace OHOS {
32 namespace UserIam {
33 namespace UserAuth {
34 using namespace OHOS::UserIam::UserAuth;
35 class DriverManagerUnitTest : public testing::Test {
36 public:
37 static void SetUpTestCase();
38 static void TearDownTestCase();
39 void SetUp();
40 void TearDown();
41 };
42
SetUpTestCase()43 void DriverManagerUnitTest::SetUpTestCase()
44 {
45 }
46
TearDownTestCase()47 void DriverManagerUnitTest::TearDownTestCase()
48 {
49 }
50
SetUp()51 void DriverManagerUnitTest::SetUp()
52 {
53 }
54
TearDown()55 void DriverManagerUnitTest::TearDown()
56 {
57 }
58
59 HWTEST_F(DriverManagerUnitTest, DriverManagerTest_001, TestSize.Level0)
60 {
61 std::string serviceName = "mockDriver";
62 HdiConfig config = {};
63 config.id = 10;
64 config.driver = nullptr;
65 std::map<std::string, HdiConfig> hdiName2Config;
66 EXPECT_EQ(DriverManager::GetInstance().Start(hdiName2Config), USERAUTH_SUCCESS);
67 hdiName2Config.emplace(serviceName, config);
68 EXPECT_EQ(DriverManager::GetInstance().Start(hdiName2Config), USERAUTH_ERROR);
69 EXPECT_EQ(DriverManager::GetInstance().GetDriverByServiceName(serviceName), nullptr);
70 config.driver = Common::MakeShared<MockIAuthDriverHdi>();
71 EXPECT_EQ(DriverManager::GetInstance().Start(hdiName2Config), USERAUTH_ERROR);
72 DriverManager::GetInstance().GetDriverByServiceName(serviceName);
73 EXPECT_EQ(DriverManager::GetInstance().Start(hdiName2Config), USERAUTH_ERROR);
74 DriverManager::GetInstance().OnFrameworkReady();
75 DriverManager::GetInstance().SubscribeHdiDriverStatus();
76 DriverManager::GetInstance().OnAllHdiDisconnect();
77 }
78
79 HWTEST_F(DriverManagerUnitTest, DriverManagerTest_002, TestSize.Level0)
80 {
81 std::string serviceName = "mockDriver";
82 HdiConfig config = {};
83 config.id = 10;
84 config.driver = nullptr;
85 std::map<std::string, HdiConfig> hdiName2Config;
86 EXPECT_EQ(IDriverManager::Start(hdiName2Config), USERAUTH_SUCCESS);
87 hdiName2Config.emplace(serviceName, config);
88 EXPECT_EQ(IDriverManager::Start(hdiName2Config), USERAUTH_ERROR);
89 config.driver = Common::MakeShared<MockIAuthDriverHdi>();
90 EXPECT_EQ(IDriverManager::Start(hdiName2Config), USERAUTH_ERROR);
91 }
92
93 HWTEST_F(DriverManagerUnitTest, DriverManagerTest_003, TestSize.Level0)
94 {
95 std::string serviceName1 = "mockDriver1";
96 std::string serviceName2 = "mockDriver2";
97 HdiConfig config = {};
98 config.id = 10;
99 config.driver = Common::MakeShared<MockIAuthDriverHdi>();
100 std::map<std::string, HdiConfig> hdiName2Config;
101 hdiName2Config.emplace(serviceName1, config);
102 EXPECT_EQ(DriverManager::GetInstance().HdiConfigIsValid(hdiName2Config), true);
103 hdiName2Config.emplace(serviceName2, config);
104 EXPECT_EQ(DriverManager::GetInstance().HdiConfigIsValid(hdiName2Config), false);
105 EXPECT_EQ(IDriverManager::Start(hdiName2Config), USERAUTH_ERROR);
106 }
107
108 HWTEST_F(DriverManagerUnitTest, DriverManagerTest_004, TestSize.Level0)
109 {
110 std::string serviceName = "mockDriver";
111 HdiConfig config = {};
112 config.id = 10;
113 config.driver = Common::MakeShared<MockIAuthDriverHdi>();
114 std::map<std::string, HdiConfig> hdiName2Config;
115 hdiName2Config.emplace(serviceName, config);
116 EXPECT_EQ(DriverManager::GetInstance().HdiConfigIsValid(hdiName2Config), true);
117 std::shared_ptr<Driver> mockDriver = Common::MakeShared<Driver>(serviceName, config);
118 DriverManager::GetInstance().serviceName2Driver_.emplace(serviceName, mockDriver);
119 EXPECT_EQ(DriverManager::GetInstance().Start(hdiName2Config), USERAUTH_SUCCESS);
120 EXPECT_NE(DriverManager::GetInstance().GetDriverByServiceName(serviceName), nullptr);
121 DriverManager::GetInstance().OnFrameworkReady();
122 DriverManager::GetInstance().SubscribeHdiDriverStatus();
123 DriverManager::GetInstance().OnAllHdiDisconnect();
124 }
125
126 HWTEST_F(DriverManagerUnitTest, DriverManagerTest_005, TestSize.Level0)
127 {
128 std::string serviceName = "mockDriver";
129 HdiConfig config = {};
130 config.id = 10;
131 config.driver = Common::MakeShared<MockIAuthDriverHdi>();
132 std::map<std::string, HdiConfig> hdiName2Config;
133 hdiName2Config.emplace(serviceName, config);
134 EXPECT_EQ(DriverManager::GetInstance().HdiConfigIsValid(hdiName2Config), true);
135 DriverManager::GetInstance().serviceName2Driver_.emplace(serviceName, nullptr);
136 EXPECT_EQ(DriverManager::GetInstance().Start(hdiName2Config), USERAUTH_SUCCESS);
137 EXPECT_NE(DriverManager::GetInstance().GetDriverByServiceName(serviceName), nullptr);
138 DriverManager::GetInstance().OnFrameworkReady();
139 DriverManager::GetInstance().SubscribeHdiDriverStatus();
140 DriverManager::GetInstance().OnAllHdiDisconnect();
141 DriverManager::GetInstance().OnAllHdiDisconnect();
142 }
143
144 HWTEST_F(DriverManagerUnitTest, DriverManager_OnFrameworkDownTest_001, TestSize.Level0)
145 {
146 std::string serviceName = "mockDriver";
147 HdiConfig config = {};
148 config.id = 10;
149 config.driver = Common::MakeShared<MockIAuthDriverHdi>();
150 std::map<std::string, HdiConfig> hdiName2Config;
151 hdiName2Config.emplace(serviceName, config);
152 EXPECT_EQ(DriverManager::GetInstance().HdiConfigIsValid(hdiName2Config), true);
153 DriverManager::GetInstance().serviceName2Driver_.emplace(serviceName, nullptr);
154 EXPECT_EQ(DriverManager::GetInstance().Start(hdiName2Config), USERAUTH_SUCCESS);
155 config.driver->OnFrameworkDown();
156 DriverManager::GetInstance().OnFrameworkDown();
157 for (auto const &pair : DriverManager::GetInstance().serviceName2Driver_) {
158 if (pair.second == nullptr) {
159 continue;
160 }
161 EXPECT_EQ(pair.second->isFwkReady_, false);
162 }
163 }
164 } // namespace UserAuth
165 } // namespace UserIam
166 } // namespace OHOS
167