commit 34618521d487a37d21f25812f324380f1eb394a8
author Kelvin Zhang <zhangkelvin@google.com> Mon Sep 28 09:21:02 2020 -0400
committer Kelvin Zhang <zhangkelvin@google.com> Fri Oct 23 09:52:19 2020 -0400
tree 45107761577a19fda261d73a6dfc1bba30d2e48d
parent 19e55297ac3805c7109325ea7d536887e302403e

Add CowWriter interface methods to DynamicPartiitonControl

Test: treehugger
Change-Id: I110c65e75a4594af51e68390e570ac31b50de8b5

mock_dynamic_partition_control.h

diff --git a/mock_dynamic_partition_control.h b/mock_dynamic_partition_control.h
index e85df32..5144cbb 100644
--- a/mock_dynamic_partition_control.h
+++ b/mock_dynamic_partition_control.h
@@ -22,6 +22,9 @@
 
 #include <gmock/gmock.h>
 
+#include <libsnapshot/cow_writer.h>
+
+#include "libsnapshot/snapshot_writer.h"
 #include "update_engine/common/boot_control_interface.h"
 #include "update_engine/common/dynamic_partition_control_interface.h"
 #include "update_engine/dynamic_partition_control_android.h"
@@ -81,6 +84,12 @@
               PrepareDynamicPartitionsForUpdate,
               (uint32_t, uint32_t, const DeltaArchiveManifest&, bool),
               (override));
+  MOCK_METHOD(std::unique_ptr<android::snapshot::ISnapshotWriter>,
+              OpenCowWriter,
+              (const std::string& unsuffixed_partition_name,
+               const std::optional<std::string>& source_path,
+               bool is_append),
+              (override));
 
   void set_fake_mapped_devices(const std::set<std::string>& fake) override {
     DynamicPartitionControlAndroid::set_fake_mapped_devices(fake);