diff options
Diffstat (limited to 'mock_dynamic_partition_control.h')
-rw-r--r-- | mock_dynamic_partition_control.h | 110 |
1 files changed, 0 insertions, 110 deletions
diff --git a/mock_dynamic_partition_control.h b/mock_dynamic_partition_control.h deleted file mode 100644 index 1e4e5fd8..00000000 --- a/mock_dynamic_partition_control.h +++ /dev/null @@ -1,110 +0,0 @@ -// -// Copyright (C) 2018 The Android Open Source Project -// -// Licensed under the Apache License, Version 2.0 (the "License"); -// you may not use this file except in compliance with the License. -// You may obtain a copy of the License at -// -// http://www.apache.org/licenses/LICENSE-2.0 -// -// Unless required by applicable law or agreed to in writing, software -// distributed under the License is distributed on an "AS IS" BASIS, -// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -// See the License for the specific language governing permissions and -// limitations under the License. -// - -#include <stdint.h> - -#include <memory> -#include <set> -#include <string> - -#include <gmock/gmock.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" - -namespace chromeos_update_engine { - -class MockDynamicPartitionControl : public DynamicPartitionControlInterface { - public: - MOCK_METHOD5(MapPartitionOnDeviceMapper, - bool(const std::string&, - const std::string&, - uint32_t, - bool, - std::string*)); - MOCK_METHOD0(Cleanup, void()); - MOCK_METHOD0(GetDynamicPartitionsFeatureFlag, FeatureFlag()); - MOCK_METHOD5( - PreparePartitionsForUpdate, - bool(uint32_t, uint32_t, const DeltaArchiveManifest&, bool, uint64_t*)); - MOCK_METHOD0(GetVirtualAbFeatureFlag, FeatureFlag()); - MOCK_METHOD1(FinishUpdate, bool(bool)); - MOCK_METHOD0(CleanupSuccessfulUpdate, ErrorCode()); - MOCK_METHOD3(GetCleanupPreviousUpdateAction, - std::unique_ptr<AbstractAction>( - BootControlInterface*, - PrefsInterface*, - CleanupPreviousUpdateActionDelegateInterface*)); -}; - -class MockDynamicPartitionControlAndroid - : public DynamicPartitionControlAndroid { - public: - MOCK_METHOD5(MapPartitionOnDeviceMapper, - bool(const std::string&, - const std::string&, - uint32_t, - bool, - std::string*)); - MOCK_METHOD1(UnmapPartitionOnDeviceMapper, bool(const std::string&)); - MOCK_METHOD0(Cleanup, void()); - MOCK_METHOD1(DeviceExists, bool(const std::string&)); - MOCK_METHOD1(GetState, ::android::dm::DmDeviceState(const std::string&)); - MOCK_METHOD2(GetDmDevicePathByName, bool(const std::string&, std::string*)); - MOCK_METHOD3(LoadMetadataBuilder, - std::unique_ptr<::android::fs_mgr::MetadataBuilder>( - const std::string&, uint32_t, uint32_t)); - MOCK_METHOD3(StoreMetadata, - bool(const std::string&, - android::fs_mgr::MetadataBuilder*, - uint32_t)); - MOCK_METHOD1(GetDeviceDir, bool(std::string*)); - MOCK_METHOD0(GetDynamicPartitionsFeatureFlag, FeatureFlag()); - MOCK_METHOD1(GetSuperPartitionName, std::string(uint32_t)); - MOCK_METHOD0(GetVirtualAbFeatureFlag, FeatureFlag()); - MOCK_METHOD1(FinishUpdate, bool(bool)); - MOCK_METHOD5( - GetSystemOtherPath, - bool(uint32_t, uint32_t, const std::string&, std::string*, bool*)); - MOCK_METHOD2(EraseSystemOtherAvbFooter, bool(uint32_t, uint32_t)); - MOCK_METHOD0(IsAvbEnabledOnSystemOther, std::optional<bool>()); - - void set_fake_mapped_devices(const std::set<std::string>& fake) override { - DynamicPartitionControlAndroid::set_fake_mapped_devices(fake); - } - - bool RealGetSystemOtherPath(uint32_t source_slot, - uint32_t target_slot, - const std::string& partition_name_suffix, - std::string* path, - bool* should_unmap) { - return DynamicPartitionControlAndroid::GetSystemOtherPath( - source_slot, target_slot, partition_name_suffix, path, should_unmap); - } - - bool RealEraseSystemOtherAvbFooter(uint32_t source_slot, - uint32_t target_slot) { - return DynamicPartitionControlAndroid::EraseSystemOtherAvbFooter( - source_slot, target_slot); - } - - std::optional<bool> RealIsAvbEnabledInFstab(const std::string& path) { - return DynamicPartitionControlAndroid::IsAvbEnabledInFstab(path); - } -}; - -} // namespace chromeos_update_engine |