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 #include "func_code.h"
20 #include "managed_event.h"
21 #define private public
22 #include "enterprise_conn_manager.h"
23 #undef private
24
25 using namespace testing::ext;
26 using namespace testing;
27
28 namespace OHOS {
29 namespace EDM {
30 namespace TEST {
31 constexpr int32_t DEFAULT_USERID = 100;
32 class EnterpriseConnManagerTest : public testing::Test {
33 protected:
34 void SetUp() override;
35
36 void TearDown() override;
37
38 std::shared_ptr<EnterpriseConnManager> enterpriseConnManagerTest {nullptr};
39 };
40
SetUp()41 void EnterpriseConnManagerTest::SetUp()
42 {
43 enterpriseConnManagerTest = std::make_shared<EnterpriseConnManager>();
44 }
45
TearDown()46 void EnterpriseConnManagerTest::TearDown()
47 {
48 if (enterpriseConnManagerTest) {
49 enterpriseConnManagerTest.reset();
50 }
51 }
52
53 /**
54 * @tc.name: TestAdminConnectAbility
55 * @tc.desc: Test EnterpriseConnManager::CreateAdminConnection func.
56 * @tc.type: FUNC
57 */
58 HWTEST_F(EnterpriseConnManagerTest, TestAdminConnectAbility, TestSize.Level1)
59 {
60 std::string bundleName{"com.edm.test.demo"};
61 std::string abilityName{"com.edm.test.demo.Ability"};
62 AAFwk::Want connectWant;
63 connectWant.SetElementName(bundleName, abilityName);
64 std::shared_ptr<EnterpriseConnManager> manager = DelayedSingleton<EnterpriseConnManager>::GetInstance();
65 sptr<IEnterpriseConnection> connection = manager->CreateAdminConnection(connectWant,
66 IEnterpriseAdmin::COMMAND_ON_ADMIN_ENABLED, DEFAULT_USERID);
67 bool ret = manager->ConnectAbility(connection);
68 EXPECT_TRUE(!ret);
69 }
70
71 /**
72 * @tc.name: TestBundleConnectAbility
73 * @tc.desc: Test EnterpriseConnManager::CreateBundleConnection func.
74 * @tc.type: FUNC
75 */
76 HWTEST_F(EnterpriseConnManagerTest, TestBundleConnectAbility, TestSize.Level1)
77 {
78 std::string bundleName{"com.edm.test.demo"};
79 std::string abilityName{"com.edm.test.demo.Ability"};
80 AAFwk::Want connectWant;
81 connectWant.SetElementName(bundleName, abilityName);
82 std::shared_ptr<EnterpriseConnManager> manager = DelayedSingleton<EnterpriseConnManager>::GetInstance();
83 sptr<IEnterpriseConnection> connection = manager->CreateBundleConnection(connectWant,
84 static_cast<uint32_t>(ManagedEvent::BUNDLE_ADDED), DEFAULT_USERID, "com.edm.test.bundle", DEFAULT_USERID);
85 bool ret = manager->ConnectAbility(connection);
86 EXPECT_TRUE(!ret);
87
88 ret = manager->ConnectAbility(nullptr);
89 EXPECT_TRUE(!ret);
90 }
91
92 /**
93 * @tc.name: TestUpdateConnectAbility
94 * @tc.desc: Test EnterpriseConnManager::CreateUpdateConnection func.
95 * @tc.type: FUNC
96 */
97 HWTEST_F(EnterpriseConnManagerTest, TestUpdateConnectAbility, TestSize.Level1)
98 {
99 std::string bundleName{"com.edm.test.demo"};
100 std::string abilityName{"com.edm.test.demo.Ability"};
101 AAFwk::Want connectWant;
102 connectWant.SetElementName(bundleName, abilityName);
103 UpdateInfo updateInfo;
104 std::shared_ptr<EnterpriseConnManager> manager = DelayedSingleton<EnterpriseConnManager>::GetInstance();
105 sptr<IEnterpriseConnection> connection = manager->CreateUpdateConnection(connectWant,
106 DEFAULT_USERID, updateInfo);
107 bool ret = manager->ConnectAbility(connection);
108 EXPECT_TRUE(!ret);
109 }
110 } // namespace TEST
111 } // namespace EDM
112 } // namespace OHOS