1 /*
2 * Copyright (c) 2024 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 #include <gtest/gtest.h>
16 #include "clipboard_policy_plugin.h"
17 #include "clipboard_policy_serializer.h"
18 #include "utils.h"
19
20 using namespace testing::ext;
21
22 namespace OHOS {
23 namespace EDM {
24 namespace TEST {
25 class ClipboardPolicyPluginTest : public testing::Test {
26 protected:
27 static void SetUpTestSuite(void);
28
29 static void TearDownTestSuite(void);
30 };
31 const std::string POLICY_DATA = "[{\"tokenId\":1,\"clipboardPolicy\":1},{\"tokenId\":2,\"clipboardPolicy\":2}]";
SetUpTestSuite(void)32 void ClipboardPolicyPluginTest::SetUpTestSuite(void)
33 {
34 Utils::SetEdmServiceEnable();
35 Utils::SetEdmInitialEnv();
36 }
37
TearDownTestSuite(void)38 void ClipboardPolicyPluginTest::TearDownTestSuite(void)
39 {
40 Utils::SetEdmServiceDisable();
41 Utils::ResetTokenTypeAndUid();
42 ASSERT_TRUE(Utils::IsOriginalUTEnv());
43 std::cout << "now ut process is orignal ut env : " << Utils::IsOriginalUTEnv() << std::endl;
44 }
45
46 /**
47 * @tc.name: TestOnSetPolicyEmpty
48 * @tc.desc: Test ClipboardPolicyPluginTest::OnSetPolicy when data is empty.
49 * @tc.type: FUNC
50 */
51 HWTEST_F(ClipboardPolicyPluginTest, TestOnSetPolicyEmpty, TestSize.Level1)
52 {
53 ClipboardPolicyPlugin plugin;
54 std::map<int32_t, ClipboardPolicy> data;
55 std::map<int32_t, ClipboardPolicy> currentData;
56 ErrCode ret = plugin.OnSetPolicy(data, currentData, 0);
57 ASSERT_TRUE(ret == EdmReturnErrCode::PARAM_ERROR);
58 }
59
60 /**
61 * @tc.name: TestOnSetPolicy
62 * @tc.desc: Test ClipboardPolicyPluginTest::OnSetPolicy.
63 * @tc.type: FUNC
64 */
65 HWTEST_F(ClipboardPolicyPluginTest, TestOnSetPolicy, TestSize.Level1)
66 {
67 ClipboardPolicyPlugin plugin;
68 std::map<int32_t, ClipboardPolicy> data;
69 data.insert(std::make_pair(1, ClipboardPolicy::IN_APP));
70 std::map<int32_t, ClipboardPolicy> currentData;
71 ErrCode ret = plugin.OnSetPolicy(data, currentData, 0);
72 ASSERT_TRUE(ret == ERR_OK);
73 ASSERT_TRUE(currentData[1] == ClipboardPolicy::IN_APP);
74 }
75
76 /**
77 * @tc.name: TestOnSetPolicyFail
78 * @tc.desc: Test ClipboardPolicyPluginTest::OnSetPolicy.
79 * @tc.type: FUNC
80 */
81 HWTEST_F(ClipboardPolicyPluginTest, TestOnSetPolicyFail, TestSize.Level1)
82 {
83 ClipboardPolicyPlugin plugin;
84 std::map<int32_t, ClipboardPolicy> data;
85 data.insert(std::make_pair(101, ClipboardPolicy::IN_APP));
86 std::map<int32_t, ClipboardPolicy> currentData;
87 for (int i = 1; i <= 100; i++) {
88 currentData.insert(std::make_pair(i, ClipboardPolicy::IN_APP));
89 }
90 ErrCode ret = plugin.OnSetPolicy(data, currentData, 0);
91 ASSERT_TRUE(ret == EdmReturnErrCode::PARAM_ERROR);
92 }
93
94 /**
95 * @tc.name: TestOnGetPolicy
96 * @tc.desc: Test ClipboardPolicyPluginTest::OnGetPolicy.
97 * @tc.type: FUNC
98 */
99 HWTEST_F(ClipboardPolicyPluginTest, TestOnGetPolicy, TestSize.Level1)
100 {
101 ClipboardPolicyPlugin plugin;
102 MessageParcel data;
103 MessageParcel reply;
104 std::string policyData = POLICY_DATA;
105 plugin.OnGetPolicy(policyData, data, reply, 0);
106 int32_t ret = reply.ReadInt32();
107 std::string policy = reply.ReadString();
108 std::map<int32_t, ClipboardPolicy> policyMap;
109 auto serializer_ = ClipboardSerializer::GetInstance();
110 serializer_->Deserialize(policy, policyMap);
111 ASSERT_TRUE(ret == ERR_OK);
112 ASSERT_TRUE(policyMap.size() == 2);
113 }
114
115 /**
116 * @tc.name: TestOnGetPolicyWithTokenId
117 * @tc.desc: Test ClipboardPolicyPluginTest::OnGetPolicy.
118 * @tc.type: FUNC
119 */
120 HWTEST_F(ClipboardPolicyPluginTest, TestOnGetPolicyWithTokenId, TestSize.Level1)
121 {
122 ClipboardPolicyPlugin plugin;
123 MessageParcel data;
124 MessageParcel reply;
125 std::string policyData = POLICY_DATA;
126 data.WriteInt32(1);
127 plugin.OnGetPolicy(policyData, data, reply, 0);
128 int32_t ret = reply.ReadInt32();
129 std::string policy = reply.ReadString();
130 std::map<int32_t, ClipboardPolicy> policyMap;
131 auto serializer_ = ClipboardSerializer::GetInstance();
132 serializer_->Deserialize(policy, policyMap);
133 ASSERT_TRUE(ret == ERR_OK);
134 ASSERT_TRUE(policyMap[1] == ClipboardPolicy::IN_APP);
135 }
136
137 } // namespace TEST
138 } // namespace EDM
139 } // namespace OHOS