commit f033ecb6329c9bab43c91b1a2af812d3d2dd7daa
author Yifan Hong <elsk@google.com> Tue Jan 07 18:13:56 2020 -0800
committer Yifan Hong <elsk@google.com> Thu Jan 16 05:05:48 2020 +0000
tree 99f2176378d28852e0dad5c1984453e73e6fcf0c
parent 0b664d1964f88d865643b7e157842bd9ee4eed0e

DynamicPartitionControl: Add required_size to Prepare

Add out parameter required_size to PreparePartitionsForUpdate
to indicate the total size required on /userdata in order
to apply the update.

Bug: 138808058
Test: update_engine_unittests

Change-Id: I2768d13671e212fd24a1a22811b50c9738834459

dynamic_partition_control_android_unittest.cc

diff --git a/dynamic_partition_control_android_unittest.cc b/dynamic_partition_control_android_unittest.cc
index 207a97e..3e8375c 100644
--- a/dynamic_partition_control_android_unittest.cc
+++ b/dynamic_partition_control_android_unittest.cc
@@ -120,7 +120,11 @@
   }
   bool PreparePartitionsForUpdate(const PartitionSizes& partition_sizes) {
     return dynamicControl().PreparePartitionsForUpdate(
-        source(), target(), PartitionSizesToManifest(partition_sizes), true);
+        source(),
+        target(),
+        PartitionSizesToManifest(partition_sizes),
+        true,
+        nullptr);
   }
   void SetSlots(const TestParam& slots) { slots_ = slots; }
 
@@ -317,7 +321,7 @@
   // DynamicPartitionControlAndroidTest::PreparePartitionsForUpdate(), since we
   // don't want any default group in the PartitionMetadata.
   EXPECT_TRUE(dynamicControl().PreparePartitionsForUpdate(
-      source(), target(), {}, true));
+      source(), target(), {}, true, nullptr));
 
   // Should use dynamic source partitions.
   EXPECT_CALL(dynamicControl(), GetState(S("system")))
@@ -371,7 +375,8 @@
       source(),
       target(),
       PartitionSizesToManifest({{"system", 2_GiB}, {"vendor", 1_GiB}}),
-      false));
+      false,
+      nullptr));
 
   // Dynamic partition "system".
   EXPECT_CALL(dynamicControl(), GetState(S("system")))
@@ -622,7 +627,11 @@
 
 TEST_P(DynamicPartitionControlAndroidTestP, ShouldSkipOperationTest) {
   ASSERT_TRUE(dynamicControl().PreparePartitionsForUpdate(
-      source(), target(), PartitionSizesToManifest({{"foo", 4_MiB}}), false));
+      source(),
+      target(),
+      PartitionSizesToManifest({{"foo", 4_MiB}}),
+      false,
+      nullptr));
   dynamicControl().set_fake_mapped_devices({T("foo")});
 
   InstallOperation iop;