LocalData/Tests/LocalDataTests/Mocks/MockMigration.swift
Matt Bruce 70534f647b Update Migrations, Models, Protocols (+2 more) and tests, docs
Summary:
- Sources: update Migrations, Models, Protocols (+2 more)
- Tests: update tests for AnyStorageKeyTests.swift, MigrationAdditionalTests.swift, MigrationIntegrationTests.swift (+3 more)
- Docs: update docs for Migration, Migration_Refactor_Plan_Clean, Proposal (+1 more)

Stats:
- 31 files changed, 2820 insertions(+), 80 deletions(-)
2026-01-18 13:43:10 -06:00

43 lines
1.3 KiB
Swift

import Foundation
@testable import LocalData
struct MockMigration<Destination: StorageKey>: StorageMigration {
typealias DestinationKey = Destination
let destinationKey: Destination
let shouldSucceed: Bool
let shouldMigrateResult: Bool
let migrationDelay: TimeInterval
func shouldMigrate(using router: StorageRouter, context: MigrationContext) async throws -> Bool {
try await Task.sleep(for: .seconds(0.1))
return shouldMigrateResult
}
func migrate(using router: StorageRouter, context: MigrationContext) async throws -> MigrationResult {
try await Task.sleep(for: .seconds(migrationDelay))
if shouldSucceed {
return MigrationResult(success: true, migratedCount: 1)
}
return MigrationResult(
success: false,
errors: [.transformationFailed("Mock failure")]
)
}
}
struct FailingMigration<Destination: StorageKey>: StorageMigration {
typealias DestinationKey = Destination
let destinationKey: Destination
let error: MigrationError
func shouldMigrate(using router: StorageRouter, context: MigrationContext) async throws -> Bool { true }
func migrate(using router: StorageRouter, context: MigrationContext) async throws -> MigrationResult {
MigrationResult(success: false, errors: [error])
}
}