111 lines
4.2 KiB
C++
111 lines
4.2 KiB
C++
//
|
|
// 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
|