commit ca9704edc3018b6029d3b9e5aa81bee033d02911
author Kelvin Zhang <zhangkelvin@google.com> Thu Jul 15 16:29:25 2021 -0400
committer Tianjie Xu <xunchang@google.com> Fri Jul 30 22:20:48 2021 +0000
tree 0d077a8ef6e94cc3e3e70c0248a3774ee686a178
parent b170676ab80f30bf6bff76d5427efd8b4fd66971

Add GetNthBlockTest

Test: th
Change-Id: Id7107f9be1fad54cfd4b59d74e72a2f6e45e1a40

payload_consumer/install_operation_executor_unittest.cc

diff --git a/payload_consumer/install_operation_executor_unittest.cc b/payload_consumer/install_operation_executor_unittest.cc
index 2a99782..3c2fb0f 100644
--- a/payload_consumer/install_operation_executor_unittest.cc
+++ b/payload_consumer/install_operation_executor_unittest.cc
@@ -205,4 +205,22 @@
   }
   VerityUntouchedExtents(op);
 }
+
+TEST_F(InstallOperationExecutorTest, GetNthBlockTest) {
+  std::vector<Extent> extents;
+  extents.emplace_back(ExtentForRange(10, 3));
+  extents.emplace_back(ExtentForRange(20, 2));
+  extents.emplace_back(ExtentForRange(30, 1));
+  extents.emplace_back(ExtentForRange(40, 4));
+
+  ASSERT_EQ(GetNthBlock(extents, 0), 10U);
+  ASSERT_EQ(GetNthBlock(extents, 2), 12U);
+  ASSERT_EQ(GetNthBlock(extents, 3), 20U);
+  ASSERT_EQ(GetNthBlock(extents, 4), 21U);
+  ASSERT_EQ(GetNthBlock(extents, 5), 30U);
+  ASSERT_EQ(GetNthBlock(extents, 6), 40U);
+  ASSERT_EQ(GetNthBlock(extents, 7), 41U);
+  ASSERT_EQ(GetNthBlock(extents, 8), 42U);
+}
+
 }  // namespace chromeos_update_engine