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 "single_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 SingleExecuteStrategyTest : 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(SingleExecuteStrategyTest, 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<SingleExecuteStrategy> singleExecuteStrategy = std::make_unique<SingleExecuteStrategy>(); 51 ErrCode eRet = singleExecuteStrategy->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(SingleExecuteStrategyTest, 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<SingleExecuteStrategy> singleExecuteStrategy = std::make_unique<SingleExecuteStrategy>(); 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 = singleExecuteStrategy->OnGetExecute(funcCode, policyData, data, reply, userId); 75 ASSERT_TRUE(eRet == ERR_OK); 76 } 77 78 /** 79 * @tc.name: TestOnSetExecuteWhenFirstIfEstablished 80 * @tc.desc: Test OnSetExecute func When the First If condition is met. 81 * @tc.type: FUNC 82 */ 83 HWTEST_F(SingleExecuteStrategyTest, TestOnSetExecuteWhenFirstIfEstablished, TestSize.Level1) 84 { 85 HandlePolicyData policyData; 86 MessageParcel data; 87 MessageParcel reply; 88 int32_t userId = 0; 89 90 std::uint32_t funcCode = 0; 91 std::unique_ptr<SingleExecuteStrategy> singleExecuteStrategy = std::make_unique<SingleExecuteStrategy>(); 92 ErrCode eRet = singleExecuteStrategy->OnSetExecute(funcCode, data, reply, policyData, userId); 93 ASSERT_TRUE(eRet == ERR_EDM_HANDLE_POLICY_FAILED); 94 } 95 96 /** 97 * @tc.name: TestOnSetExecuteWhenSecondIfEstablished 98 * @tc.desc: Test OnSetExecute func When the Second If condition is met. 99 * @tc.type: FUNC 100 */ 101 HWTEST_F(SingleExecuteStrategyTest, TestOnSetExecuteWhenSecondIfEstablished, TestSize.Level1) 102 { 103 HandlePolicyData policyData; 104 MessageParcel data; 105 MessageParcel reply; 106 int32_t userId = 0; 107 108 std::unique_ptr<SingleExecuteStrategy> singleExecuteStrategy = std::make_unique<SingleExecuteStrategy>(); 109 std::shared_ptr<IPlugin> plugin = std::make_shared<TestPlugin>(); 110 std::uint32_t policyCode = plugin->GetCode(); 111 std::uint32_t funcCode = FuncCodeUtils::CreateFuncCode(static_cast<std::uint32_t>(FuncFlag::POLICY_FLAG), 112 static_cast<std::uint32_t>(FuncOperateType::SET), policyCode); 113 bool bRet = PluginManager::GetInstance()->AddPlugin(plugin); 114 ASSERT_TRUE(bRet); 115 ErrCode eRet = singleExecuteStrategy->OnSetExecute(funcCode, data, reply, policyData, userId); 116 ASSERT_TRUE(eRet == ERR_OK); 117 } 118 } // namespace TEST 119 } // namespace EDM 120 } // namespace OHOS