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, BlocksInExtents(extents));
58 extents.push_back(ExtentForRange(0, 1));
59 EXPECT_EQ(1U, BlocksInExtents(extents));
60 extents.push_back(ExtentForRange(23, 55));
61 EXPECT_EQ(56U, BlocksInExtents(extents));
62 extents.push_back(ExtentForRange(1, 2));
63 EXPECT_EQ(58U, BlocksInExtents(extents));
64 }
65 {
66 google::protobuf::RepeatedPtrField<Extent> extents;
67 EXPECT_EQ(0U, BlocksInExtents(extents));
68 *extents.Add() = ExtentForRange(0, 1);
69 EXPECT_EQ(1U, BlocksInExtents(extents));
70 *extents.Add() = ExtentForRange(23, 55);
71 EXPECT_EQ(56U, BlocksInExtents(extents));
72 *extents.Add() = ExtentForRange(1, 2);
73 EXPECT_EQ(58U, 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((vector<Extent>{
90 ExtentForRange(1, 1),
91 ExtentForRange(3, 3),
92 ExtentForRange(8, 2)}), 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 = {
109 ExtentForRange(0, 3),
110 ExtentForRange(3, 2),
111 ExtentForRange(5, 1),
112 ExtentForRange(8, 4),
113 ExtentForRange(13, 1),
114 ExtentForRange(14, 2)
115 };
116 NormalizeExtents(&extents);
117 EXPECT_EQ(3U, extents.size());
118 EXPECT_EQ(ExtentForRange(0, 6), extents[0]);
119 EXPECT_EQ(ExtentForRange(8, 4), extents[1]);
120 EXPECT_EQ(ExtentForRange(13, 3), extents[2]);
121 }
122
TEST(ExtentUtilsTest,ExtentsSublistTest)123 TEST(ExtentUtilsTest, ExtentsSublistTest) {
124 vector<Extent> extents = {
125 ExtentForRange(10, 10),
126 ExtentForRange(30, 10),
127 ExtentForRange(50, 10)
128 };
129
130 // Simple empty result cases.
131 EXPECT_EQ(vector<Extent>(),
132 ExtentsSublist(extents, 1000, 20));
133 EXPECT_EQ(vector<Extent>(),
134 ExtentsSublist(extents, 5, 0));
135 EXPECT_EQ(vector<Extent>(),
136 ExtentsSublist(extents, 30, 1));
137
138 // Normal test cases.
139 EXPECT_EQ(vector<Extent>{ ExtentForRange(13, 2) },
140 ExtentsSublist(extents, 3, 2));
141 EXPECT_EQ(vector<Extent>{ ExtentForRange(15, 5) },
142 ExtentsSublist(extents, 5, 5));
143 EXPECT_EQ((vector<Extent>{ ExtentForRange(15, 5), ExtentForRange(30, 5) }),
144 ExtentsSublist(extents, 5, 10));
145 EXPECT_EQ((vector<Extent>{
146 ExtentForRange(13, 7),
147 ExtentForRange(30, 10),
148 ExtentForRange(50, 3), }),
149 ExtentsSublist(extents, 3, 20));
150
151 // Extact match case.
152 EXPECT_EQ(vector<Extent>{ ExtentForRange(30, 10) },
153 ExtentsSublist(extents, 10, 10));
154 EXPECT_EQ(vector<Extent>{ ExtentForRange(50, 10) },
155 ExtentsSublist(extents, 20, 10));
156
157 // Cases where the requested num_blocks is too big.
158 EXPECT_EQ(vector<Extent>{ ExtentForRange(53, 7) },
159 ExtentsSublist(extents, 23, 100));
160 EXPECT_EQ((vector<Extent>{ ExtentForRange(34, 6), ExtentForRange(50, 10) }),
161 ExtentsSublist(extents, 14, 100));
162 }
163
164 } // namespace chromeos_update_engine
165