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 #ifndef EDM_UNIT_TEST_PLUGIN_MANAGER_TEST_H 17 #define EDM_UNIT_TEST_PLUGIN_MANAGER_TEST_H 18 19 #include <gtest/gtest.h> 20 #include <iremote_object.h> 21 #include "edm_log.h" 22 #include "iplugin.h" 23 24 namespace OHOS { 25 namespace EDM { 26 namespace TEST { 27 class TestPlugin : public IPlugin { 28 public: TestPlugin()29 TestPlugin() 30 { 31 policyCode_ = 0; 32 policyName_ = "TestPlugin"; 33 permissionConfig_.typePermissions[IPlugin::PermissionType::NORMAL_DEVICE_ADMIN] = 34 "ohos.permission.EDM_TEST_PERMISSION"; 35 permissionConfig_.apiType = IPlugin::ApiType::PUBLIC; 36 EDMLOGD("TestPlugin constructor"); 37 } 38 OnHandlePolicy(std::uint32_t funcCode,MessageParcel & data,MessageParcel & reply,HandlePolicyData & policyData,int32_t userId)39 ErrCode OnHandlePolicy(std::uint32_t funcCode, MessageParcel &data, MessageParcel &reply, 40 HandlePolicyData &policyData, int32_t userId) override 41 { 42 return ERR_OK; 43 } 44 MergePolicyData(const std::string & adminName,std::string & policyData)45 ErrCode MergePolicyData(const std::string &adminName, std::string &policyData) override 46 { 47 return IPlugin::MergePolicyData(adminName, policyData); 48 } 49 OnHandlePolicyDone(std::uint32_t funcCode,const std::string & adminName,bool isGlobalChanged,int32_t userId)50 void OnHandlePolicyDone(std::uint32_t funcCode, const std::string &adminName, bool isGlobalChanged, 51 int32_t userId) override {} 52 OnAdminRemove(const std::string & adminName,const std::string & policyData,int32_t userId)53 ErrCode OnAdminRemove(const std::string &adminName, const std::string &policyData, int32_t userId) override 54 { 55 return ERR_OK; 56 } 57 OnAdminRemoveDone(const std::string & adminName,const std::string & policyData,int32_t userId)58 void OnAdminRemoveDone(const std::string &adminName, const std::string &policyData, int32_t userId) override {} 59 WritePolicyToParcel(const std::string & policyData,MessageParcel & reply)60 ErrCode WritePolicyToParcel(const std::string &policyData, MessageParcel &reply) override 61 { 62 return IPlugin::WritePolicyToParcel(policyData, reply); 63 } 64 OnGetPolicy(std::string & policyData,MessageParcel & data,MessageParcel & reply,int32_t userId)65 ErrCode OnGetPolicy(std::string &policyData, MessageParcel &data, MessageParcel &reply, int32_t userId) 66 { 67 return ERR_OK; 68 } 69 70 ~TestPlugin() override = default; 71 }; 72 73 class TestExtensionPlugin : public IPlugin { 74 public: TestExtensionPlugin()75 TestExtensionPlugin() 76 { 77 policyCode_ = 1; 78 policyName_ = "TestExtensionPlugin"; 79 permissionConfig_.typePermissions.clear(); 80 permissionConfig_.typePermissions.emplace(IPlugin::PermissionType::NORMAL_DEVICE_ADMIN, 81 "ohos.permission.EDM_TEST_PERMISSION"); 82 permissionConfig_.apiType = IPlugin::ApiType::PUBLIC; 83 EDMLOGD("TestExtensionPlugin constructor"); 84 } 85 OnHandlePolicy(std::uint32_t funcCode,MessageParcel & data,MessageParcel & reply,HandlePolicyData & policyData,int32_t userId)86 ErrCode OnHandlePolicy(std::uint32_t funcCode, MessageParcel &data, MessageParcel &reply, 87 HandlePolicyData &policyData, int32_t userId) override 88 { 89 return ERR_OK; 90 } 91 MergePolicyData(const std::string & adminName,std::string & policyData)92 ErrCode MergePolicyData(const std::string &adminName, std::string &policyData) override 93 { 94 return IPlugin::MergePolicyData(adminName, policyData); 95 } 96 OnHandlePolicyDone(std::uint32_t funcCode,const std::string & adminName,bool isGlobalChanged,int32_t userId)97 void OnHandlePolicyDone(std::uint32_t funcCode, const std::string &adminName, bool isGlobalChanged, 98 int32_t userId) override {} 99 OnAdminRemove(const std::string & adminName,const std::string & policyData,int32_t userId)100 ErrCode OnAdminRemove(const std::string &adminName, const std::string &policyData, int32_t userId) override 101 { 102 return ERR_OK; 103 } 104 OnAdminRemoveDone(const std::string & adminName,const std::string & policyData,int32_t userId)105 void OnAdminRemoveDone(const std::string &adminName, const std::string &policyData, int32_t userId) override {} 106 WritePolicyToParcel(const std::string & policyData,MessageParcel & reply)107 ErrCode WritePolicyToParcel(const std::string &policyData, MessageParcel &reply) override 108 { 109 return IPlugin::WritePolicyToParcel(policyData, reply); 110 } 111 OnGetPolicy(std::string & policyData,MessageParcel & data,MessageParcel & reply,int32_t userId)112 ErrCode OnGetPolicy(std::string &policyData, MessageParcel &data, MessageParcel &reply, int32_t userId) 113 { 114 return ERR_OK; 115 } 116 117 ~TestExtensionPlugin() override = default; 118 }; 119 120 class PluginManagerTest : public testing::Test { 121 protected: 122 virtual void SetUp() override; 123 124 virtual void TearDown() override; 125 }; 126 } // namespace TEST 127 } // namespace EDM 128 } // namespace OHOS 129 #endif // EDM_UNIT_TEST_PLUGIN_MANAGER_TEST_H 130