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/payload_consumer/payload_constants.h"
25 #include "update_engine/payload_generator/extent_ranges.h"
26
27 using std::vector;
28
29 namespace chromeos_update_engine {
30
31 class ExtentUtilsTest : public ::testing::Test {};
32
TEST(ExtentUtilsTest,AppendSparseToExtentsTest)33 TEST(ExtentUtilsTest, AppendSparseToExtentsTest) {
34 vector<Extent> extents;
35
36 EXPECT_EQ(0U, extents.size());
37 AppendBlockToExtents(&extents, kSparseHole);
38 EXPECT_EQ(1U, extents.size());
39 AppendBlockToExtents(&extents, 0);
40 EXPECT_EQ(2U, extents.size());
41 AppendBlockToExtents(&extents, kSparseHole);
42 AppendBlockToExtents(&extents, kSparseHole);
43
44 ASSERT_EQ(3U, extents.size());
45 EXPECT_EQ(kSparseHole, extents[0].start_block());
46 EXPECT_EQ(1U, extents[0].num_blocks());
47 EXPECT_EQ(0U, extents[1].start_block());
48 EXPECT_EQ(1U, extents[1].num_blocks());
49 EXPECT_EQ(kSparseHole, extents[2].start_block());
50 EXPECT_EQ(2U, extents[2].num_blocks());
51 }
52
TEST(ExtentUtilsTest,BlocksInExtentsTest)53 TEST(ExtentUtilsTest, BlocksInExtentsTest) {
54 {
55 vector<Extent> extents;
56 EXPECT_EQ(0U, utils::BlocksInExtents(extents));
57 extents.push_back(ExtentForRange(0, 1));
58 EXPECT_EQ(1U, utils::BlocksInExtents(extents));
59 extents.push_back(ExtentForRange(23, 55));
60 EXPECT_EQ(56U, utils::BlocksInExtents(extents));
61 extents.push_back(ExtentForRange(1, 2));
62 EXPECT_EQ(58U, utils::BlocksInExtents(extents));
63 }
64 {
65 google::protobuf::RepeatedPtrField<Extent> extents;
66 EXPECT_EQ(0U, utils::BlocksInExtents(extents));
67 *extents.Add() = ExtentForRange(0, 1);
68 EXPECT_EQ(1U, utils::BlocksInExtents(extents));
69 *extents.Add() = ExtentForRange(23, 55);
70 EXPECT_EQ(56U, utils::BlocksInExtents(extents));
71 *extents.Add() = ExtentForRange(1, 2);
72 EXPECT_EQ(58U, utils::BlocksInExtents(extents));
73 }
74 }
75
TEST(ExtentUtilsTest,ExtendExtentsTest)76 TEST(ExtentUtilsTest, ExtendExtentsTest) {
77 InstallOperation first_op;
78 *(first_op.add_src_extents()) = ExtentForRange(1, 1);
79 *(first_op.add_src_extents()) = ExtentForRange(3, 1);
80
81 InstallOperation second_op;
82 *(second_op.add_src_extents()) = ExtentForRange(4, 2);
83 *(second_op.add_src_extents()) = ExtentForRange(8, 2);
84
85 ExtendExtents(first_op.mutable_src_extents(), second_op.src_extents());
86 vector<Extent> first_op_vec;
87 ExtentsToVector(first_op.src_extents(), &first_op_vec);
88 EXPECT_EQ(
89 (vector<Extent>{
90 ExtentForRange(1, 1), ExtentForRange(3, 3), ExtentForRange(8, 2)}),
91 first_op_vec);
92 }
93
TEST(ExtentUtilsTest,NormalizeExtentsSimpleList)94 TEST(ExtentUtilsTest, NormalizeExtentsSimpleList) {
95 // Make sure it works when there's just one extent.
96 vector<Extent> extents;
97 NormalizeExtents(&extents);
98 EXPECT_EQ(0U, extents.size());
99
100 extents = {ExtentForRange(0, 3)};
101 NormalizeExtents(&extents);
102 EXPECT_EQ(1U, extents.size());
103 EXPECT_EQ(ExtentForRange(0, 3), extents[0]);
104 }
105
TEST(ExtentUtilsTest,NormalizeExtentsTest)106 TEST(ExtentUtilsTest, NormalizeExtentsTest) {
107 vector<Extent> extents = {ExtentForRange(0, 3),
108 ExtentForRange(3, 2),
109 ExtentForRange(5, 1),
110 ExtentForRange(8, 4),
111 ExtentForRange(13, 1),
112 ExtentForRange(14, 2)};
113 NormalizeExtents(&extents);
114 EXPECT_EQ(3U, extents.size());
115 EXPECT_EQ(ExtentForRange(0, 6), extents[0]);
116 EXPECT_EQ(ExtentForRange(8, 4), extents[1]);
117 EXPECT_EQ(ExtentForRange(13, 3), extents[2]);
118 }
119
TEST(ExtentUtilsTest,ExtentsSublistTest)120 TEST(ExtentUtilsTest, ExtentsSublistTest) {
121 vector<Extent> extents = {
122 ExtentForRange(10, 10), ExtentForRange(30, 10), ExtentForRange(50, 10)};
123
124 // Simple empty result cases.
125 EXPECT_EQ(vector<Extent>(), ExtentsSublist(extents, 1000, 20));
126 EXPECT_EQ(vector<Extent>(), ExtentsSublist(extents, 5, 0));
127 EXPECT_EQ(vector<Extent>(), ExtentsSublist(extents, 30, 1));
128
129 // Normal test cases.
130 EXPECT_EQ(vector<Extent>{ExtentForRange(13, 2)},
131 ExtentsSublist(extents, 3, 2));
132 EXPECT_EQ(vector<Extent>{ExtentForRange(15, 5)},
133 ExtentsSublist(extents, 5, 5));
134 EXPECT_EQ((vector<Extent>{ExtentForRange(15, 5), ExtentForRange(30, 5)}),
135 ExtentsSublist(extents, 5, 10));
136 EXPECT_EQ((vector<Extent>{
137 ExtentForRange(13, 7),
138 ExtentForRange(30, 10),
139 ExtentForRange(50, 3),
140 }),
141 ExtentsSublist(extents, 3, 20));
142
143 // Extact match case.
144 EXPECT_EQ(vector<Extent>{ExtentForRange(30, 10)},
145 ExtentsSublist(extents, 10, 10));
146 EXPECT_EQ(vector<Extent>{ExtentForRange(50, 10)},
147 ExtentsSublist(extents, 20, 10));
148
149 // Cases where the requested num_blocks is too big.
150 EXPECT_EQ(vector<Extent>{ExtentForRange(53, 7)},
151 ExtentsSublist(extents, 23, 100));
152 EXPECT_EQ((vector<Extent>{ExtentForRange(34, 6), ExtentForRange(50, 10)}),
153 ExtentsSublist(extents, 14, 100));
154 }
155
156 } // namespace chromeos_update_engine
157