55 lines
2.0 KiB
Swift
55 lines
2.0 KiB
Swift
//
|
|
// Copyright 2025 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
import Testing
|
|
|
|
@testable import SignalServiceKit
|
|
|
|
@MainActor
|
|
struct BackupDisablingManagerTest {
|
|
@Test
|
|
func testPreviousFailurePersistsUntilFutureSuccess() async throws {
|
|
let mockBackupAttachmentDownloadQueueStatusManager = MockBackupAttachmentDownloadQueueStatusManager()
|
|
let mockBackupKeyService = MockBackupKeyService()
|
|
let mockBackupPlanManager = MockBackupPlanManager()
|
|
let mockBackupListMediaManager = MockBackupListMediaManager()
|
|
let mockDB = InMemoryDB()
|
|
let mockTSAccountManager = MockTSAccountManager()
|
|
|
|
let disablingManager = BackupDisablingManager(
|
|
authCredentialStore: AuthCredentialStore(dateProvider: { Date() }),
|
|
backupAttachmentDownloadQueueStatusManager: mockBackupAttachmentDownloadQueueStatusManager,
|
|
backupCDNCredentialStore: BackupCDNCredentialStore(),
|
|
backupKeyService: mockBackupKeyService,
|
|
backupListMediaManager: mockBackupListMediaManager,
|
|
backupPlanManager: mockBackupPlanManager,
|
|
backupSettingsStore: BackupSettingsStore(),
|
|
db: mockDB,
|
|
tsAccountManager: mockTSAccountManager
|
|
)
|
|
|
|
struct DeleteBackupError: Error {}
|
|
mockBackupKeyService.deleteBackupKeyMock = { throw DeleteBackupError() }
|
|
|
|
_ = await disablingManager.startDisablingBackups()
|
|
await disablingManager.disableRemotelyIfNecessary()
|
|
mockDB.read { tx in
|
|
#expect(disablingManager.disableRemotelyFailed(tx: tx))
|
|
}
|
|
|
|
mockDB.write { tx in
|
|
try! mockBackupPlanManager.setBackupPlan(.free, tx: tx)
|
|
#expect(!disablingManager.disableRemotelyFailed(tx: tx))
|
|
}
|
|
|
|
mockBackupKeyService.deleteBackupKeyMock = { }
|
|
_ = await disablingManager.startDisablingBackups()
|
|
await disablingManager.disableRemotelyIfNecessary()
|
|
mockDB.read { tx in
|
|
#expect(!disablingManager.disableRemotelyFailed(tx: tx))
|
|
}
|
|
}
|
|
}
|