1 //
2 // Copyright (C) 2015 The Android Open Source Project
3 //
4 // Licensed under the Apache License, Version 2.0 (the "License");
5 // you may not use this file except in compliance with the License.
6 // You may obtain a copy of the License at
7 //
8 // http://www.apache.org/licenses/LICENSE-2.0
9 //
10 // Unless required by applicable law or agreed to in writing, software
11 // distributed under the License is distributed on an "AS IS" BASIS,
12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 // See the License for the specific language governing permissions and
14 // limitations under the License.
15 //
16
17 #include "update_engine/payload_generator/extent_utils.h"
18
19 #include <utility>
20 #include <vector>
21
22 #include <gtest/gtest.h>
23
24 #include "update_engine/common/test_utils.h"
25 #include "update_engine/payload_consumer/payload_constants.h"
26 #include "update_engine/payload_generator/extent_ranges.h"
27
28 using std::vector;
29
30 namespace chromeos_update_engine {
31
32 class ExtentUtilsTest : public ::testing::Test {};
33
TEST(ExtentUtilsTest,AppendSparseToExtentsTest)34 TEST(ExtentUtilsTest, AppendSparseToExtentsTest) {
35 vector<Extent> extents;
36
37 EXPECT_EQ(0U, extents.size());
38 AppendBlockToExtents(&extents, kSparseHole);
39 EXPECT_EQ(1U, extents.size());
40 AppendBlockToExtents(&extents, 0);
41 EXPECT_EQ(2U, extents.size());
42 AppendBlockToExtents(&extents, kSparseHole);
43 AppendBlockToExtents(&extents, kSparseHole);
44
45 ASSERT_EQ(3U, extents.size());
46 EXPECT_EQ(kSparseHole, extents[0].start_block());
47 EXPECT_EQ(1U, extents[0].num_blocks());
48 EXPECT_EQ(0U, extents[1].start_block());
49 EXPECT_EQ(1U, extents[1].num_blocks());
50 EXPECT_EQ(kSparseHole, extents[2].start_block());
51 EXPECT_EQ(2U, extents[2].num_blocks());
52 }
53
TEST(ExtentUtilsTest,BlocksInExtentsTest)54 TEST(ExtentUtilsTest, BlocksInExtentsTest) {
55 {
56 vector<Extent> extents;
57 EXPECT_EQ(0U, utils::BlocksInExtents(extents));
58 extents.push_back(ExtentForRange(0, 1));
59 EXPECT_EQ(1U, utils::BlocksInExtents(extents));
60 extents.push_back(ExtentForRange(23, 55));
61 EXPECT_EQ(56U, utils::BlocksInExtents(extents));
62 extents.push_back(ExtentForRange(1, 2));
63 EXPECT_EQ(58U, utils::BlocksInExtents(extents));
64 }
65 {
66 google::protobuf::RepeatedPtrField<Extent> extents;
67 EXPECT_EQ(0U, utils::BlocksInExtents(extents));
68 *extents.Add() = ExtentForRange(0, 1);
69 EXPECT_EQ(1U, utils::BlocksInExtents(extents));
70 *extents.Add() = ExtentForRange(23, 55);
71 EXPECT_EQ(56U, utils::BlocksInExtents(extents));
72 *extents.Add() = ExtentForRange(1, 2);
73 EXPECT_EQ(58U, utils::BlocksInExtents(extents));
74 }
75 }
76
TEST(ExtentUtilsTest,ExtendExtentsTest)77 TEST(ExtentUtilsTest, ExtendExtentsTest) {
78 InstallOperation first_op;
79 *(first_op.add_src_extents()) = ExtentForRange(1, 1);
80 *(first_op.add_src_extents()) = ExtentForRange(3, 1);
81
82 InstallOperation second_op;
83 *(second_op.add_src_extents()) = ExtentForRange(4, 2);
84 *(second_op.add_src_extents()) = ExtentForRange(8, 2);
85
86 ExtendExtents(first_op.mutable_src_extents(), second_op.src_extents());
87 vector<Extent> first_op_vec;
88 ExtentsToVector(first_op.src_extents(), &first_op_vec);
89 EXPECT_EQ(
90 (vector<Extent>{
91 ExtentForRange(1, 1), ExtentForRange(3, 3), ExtentForRange(8, 2)}),
92 first_op_vec);
93 }
94
TEST(ExtentUtilsTest,NormalizeExtentsSimpleList)95 TEST(ExtentUtilsTest, NormalizeExtentsSimpleList) {
96 // Make sure it works when there's just one extent.
97 vector<Extent> extents;
98 NormalizeExtents(&extents);
99 EXPECT_EQ(0U, extents.size());
100
101 extents = {ExtentForRange(0, 3)};
102 NormalizeExtents(&extents);
103 EXPECT_EQ(1U, extents.size());
104 EXPECT_EQ(ExtentForRange(0, 3), extents[0]);
105 }
106
TEST(ExtentUtilsTest,NormalizeExtentsTest)107 TEST(ExtentUtilsTest, NormalizeExtentsTest) {
108 vector<Extent> extents = {ExtentForRange(0, 3),
109 ExtentForRange(3, 2),
110 ExtentForRange(5, 1),
111 ExtentForRange(8, 4),
112 ExtentForRange(13, 1),
113 ExtentForRange(14, 2)};
114 NormalizeExtents(&extents);
115 EXPECT_EQ(3U, extents.size());
116 EXPECT_EQ(ExtentForRange(0, 6), extents[0]);
117 EXPECT_EQ(ExtentForRange(8, 4), extents[1]);
118 EXPECT_EQ(ExtentForRange(13, 3), extents[2]);
119 }
120
TEST(ExtentUtilsTest,ExtentsSublistTest)121 TEST(ExtentUtilsTest, ExtentsSublistTest) {
122 vector<Extent> extents = {
123 ExtentForRange(10, 10), ExtentForRange(30, 10), ExtentForRange(50, 10)};
124
125 // Simple empty result cases.
126 EXPECT_EQ(vector<Extent>(), ExtentsSublist(extents, 1000, 20));
127 EXPECT_EQ(vector<Extent>(), ExtentsSublist(extents, 5, 0));
128 EXPECT_EQ(vector<Extent>(), ExtentsSublist(extents, 30, 1));
129
130 // Normal test cases.
131 EXPECT_EQ(vector<Extent>{ExtentForRange(13, 2)},
132 ExtentsSublist(extents, 3, 2));
133 EXPECT_EQ(vector<Extent>{ExtentForRange(15, 5)},
134 ExtentsSublist(extents, 5, 5));
135 EXPECT_EQ((vector<Extent>{ExtentForRange(15, 5), ExtentForRange(30, 5)}),
136 ExtentsSublist(extents, 5, 10));
137 EXPECT_EQ((vector<Extent>{
138 ExtentForRange(13, 7),
139 ExtentForRange(30, 10),
140 ExtentForRange(50, 3),
141 }),
142 ExtentsSublist(extents, 3, 20));
143
144 // Extact match case.
145 EXPECT_EQ(vector<Extent>{ExtentForRange(30, 10)},
146 ExtentsSublist(extents, 10, 10));
147 EXPECT_EQ(vector<Extent>{ExtentForRange(50, 10)},
148 ExtentsSublist(extents, 20, 10));
149
150 // Cases where the requested num_blocks is too big.
151 EXPECT_EQ(vector<Extent>{ExtentForRange(53, 7)},
152 ExtentsSublist(extents, 23, 100));
153 EXPECT_EQ((vector<Extent>{ExtentForRange(34, 6), ExtentForRange(50, 10)}),
154 ExtentsSublist(extents, 14, 100));
155 }
156
157 } // namespace chromeos_update_engine
158