1 // 2 // Copyright (C) 2018 The Android Open Source Project 3 // 4 // Licensed under the Apache License, Version 2.0 (the "License"); 5 // you may not use this file except in compliance with the License. 6 // You may obtain a copy of the License at 7 // 8 // http://www.apache.org/licenses/LICENSE-2.0 9 // 10 // Unless required by applicable law or agreed to in writing, software 11 // distributed under the License is distributed on an "AS IS" BASIS, 12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 // See the License for the specific language governing permissions and 14 // limitations under the License. 15 // 16 17 #include <stdint.h> 18 19 #include <memory> 20 #include <set> 21 #include <string> 22 #include <vector> 23 24 #include <gmock/gmock.h> 25 26 #include "update_engine/common/dynamic_partition_control_interface.h" 27 #include "update_engine/payload_consumer/file_descriptor.h" 28 29 namespace chromeos_update_engine { 30 31 class MockDynamicPartitionControl : public DynamicPartitionControlInterface { 32 public: 33 MOCK_METHOD(void, Cleanup, (), (override)); 34 MOCK_METHOD(bool, GetDeviceDir, (std::string*), (override)); 35 MOCK_METHOD(FeatureFlag, GetDynamicPartitionsFeatureFlag, (), (override)); 36 MOCK_METHOD(FeatureFlag, GetVirtualAbCompressionFeatureFlag, (), (override)); 37 MOCK_METHOD(FeatureFlag, 38 GetVirtualAbCompressionXorFeatureFlag, 39 (), 40 (override)); 41 MOCK_METHOD(FeatureFlag, GetVirtualAbFeatureFlag, (), (override)); 42 MOCK_METHOD(bool, FinishUpdate, (bool), (override)); 43 MOCK_METHOD(std::unique_ptr<FileDescriptor>, 44 OpenCowFd, 45 (const std::string& unsuffixed_partition_name, 46 const std::optional<std::string>& source_path, 47 bool is_append), 48 (override)); 49 MOCK_METHOD(bool, MapAllPartitions, (), (override)); 50 MOCK_METHOD(bool, UnmapAllPartitions, (), (override)); 51 52 MOCK_METHOD(bool, 53 OptimizeOperation, 54 (const std::string&, const InstallOperation&, InstallOperation*), 55 (override)); 56 57 MOCK_METHOD(std::unique_ptr<android::snapshot::ISnapshotWriter>, 58 OpenCowWriter, 59 (const std::string&, const std::optional<std::string>&, bool), 60 (override)); 61 62 MOCK_METHOD( 63 bool, 64 PreparePartitionsForUpdate, 65 (uint32_t, uint32_t, const DeltaArchiveManifest&, bool, uint64_t*), 66 (override)); 67 68 MOCK_METHOD(bool, ResetUpdate, (PrefsInterface*), (override)); 69 MOCK_METHOD(std::unique_ptr<AbstractAction>, 70 GetCleanupPreviousUpdateAction, 71 (BootControlInterface*, 72 PrefsInterface*, 73 CleanupPreviousUpdateActionDelegateInterface*), 74 (override)); 75 MOCK_METHOD(bool, 76 ListDynamicPartitionsForSlot, 77 (uint32_t, uint32_t, std::vector<std::string>*), 78 (override)); 79 80 MOCK_METHOD(bool, 81 VerifyExtentsForUntouchedPartitions, 82 (uint32_t, uint32_t, const std::vector<std::string>&), 83 (override)); 84 MOCK_METHOD(bool, 85 IsDynamicPartition, 86 (const std::string&, uint32_t slot), 87 (override)); 88 MOCK_METHOD(bool, UpdateUsesSnapshotCompression, (), (override)); 89 }; 90 91 } // namespace chromeos_update_engine 92