commit d692ecc12d51b77f1b0d4db5d8fd4d5b81be8355
author Kelvin Zhang <zhangkelvin@google.com> Sat Jul 31 00:08:32 2021 +0000
committer Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com> Sat Jul 31 00:08:32 2021 +0000
tree 0d077a8ef6e94cc3e3e70c0248a3774ee686a178
parent 4dd9cc7cdd2686e202588ad522ab0e45d256e378
parent ca9704edc3018b6029d3b9e5aa81bee033d02911

Add GetNthBlockTest am: ca9704edc3

Original change: https://android-review.googlesource.com/c/platform/system/update_engine/+/1769945

Change-Id: I4028c6b96794e3275ada8fca36ad966601fa4923

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