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 "func_code.h" 19 #include "iplugin_template_test.h" 20 #include "plugin_manager.h" 21 #include "plugin_manager_test.h" 22 #include "replace_execute_strategy.h" 23 24 using namespace testing::ext; 25 using namespace testing; 26 27 namespace OHOS { 28 namespace EDM { 29 namespace TEST { 30 class ReplaceExecuteStrategyTest : 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(ReplaceExecuteStrategyTest, 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<ReplaceExecuteStrategy> replaceExecuteStrategy = std::make_unique<ReplaceExecuteStrategy>(); 51 ErrCode eRet = replaceExecuteStrategy->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(ReplaceExecuteStrategyTest, 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<ReplaceExecuteStrategy> replaceExecuteStrategy = std::make_unique<ReplaceExecuteStrategy>(); 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 std::shared_ptr<IPlugin> extensionPlugin = std::make_shared<TestExtensionPlugin>(); 75 std::uint32_t extpolicyCode = extensionPlugin->GetCode(); 76 bRet = PluginManager::GetInstance()->AddExtensionPlugin(extensionPlugin, extpolicyCode, ExecuteStrategy::REPLACE); 77 ASSERT_TRUE(bRet); 78 ErrCode eRet = replaceExecuteStrategy->OnGetExecute(funcCode, policyData, data, reply, userId); 79 ASSERT_TRUE(eRet == ERR_OK); 80 } 81 82 /** 83 * @tc.name: TestOnSetExecuteWhenFirstIfEstablished 84 * @tc.desc: Test OnSetExecute func When the First If condition is met. 85 * @tc.type: FUNC 86 */ 87 HWTEST_F(ReplaceExecuteStrategyTest, TestOnSetExecuteWhenFirstIfEstablished, TestSize.Level1) 88 { 89 HandlePolicyData policyData; 90 MessageParcel data; 91 MessageParcel reply; 92 int32_t userId = 0; 93 94 std::uint32_t funcCode = 0; 95 std::unique_ptr<ReplaceExecuteStrategy> replaceExecuteStrategy = std::make_unique<ReplaceExecuteStrategy>(); 96 ErrCode eRet = replaceExecuteStrategy->OnSetExecute(funcCode, data, reply, policyData, userId); 97 ASSERT_TRUE(eRet == ERR_EDM_HANDLE_POLICY_FAILED); 98 } 99 100 /** 101 * @tc.name: TestOnSetExecuteWhenSecondIfEstablished 102 * @tc.desc: Test OnSetExecute func When the Second If condition is met. 103 * @tc.type: FUNC 104 */ 105 HWTEST_F(ReplaceExecuteStrategyTest, TestOnSetExecuteWhenSecondIfEstablished, TestSize.Level1) 106 { 107 HandlePolicyData policyData; 108 MessageParcel data; 109 MessageParcel reply; 110 int32_t userId = 0; 111 112 std::unique_ptr<ReplaceExecuteStrategy> replaceExecuteStrategy = std::make_unique<ReplaceExecuteStrategy>(); 113 std::shared_ptr<IPlugin> plugin = std::make_shared<TestPlugin>(); 114 std::uint32_t policyCode = plugin->GetCode(); 115 std::uint32_t funcCode = FuncCodeUtils::CreateFuncCode(static_cast<std::uint32_t>(FuncFlag::POLICY_FLAG), 116 static_cast<std::uint32_t>(FuncOperateType::SET), policyCode); 117 bool bRet = PluginManager::GetInstance()->AddPlugin(plugin); 118 ASSERT_TRUE(bRet); 119 std::shared_ptr<IPlugin> extensionPlugin = std::make_shared<TestExtensionPlugin>(); 120 std::uint32_t extpolicyCode = extensionPlugin->GetCode(); 121 bRet = PluginManager::GetInstance()->AddExtensionPlugin(extensionPlugin, extpolicyCode, ExecuteStrategy::REPLACE); 122 ASSERT_TRUE(bRet); 123 ErrCode eRet = replaceExecuteStrategy->OnSetExecute(funcCode, data, reply, policyData, userId); 124 ASSERT_TRUE(eRet == ERR_OK); 125 } 126 } // namespace TEST 127 } // namespace EDM 128 } // namespace OHOS