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

mock_dynamic_partition_control.h

diff --git a/mock_dynamic_partition_control.h b/mock_dynamic_partition_control.h
index 09b825d..ffabac7 100644
--- a/mock_dynamic_partition_control.h
+++ b/mock_dynamic_partition_control.h
@@ -38,8 +38,9 @@
                     std::string*));
   MOCK_METHOD0(Cleanup, void());
   MOCK_METHOD0(GetDynamicPartitionsFeatureFlag, FeatureFlag());
-  MOCK_METHOD4(PreparePartitionsForUpdate,
-               bool(uint32_t, uint32_t, const DeltaArchiveManifest&, bool));
+  MOCK_METHOD5(
+      PreparePartitionsForUpdate,
+      bool(uint32_t, uint32_t, const DeltaArchiveManifest&, bool, uint64_t*));
   MOCK_METHOD0(GetVirtualAbFeatureFlag, FeatureFlag());
   MOCK_METHOD0(FinishUpdate, bool());
 };