1 /* 2 * Copyright (c) 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 #include <fcntl.h> 17 #include <gtest/gtest.h> 18 #include <memory> 19 #include <sys/ioctl.h> 20 #include "log/log.h" 21 #include "securec.h" 22 #include "updater/hardware_fault_retry.h" 23 #include "updater/updater.h" 24 #include "updater/updater_const.h" 25 #include "utils.h" 26 27 using namespace Updater; 28 using namespace std; 29 using namespace testing::ext; 30 31 namespace { 32 class HardwareFaultRetryUnitTest : public testing::Test { 33 public: HardwareFaultRetryUnitTest()34 HardwareFaultRetryUnitTest() 35 { 36 InitUpdaterLogger("UPDATER", TMP_LOG, TMP_STAGE_LOG, TMP_ERROR_CODE_PATH); 37 } ~HardwareFaultRetryUnitTest()38 ~HardwareFaultRetryUnitTest() {} 39 SetUpTestCase(void)40 static void SetUpTestCase(void) {} TearDownTestCase(void)41 static void TearDownTestCase(void) {} SetUp()42 void SetUp() {} TearDown()43 void TearDown() 44 { 45 (void)ClearMisc(); 46 } TestBody()47 void TestBody() {} 48 }; 49 50 HWTEST_F(HardwareFaultRetryUnitTest, VerifyFailRetry, TestSize.Level1) 51 { 52 HardwareFaultRetry::GetInstance().SetFaultInfo(VERIFY_FAILED_REBOOT); 53 HardwareFaultRetry::GetInstance().SetRetryCount(0); 54 HardwareFaultRetry::GetInstance().DoRetryAction(); 55 56 bool ret = Utils::CheckFaultInfo(VERIFY_FAILED_REBOOT); 57 EXPECT_EQ(ret, true); 58 } 59 60 HWTEST_F(HardwareFaultRetryUnitTest, IOFailRetry, TestSize.Level1) 61 { 62 HardwareFaultRetry::GetInstance().SetFaultInfo(IO_FAILED_REBOOT); 63 HardwareFaultRetry::GetInstance().SetRetryCount(0); 64 HardwareFaultRetry::GetInstance().DoRetryAction(); 65 66 bool ret = Utils::CheckFaultInfo(IO_FAILED_REBOOT); 67 EXPECT_EQ(ret, true); 68 } 69 70 HWTEST_F(HardwareFaultRetryUnitTest, RetryMoreThanMax, TestSize.Level1) 71 { 72 HardwareFaultRetry::GetInstance().SetFaultInfo(IO_FAILED_REBOOT); 73 HardwareFaultRetry::GetInstance().SetRetryCount(MAX_RETRY_COUNT); 74 HardwareFaultRetry::GetInstance().DoRetryAction(); 75 76 bool ret = Utils::CheckFaultInfo(IO_FAILED_REBOOT); 77 EXPECT_EQ(ret, false); 78 } 79 80 HWTEST_F(HardwareFaultRetryUnitTest, SetEffectiveValueTest, TestSize.Level1) 81 { 82 HardwareFaultRetry::GetInstance().SetEffectiveValue(false); 83 HardwareFaultRetry::GetInstance().SetFaultInfo(VERIFY_FAILED_REBOOT); 84 HardwareFaultRetry::GetInstance().SetRetryCount(0); 85 HardwareFaultRetry::GetInstance().DoRetryAction(); 86 87 EXPECT_FALSE(Utils::CheckFaultInfo(VERIFY_FAILED_REBOOT)); 88 } 89 90 HWTEST_F(HardwareFaultRetryUnitTest, RegisterAndRemoveTest, TestSize.Level1) 91 { 92 HardwareFaultRetry::GetInstance().SetEffectiveValue(true); 93 HardwareFaultRetry::GetInstance().RegisterDefaultFunc(PROCESS_BIN_FAIL_RETRY); 94 HardwareFaultRetry::GetInstance().SetFaultInfo(PROCESS_BIN_FAIL_RETRY); 95 HardwareFaultRetry::GetInstance().SetRetryCount(0); 96 HardwareFaultRetry::GetInstance().DoRetryAction(); 97 EXPECT_EQ(HardwareFaultRetry::GetInstance().IsRetry(), true); 98 99 EXPECT_EQ(Utils::CheckFaultInfo(PROCESS_BIN_FAIL_RETRY), true); 100 101 (void)ClearMisc(); 102 HardwareFaultRetry::GetInstance().RemoveFunc(PROCESS_BIN_FAIL_RETRY); 103 HardwareFaultRetry::GetInstance().SetFaultInfo(PROCESS_BIN_FAIL_RETRY); 104 HardwareFaultRetry::GetInstance().SetRetryCount(0); 105 HardwareFaultRetry::GetInstance().DoRetryAction(); 106 107 EXPECT_EQ(Utils::CheckFaultInfo(PROCESS_BIN_FAIL_RETRY), false); 108 } 109 }