commit 302fa70565228bd5aeb6c0aab87b522d60a26008
author Yifan Hong <elsk@google.com> Thu Apr 16 09:48:29 2020 -0700
committer Yifan Hong <elsk@google.com> Mon Apr 20 18:49:59 2020 +0000
tree 07bdc0cfca4a74f633c311559bb17ee995169135
parent 5c5743f6c6e1e3a7f3928265fe46ad73e132f098

Add DynamicPartitionControlAndroid unittest for Virtual A/B

Test: run it
Fixes: 153555889

Change-Id: I8f938987721c09ec263a82f223e4920365d203fc

mock_dynamic_partition_control.h

diff --git a/mock_dynamic_partition_control.h b/mock_dynamic_partition_control.h
index c933e87..1aaebd8 100644
--- a/mock_dynamic_partition_control.h
+++ b/mock_dynamic_partition_control.h
@@ -72,6 +72,11 @@
               (uint32_t, uint32_t),
               (override));
   MOCK_METHOD(std::optional<bool>, IsAvbEnabledOnSystemOther, (), (override));
+  MOCK_METHOD(bool, IsRecovery, (), (override));
+  MOCK_METHOD(bool,
+              PrepareDynamicPartitionsForUpdate,
+              (uint32_t, uint32_t, const DeltaArchiveManifest&, bool),
+              (override));
 
   void set_fake_mapped_devices(const std::set<std::string>& fake) override {
     DynamicPartitionControlAndroid::set_fake_mapped_devices(fake);
@@ -95,6 +100,15 @@
   std::optional<bool> RealIsAvbEnabledInFstab(const std::string& path) {
     return DynamicPartitionControlAndroid::IsAvbEnabledInFstab(path);
   }
+
+  bool RealPrepareDynamicPartitionsForUpdate(
+      uint32_t source_slot,
+      uint32_t target_slot,
+      const DeltaArchiveManifest& manifest,
+      bool delete_source) {
+    return DynamicPartitionControlAndroid::PrepareDynamicPartitionsForUpdate(
+        source_slot, target_slot, manifest, delete_source);
+  }
 };
 
 }  // namespace chromeos_update_engine