1 /*
2  * Copyright (c) 2022 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 <gtest/gtest.h>
17 #include "ptable_manager.h"
18 #include "ptable.h"
19 
20 using namespace Updater;
21 using namespace testing;
22 using namespace testing::ext;
23 
24 namespace {
25 class PtableTest : public Ptable {
26 public:
PtableTest()27     PtableTest() {}
28 
~PtableTest()29     ~PtableTest() {}
30 
ParsePartitionFromBuffer(uint8_t * ptbImgBuffer,const uint32_t imgBufSize)31     bool ParsePartitionFromBuffer(uint8_t *ptbImgBuffer, const uint32_t imgBufSize) override
32     {
33         return true;
34     }
35 
LoadPtableFromDevice()36     bool LoadPtableFromDevice() override
37     {
38         return true;
39     }
40 
WritePartitionTable()41     bool WritePartitionTable() override
42     {
43         return true;
44     }
45 
TestGetPartionInfoByName(const std::string & partitionName,PtnInfo & ptnInfo,int32_t & index)46     bool TestGetPartionInfoByName(const std::string &partitionName, PtnInfo &ptnInfo, int32_t &index)
47     {
48         return GetPartionInfoByName(partitionName, ptnInfo, index);
49     }
50 
TestCalculateCrc32(const uint8_t * buffer,const uint32_t len)51     int TestCalculateCrc32(const uint8_t *buffer, const uint32_t len)
52     {
53         return static_cast<int>(CalculateCrc32(buffer, len));
54     }
55 
ChangePartitionInfo(std::vector<PtnInfo> partitionInfo)56     void ChangePartitionInfo(std::vector<PtnInfo> partitionInfo)
57     {
58         partitionInfo_ = partitionInfo;
59     }
60 };
61 
62 class UTestPtable : public ::testing::Test {
63 public:
UTestPtable()64     UTestPtable() {}
65 
66     ~UTestPtable() = default;
67 
TestGetPartionInfoByName()68     void TestGetPartionInfoByName()
69     {
70         PtableTest ptableTest {};
71         std::string partionName = "";
72         int32_t index = 0;
73         PtableTest::PtnInfo ptnInfo;
74         bool ret = ptableTest.TestGetPartionInfoByName(partionName, ptnInfo, index);
75         ASSERT_EQ(ret, false);
76         std::vector<PtableTest::PtnInfo> partionInfo;
77         ptnInfo.dispName = "updater";
78         partionInfo.push_back(ptnInfo);
79         ptableTest.ChangePartitionInfo(partionInfo);
80         ret = ptableTest.TestGetPartionInfoByName(partionName, ptnInfo, index);
81         ASSERT_EQ(ret, false);
82         partionName = "updater";
83         ret = ptableTest.TestGetPartionInfoByName(partionName, ptnInfo, index);
84         ASSERT_NE(ret, false);
85     }
86 
TestCalculateCrc32()87     void TestCalculateCrc32()
88     {
89         PtableTest ptableTest {};
90         uint8_t buffer[128] = {1}; // 128 : set buffer size
91         uint8_t *nullBuffer = nullptr;
92         uint32_t len = 0;
93         int ret = ptableTest.TestCalculateCrc32(buffer, len);
94         EXPECT_EQ(ret, 0);
95         ret = ptableTest.TestCalculateCrc32(nullBuffer, len);
96         EXPECT_EQ(ret, 0);
97         len = 1;
98         ret = ptableTest.TestCalculateCrc32(nullBuffer, len);
99         EXPECT_EQ(ret, 0);
100         ret = ptableTest.TestCalculateCrc32(buffer, len);
101         EXPECT_EQ(ret, -1526341861); // -1526341861 : CalculateCrc32's result;
102     }
103 protected:
SetUp()104     void SetUp() {}
TearDown()105     void TearDown() {}
TestBody()106     void TestBody() {}
107 };
108 
109 HWTEST_F(UTestPtable, TestGetPartionInfoByName, TestSize.Level1)
110 {
111     UTestPtable {}.TestGetPartionInfoByName();
112 }
113 
114 HWTEST_F(UTestPtable, TestCalculateCrc32, TestSize.Level1)
115 {
116     UTestPtable {}.TestCalculateCrc32();
117 }
118 }