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 #include <libsnapshot/cow_writer.h> 26 27 #include "update_engine/common/dynamic_partition_control_interface.h" 28 #include "update_engine/payload_consumer/file_descriptor.h" 29 30 namespace chromeos_update_engine { 31 32 class MockDynamicPartitionControl : public DynamicPartitionControlInterface { 33 public: 34 MOCK_METHOD(void, Cleanup, (), (override)); 35 MOCK_METHOD(bool, GetDeviceDir, (std::string*), (override)); 36 MOCK_METHOD(FeatureFlag, GetDynamicPartitionsFeatureFlag, (), (override)); 37 MOCK_METHOD(FeatureFlag, GetVirtualAbCompressionFeatureFlag, (), (override)); 38 MOCK_METHOD(FeatureFlag, 39 GetVirtualAbCompressionXorFeatureFlag, 40 (), 41 (override)); 42 MOCK_METHOD(FeatureFlag, 43 GetVirtualAbUserspaceSnapshotsFeatureFlag, 44 (), 45 (override)); 46 MOCK_METHOD(FeatureFlag, GetVirtualAbFeatureFlag, (), (override)); 47 MOCK_METHOD(bool, FinishUpdate, (bool), (override)); 48 MOCK_METHOD(std::unique_ptr<FileDescriptor>, 49 OpenCowFd, 50 (const std::string& unsuffixed_partition_name, 51 const std::optional<std::string>& source_path, 52 bool is_append), 53 (override)); 54 MOCK_METHOD(bool, MapAllPartitions, (), (override)); 55 MOCK_METHOD(bool, UnmapAllPartitions, (), (override)); 56 57 MOCK_METHOD(bool, 58 OptimizeOperation, 59 (const std::string&, const InstallOperation&, InstallOperation*), 60 (override)); 61 62 MOCK_METHOD(std::unique_ptr<android::snapshot::ICowWriter>, 63 OpenCowWriter, 64 (const std::string&, 65 const std::optional<std::string>&, 66 std::optional<uint64_t> label), 67 (override)); 68 69 MOCK_METHOD(bool, 70 PreparePartitionsForUpdate, 71 (uint32_t, 72 uint32_t, 73 const DeltaArchiveManifest&, 74 bool, 75 uint64_t*, 76 ErrorCode*), 77 (override)); 78 79 MOCK_METHOD(bool, ResetUpdate, (PrefsInterface*), (override)); 80 MOCK_METHOD(std::unique_ptr<AbstractAction>, 81 GetCleanupPreviousUpdateAction, 82 (BootControlInterface*, 83 PrefsInterface*, 84 CleanupPreviousUpdateActionDelegateInterface*), 85 (override)); 86 MOCK_METHOD(bool, 87 ListDynamicPartitionsForSlot, 88 (uint32_t, uint32_t, std::vector<std::string>*), 89 (override)); 90 91 MOCK_METHOD(bool, 92 VerifyExtentsForUntouchedPartitions, 93 (uint32_t, uint32_t, const std::vector<std::string>&), 94 (override)); 95 MOCK_METHOD(bool, 96 IsDynamicPartition, 97 (const std::string&, uint32_t slot), 98 (override)); 99 MOCK_METHOD(bool, UpdateUsesSnapshotCompression, (), (override)); 100 }; 101 102 } // namespace chromeos_update_engine 103