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 <gtest/gtest.h>
17 #include <string>
18 #include <vector>
19 
20 #include "admin_manager.h"
21 #include "enterprise_admin_connection.h"
22 #include "enterprise_admin_stub_mock.h"
23 #include "func_code.h"
24 
25 using namespace testing::ext;
26 using namespace testing;
27 
28 namespace OHOS {
29 namespace EDM {
30 namespace TEST {
31 constexpr uint32_t COMMAND_ON_ADMIN_CODE = 3;
32 constexpr int32_t DEFAULT_USER_ID = 100;
33 
34 class EnterpriseAdminConnectionTest : public testing::Test {
35 protected:
36     void SetUp() override;
37 
TearDown()38     void TearDown() override {}
39 
40     std::shared_ptr<EnterpriseAdminConnection> enterpriseAdminConnectionTest {nullptr};
41     sptr<EnterpriseAdminStubMock> remoteObject {nullptr};
42 };
43 
SetUp()44 void EnterpriseAdminConnectionTest::SetUp()
45 {
46     remoteObject = new (std::nothrow) EnterpriseAdminStubMock();
47 }
48 
49 /**
50  * @tc.name: TestOnAbilityConnectDone01
51  * @tc.desc: Test EnterpriseAdminConnection::OnAbilityConnectDone func.
52  * @tc.type: FUNC
53  */
54 HWTEST_F(EnterpriseAdminConnectionTest, TestOnAbilityConnectDone01, TestSize.Level1)
55 {
56     AppExecFwk::ElementName admin;
57     admin.SetBundleName("com.edm.test.demo");
58     admin.SetAbilityName("com.edm.test.demo.Ability");
59     AAFwk::Want connectWant;
60     connectWant.SetElementName("com.edm.test.demo", "com.edm.test.demo.Ability");
61     int32_t resultCode = 0;
62     enterpriseAdminConnectionTest =
63         std::make_shared<EnterpriseAdminConnection>(connectWant,
64         IEnterpriseAdmin::COMMAND_ON_ADMIN_ENABLED, DEFAULT_USER_ID, true);
65     EXPECT_CALL(*remoteObject, SendRequest(_, _, _, _))
66         .Times(1)
67         .WillOnce(Invoke(remoteObject.GetRefPtr(), &EnterpriseAdminStubMock::InvokeSendRequest));
68     enterpriseAdminConnectionTest->OnAbilityConnectDone(admin, remoteObject, resultCode);
69     enterpriseAdminConnectionTest->OnAbilityDisconnectDone(admin, resultCode);
70 
71     EXPECT_TRUE(remoteObject->code_ == IEnterpriseAdmin::COMMAND_ON_ADMIN_ENABLED);
72 }
73 
74 /**
75  * @tc.name: TestOnAbilityConnectDone02
76  * @tc.desc: Test EnterpriseAdminConnection::OnAbilityConnectDone func.
77  * @tc.type: FUNC
78  */
79 HWTEST_F(EnterpriseAdminConnectionTest, TestOnAbilityConnectDone02, TestSize.Level1)
80 {
81     AppExecFwk::ElementName admin;
82     admin.SetBundleName("com.edm.test.demo");
83     admin.SetAbilityName("com.edm.test.demo.Ability");
84     AAFwk::Want connectWant;
85     connectWant.SetElementName("com.edm.test.demo", "com.edm.test.demo.Ability");
86     int32_t resultCode = 0;
87     enterpriseAdminConnectionTest =
88         std::make_shared<EnterpriseAdminConnection>(connectWant,
89         IEnterpriseAdmin::COMMAND_ON_ADMIN_DISABLED, DEFAULT_USER_ID, true);
90     EXPECT_CALL(*remoteObject, SendRequest(_, _, _, _))
91         .Times(1)
92         .WillOnce(Invoke(remoteObject.GetRefPtr(), &EnterpriseAdminStubMock::InvokeSendRequest));
93     enterpriseAdminConnectionTest->OnAbilityConnectDone(admin, remoteObject, resultCode);
94     enterpriseAdminConnectionTest->OnAbilityDisconnectDone(admin, resultCode);
95 
96     EXPECT_TRUE(remoteObject->code_ == IEnterpriseAdmin::COMMAND_ON_ADMIN_DISABLED);
97 }
98 
99 
100 /**
101  * @tc.name: TestOnAbilityConnectDone03
102  * @tc.desc: Test EnterpriseAdminConnection::OnAbilityConnectDone func.
103  * @tc.type: FUNC
104  */
105 HWTEST_F(EnterpriseAdminConnectionTest, TestOnAbilityConnectDone03, TestSize.Level1)
106 {
107     AppExecFwk::ElementName admin;
108     admin.SetBundleName("com.edm.test.demo");
109     admin.SetAbilityName("com.edm.test.demo.Ability");
110     AAFwk::Want connectWant;
111     connectWant.SetElementName("com.edm.test.demo", "com.edm.test.demo.Ability");
112     int32_t resultCode = 0;
113     enterpriseAdminConnectionTest = std::make_shared<EnterpriseAdminConnection>(connectWant,
114         COMMAND_ON_ADMIN_CODE, DEFAULT_USER_ID, true);
115     enterpriseAdminConnectionTest->OnAbilityConnectDone(admin, remoteObject, resultCode);
116     enterpriseAdminConnectionTest->OnAbilityDisconnectDone(admin, resultCode);
117 
118     EXPECT_TRUE(remoteObject->code_ != IEnterpriseAdmin::COMMAND_ON_ADMIN_DISABLED);
119 }
120 
121 /**
122  * @tc.name: TestOnAbilityConnectDoneWithoutAdminEnabled
123  * @tc.desc: Test EnterpriseAdminConnection::OnAbilityConnectDone func.
124  * @tc.type: FUNC
125  */
126 HWTEST_F(EnterpriseAdminConnectionTest, TestOnAbilityConnectDoneWithoutAdminEnabled, TestSize.Level1)
127 {
128     AppExecFwk::ElementName admin;
129     admin.SetBundleName("com.edm.test.demo");
130     admin.SetAbilityName("com.edm.test.demo.Ability");
131     AAFwk::Want connectWant;
132     connectWant.SetElementName("com.edm.test.demo", "com.edm.test.demo.Ability");
133     int32_t resultCode = 0;
134     enterpriseAdminConnectionTest = std::make_shared<EnterpriseAdminConnection>(connectWant,
135         IEnterpriseAdmin::COMMAND_ON_ADMIN_ENABLED, DEFAULT_USER_ID, true);
136     enterpriseAdminConnectionTest->SetIsOnAdminEnabled(false);
137     enterpriseAdminConnectionTest->OnAbilityConnectDone(admin, remoteObject, resultCode);
138 
139     EXPECT_TRUE(remoteObject->code_ != IEnterpriseAdmin::COMMAND_ON_ADMIN_ENABLED);
140 }
141 
142 /**
143  * @tc.name: TestOnAbilityDisconnectDoneWithSuperAdmin
144  * @tc.desc: Test EnterpriseAdminConnection::OnAbilityDisconnectDone func.
145  * @tc.type: FUNC
146  */
147 HWTEST_F(EnterpriseAdminConnectionTest, TestOnAbilityDisconnectDoneWithSuperAdmin, TestSize.Level1)
148 {
149     AppExecFwk::ExtensionAbilityInfo abilityInfo;
150     abilityInfo.bundleName = "com.edm.test.demo";
151     abilityInfo.name = "com.edm.test.demo.Ability";
152     EntInfo entInfo;
153     entInfo.enterpriseName = "company";
154     entInfo.description = "technology company in wuhan";
155     std::vector<std::string> permissions = {"ohos.permission.EDM_TEST_PERMISSION"};
156     Admin edmAdmin(abilityInfo, AdminType::ENT, entInfo, permissions, false);
157     AdminManager::GetInstance()->SetAdminValue(DEFAULT_USER_ID, edmAdmin);
158 
159     AAFwk::Want connectWant;
160     connectWant.SetElementName(abilityInfo.bundleName, abilityInfo.name);
161     enterpriseAdminConnectionTest = std::make_shared<EnterpriseAdminConnection>(connectWant,
162         IEnterpriseAdmin::COMMAND_ON_ADMIN_ENABLED, DEFAULT_USER_ID, true);
163     AppExecFwk::ElementName admin;
164     admin.SetBundleName(abilityInfo.bundleName);
165     admin.SetAbilityName(abilityInfo.name);
166     int32_t resultCode = 0;
167     enterpriseAdminConnectionTest->OnAbilityDisconnectDone(admin, resultCode);
168     EXPECT_TRUE(AdminManager::GetInstance()->IsSuperAdmin(abilityInfo.bundleName));
169 
170     AdminManager::GetInstance()->DeleteAdmin(admin.GetBundleName(), DEFAULT_USER_ID);
171 }
172 } // namespace TEST
173 } // namespace EDM
174 } // namespace OHOS