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 17 #include "edm_log.h" 18 #include "enhance_execute_strategy.h" 19 #include "func_code.h" 20 #include "iplugin_template_test.h" 21 #include "plugin_manager.h" 22 #include "plugin_manager_test.h" 23 24 using namespace testing::ext; 25 using namespace testing; 26 27 namespace OHOS { 28 namespace EDM { 29 namespace TEST { 30 class EnhanceExecuteStrategyTest : public testing::Test { 31 public: SetUp()32 void SetUp() override {} 33 TearDown()34 void TearDown() override {} 35 }; 36 37 /** 38 * @tc.name: TestOnGetExecuteWhenFirstIfEstablished 39 * @tc.desc: Test OnGetExecute func When the First If condition is met. 40 * @tc.type: FUNC 41 */ 42 HWTEST_F(EnhanceExecuteStrategyTest, TestOnGetExecuteWhenFirstIfEstablished, TestSize.Level1) 43 { 44 std::string policyData("policyData"); 45 MessageParcel data; 46 MessageParcel reply; 47 int32_t userId = 0; 48 49 std::uint32_t funcCode = 0; 50 std::unique_ptr<EnhanceExecuteStrategy> enhanceExecuteStrategy = std::make_unique<EnhanceExecuteStrategy>(); 51 ErrCode eRet = enhanceExecuteStrategy->OnGetExecute(funcCode, policyData, data, reply, userId); 52 ASSERT_TRUE(eRet == ERR_EDM_GET_POLICY_FAILED); 53 } 54 55 /** 56 * @tc.name: TestOnGetExecuteWhenSecondIfEstablished 57 * @tc.desc: Test OnGetExecute func When the Second If condition is met. 58 * @tc.type: FUNC 59 */ 60 HWTEST_F(EnhanceExecuteStrategyTest, TestOnGetExecuteWhenSecondIfEstablished, TestSize.Level1) 61 { 62 std::string policyData("policyData"); 63 MessageParcel data; 64 MessageParcel reply; 65 int32_t userId = 0; 66 67 std::unique_ptr<EnhanceExecuteStrategy> enhanceExecuteStrategy = std::make_unique<EnhanceExecuteStrategy>(); 68 std::shared_ptr<IPlugin> plugin = std::make_shared<TestPlugin>(); 69 std::uint32_t policyCode = plugin->GetCode(); 70 std::uint32_t funcCode = FuncCodeUtils::CreateFuncCode(static_cast<std::uint32_t>(FuncFlag::POLICY_FLAG), 71 static_cast<std::uint32_t>(FuncOperateType::GET), policyCode); 72 bool bRet = PluginManager::GetInstance()->AddPlugin(plugin); 73 ASSERT_TRUE(bRet); 74 ErrCode eRet = enhanceExecuteStrategy->OnGetExecute(funcCode, policyData, data, reply, userId); 75 ASSERT_TRUE(eRet == ERR_OK); 76 } 77 78 /** 79 * @tc.name: TestOnGetExecuteWhenThirdIfEstablished 80 * @tc.desc: Test OnGetExecute func When the Third If condition is met. 81 * @tc.type: FUNC 82 */ 83 HWTEST_F(EnhanceExecuteStrategyTest, TestOnGetExecuteWhenThirdIfEstablished, TestSize.Level1) 84 { 85 std::string policyData("policyData"); 86 MessageParcel data; 87 MessageParcel reply; 88 int32_t userId = 0; 89 90 std::unique_ptr<EnhanceExecuteStrategy> enhanceExecuteStrategy = std::make_unique<EnhanceExecuteStrategy>(); 91 std::shared_ptr<IPlugin> plugin = std::make_shared<TestPlugin>(); 92 std::uint32_t policyCode = plugin->GetCode(); 93 std::uint32_t funcCode = FuncCodeUtils::CreateFuncCode(static_cast<std::uint32_t>(FuncFlag::POLICY_FLAG), 94 static_cast<std::uint32_t>(FuncOperateType::GET), policyCode); 95 bool bRet = PluginManager::GetInstance()->AddPlugin(plugin); 96 ASSERT_TRUE(bRet); 97 std::shared_ptr<IPlugin> extensionPlugin = std::make_shared<TestExtensionPlugin>(); 98 std::uint32_t extpolicyCode = extensionPlugin->GetCode(); 99 bRet = PluginManager::GetInstance()->AddExtensionPlugin(extensionPlugin, extpolicyCode, ExecuteStrategy::ENHANCE); 100 ASSERT_TRUE(bRet); 101 ErrCode eRet = enhanceExecuteStrategy->OnGetExecute(funcCode, policyData, data, reply, userId); 102 ASSERT_TRUE(eRet == ERR_OK); 103 } 104 105 /** 106 * @tc.name: TestOnSetExecuteWhenFirstIfEstablished 107 * @tc.desc: Test OnSetExecute func When the First If condition is met. 108 * @tc.type: FUNC 109 */ 110 HWTEST_F(EnhanceExecuteStrategyTest, TestOnSetExecuteWhenFirstIfEstablished, TestSize.Level1) 111 { 112 HandlePolicyData policyData; 113 MessageParcel data; 114 MessageParcel reply; 115 int32_t userId = 0; 116 117 std::uint32_t funcCode = 0; 118 std::unique_ptr<EnhanceExecuteStrategy> enhanceExecuteStrategy = std::make_unique<EnhanceExecuteStrategy>(); 119 ErrCode eRet = enhanceExecuteStrategy->OnSetExecute(funcCode, data, reply, policyData, userId); 120 ASSERT_TRUE(eRet == ERR_EDM_HANDLE_POLICY_FAILED); 121 } 122 123 /** 124 * @tc.name: TestOnSetExecuteWhenSecondIfEstablished 125 * @tc.desc: Test OnSetExecute func When the Second If condition is met. 126 * @tc.type: FUNC 127 */ 128 HWTEST_F(EnhanceExecuteStrategyTest, TestOnSetExecuteWhenSecondIfEstablished, TestSize.Level1) 129 { 130 HandlePolicyData policyData; 131 MessageParcel data; 132 MessageParcel reply; 133 int32_t userId = 0; 134 135 std::unique_ptr<EnhanceExecuteStrategy> enhanceExecuteStrategy = std::make_unique<EnhanceExecuteStrategy>(); 136 std::shared_ptr<IPlugin> plugin = std::make_shared<TestPlugin>(); 137 std::uint32_t policyCode = plugin->GetCode(); 138 std::uint32_t funcCode = FuncCodeUtils::CreateFuncCode(static_cast<std::uint32_t>(FuncFlag::POLICY_FLAG), 139 static_cast<std::uint32_t>(FuncOperateType::SET), policyCode); 140 bool bRet = PluginManager::GetInstance()->AddPlugin(plugin); 141 ASSERT_TRUE(bRet); 142 ErrCode eRet = enhanceExecuteStrategy->OnSetExecute(funcCode, data, reply, policyData, userId); 143 ASSERT_TRUE(eRet == ERR_OK); 144 } 145 146 /** 147 * @tc.name: TestOnSetExecuteWhenThirdIfEstablished 148 * @tc.desc: Test OnSetExecute func When the Third If condition is met. 149 * @tc.type: FUNC 150 */ 151 HWTEST_F(EnhanceExecuteStrategyTest, TestOnSetExecuteWhenThirdIfEstablished, TestSize.Level1) 152 { 153 HandlePolicyData policyData; 154 MessageParcel data; 155 MessageParcel reply; 156 int32_t userId = 0; 157 158 std::unique_ptr<EnhanceExecuteStrategy> enhanceExecuteStrategy = std::make_unique<EnhanceExecuteStrategy>(); 159 std::shared_ptr<IPlugin> plugin = std::make_shared<TestPlugin>(); 160 std::uint32_t policyCode = plugin->GetCode(); 161 std::uint32_t funcCode = FuncCodeUtils::CreateFuncCode(static_cast<std::uint32_t>(FuncFlag::POLICY_FLAG), 162 static_cast<std::uint32_t>(FuncOperateType::SET), policyCode); 163 bool bRet = PluginManager::GetInstance()->AddPlugin(plugin); 164 ASSERT_TRUE(bRet); 165 std::shared_ptr<IPlugin> extensionPlugin = std::make_shared<TestExtensionPlugin>(); 166 std::uint32_t extpolicyCode = extensionPlugin->GetCode(); 167 bRet = PluginManager::GetInstance()->AddExtensionPlugin(extensionPlugin, extpolicyCode, ExecuteStrategy::ENHANCE); 168 ASSERT_TRUE(bRet); 169 ErrCode eRet = enhanceExecuteStrategy->OnSetExecute(funcCode, data, reply, policyData, userId); 170 ASSERT_TRUE(eRet == ERR_OK); 171 } 172 } // namespace TEST 173 } // namespace EDM 174 } // namespace OHOS