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 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 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 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 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 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 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
