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