commit 0c0b7c0e810eaf42e958d6d7cd8b3d39b7e834bc
author Kelvin Zhang <zhangkelvin@google.com> Sat Jul 31 00:17:05 2021 +0000
committer Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com> Sat Jul 31 00:17:05 2021 +0000
tree 28049392489b45dd6c9a4114fbb22eda84c26ab3
parent aa59dc569f5ef50acccd01aed031c54754c957d6
parent d692ecc12d51b77f1b0d4db5d8fd4d5b81be8355

Add GetNthBlockTest am: ca9704edc3 am: d692ecc12d

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

Change-Id: Ic4970805a95c920e8e2345eeca6ced704e07e0a5

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