298 lines
12 KiB
Swift
298 lines
12 KiB
Swift
|
import XCTest
|
||
|
|
@testable import AtlasApp
|
||
|
|
import AtlasApplication
|
||
|
|
import AtlasDomain
|
||
|
|
import AtlasInfrastructure
|
||
|
|
|
||
|
|
@MainActor
|
||
|
|
final class AtlasAppModelTests: XCTestCase {
|
||
|
|
|
||
|
|
func testCurrentSmartCleanPlanStartsAsCachedUntilSessionRefresh() {
|
||
|
|
let model = AtlasAppModel(repository: makeRepository(), workerService: AtlasScaffoldWorkerService(allowStateOnlyCleanExecution: true))
|
||
|
|
|
||
|
|
XCTAssertFalse(model.isCurrentSmartCleanPlanFresh)
|
||
|
|
XCTAssertFalse(model.canExecuteCurrentSmartCleanPlan)
|
||
|
|
XCTAssertNil(model.smartCleanPlanIssue)
|
||
|
|
}
|
||
|
|
|
||
|
|
func testFailedSmartCleanScanKeepsCachedPlanAndExposesFailureReason() async {
|
||
|
|
let repository = makeRepository()
|
||
|
|
let worker = AtlasScaffoldWorkerService(
|
||
|
|
repository: repository,
|
||
|
|
smartCleanScanProvider: FailingSmartCleanProvider()
|
||
|
|
)
|
||
|
|
let model = AtlasAppModel(repository: repository, workerService: worker)
|
||
|
|
|
||
|
|
await model.runSmartCleanScan()
|
||
|
|
|
||
|
|
XCTAssertFalse(model.isCurrentSmartCleanPlanFresh)
|
||
|
|
XCTAssertFalse(model.canExecuteCurrentSmartCleanPlan)
|
||
|
|
XCTAssertNotNil(model.smartCleanPlanIssue)
|
||
|
|
XCTAssertTrue(model.latestScanSummary.contains("Smart Clean scan is unavailable"))
|
||
|
|
}
|
||
|
|
|
||
|
|
func testRefreshPlanPreviewKeepsPlanNonExecutableWhenFindingsLackTargets() async {
|
||
|
|
let repository = makeRepository()
|
||
|
|
let worker = AtlasScaffoldWorkerService(repository: repository, allowStateOnlyCleanExecution: true)
|
||
|
|
let model = AtlasAppModel(repository: repository, workerService: worker)
|
||
|
|
|
||
|
|
let refreshed = await model.refreshPlanPreview()
|
||
|
|
|
||
|
|
XCTAssertTrue(refreshed)
|
||
|
|
XCTAssertTrue(model.isCurrentSmartCleanPlanFresh)
|
||
|
|
XCTAssertFalse(model.canExecuteCurrentSmartCleanPlan)
|
||
|
|
}
|
||
|
|
|
||
|
|
func testRunSmartCleanScanMarksPlanAsFreshForCurrentSession() async throws {
|
||
|
|
let repository = makeRepository()
|
||
|
|
let worker = AtlasScaffoldWorkerService(
|
||
|
|
repository: repository,
|
||
|
|
smartCleanScanProvider: FakeSmartCleanProvider(),
|
||
|
|
allowStateOnlyCleanExecution: true
|
||
|
|
)
|
||
|
|
let model = AtlasAppModel(repository: repository, workerService: worker)
|
||
|
|
|
||
|
|
await model.runSmartCleanScan()
|
||
|
|
|
||
|
|
XCTAssertTrue(model.isCurrentSmartCleanPlanFresh)
|
||
|
|
XCTAssertNil(model.smartCleanPlanIssue)
|
||
|
|
XCTAssertTrue(model.canExecuteCurrentSmartCleanPlan)
|
||
|
|
}
|
||
|
|
func testRunSmartCleanScanUpdatesSummaryProgressAndPlan() async throws {
|
||
|
|
let repository = makeRepository()
|
||
|
|
let worker = AtlasScaffoldWorkerService(
|
||
|
|
repository: repository,
|
||
|
|
smartCleanScanProvider: FakeSmartCleanProvider()
|
||
|
|
)
|
||
|
|
let model = AtlasAppModel(repository: repository, workerService: worker)
|
||
|
|
|
||
|
|
await model.runSmartCleanScan()
|
||
|
|
|
||
|
|
XCTAssertEqual(model.snapshot.findings.count, 2)
|
||
|
|
XCTAssertEqual(model.currentPlan.items.count, 2)
|
||
|
|
XCTAssertEqual(model.latestScanProgress, 1)
|
||
|
|
XCTAssertTrue(model.latestScanSummary.contains("2 reclaimable item"))
|
||
|
|
}
|
||
|
|
|
||
|
|
func testExecuteCurrentPlanMovesFindingsIntoRecovery() async throws {
|
||
|
|
let repository = makeRepository()
|
||
|
|
let worker = AtlasScaffoldWorkerService(
|
||
|
|
repository: repository,
|
||
|
|
smartCleanScanProvider: FakeSmartCleanProvider(),
|
||
|
|
allowStateOnlyCleanExecution: true
|
||
|
|
)
|
||
|
|
let model = AtlasAppModel(repository: repository, workerService: worker)
|
||
|
|
let initialRecoveryCount = model.snapshot.recoveryItems.count
|
||
|
|
|
||
|
|
await model.runSmartCleanScan()
|
||
|
|
await model.executeCurrentPlan()
|
||
|
|
|
||
|
|
XCTAssertGreaterThan(model.snapshot.recoveryItems.count, initialRecoveryCount)
|
||
|
|
XCTAssertEqual(model.snapshot.taskRuns.first?.kind, .executePlan)
|
||
|
|
XCTAssertGreaterThan(model.latestScanProgress, 0)
|
||
|
|
}
|
||
|
|
|
||
|
|
func testRefreshAppsUsesInventoryProvider() async throws {
|
||
|
|
let repository = makeRepository()
|
||
|
|
let worker = AtlasScaffoldWorkerService(
|
||
|
|
repository: repository,
|
||
|
|
appsInventoryProvider: FakeInventoryProvider()
|
||
|
|
)
|
||
|
|
let model = AtlasAppModel(repository: repository, workerService: worker)
|
||
|
|
|
||
|
|
await model.refreshApps()
|
||
|
|
|
||
|
|
XCTAssertEqual(model.snapshot.apps.count, 1)
|
||
|
|
XCTAssertEqual(model.snapshot.apps.first?.name, "Sample App")
|
||
|
|
XCTAssertEqual(model.latestAppsSummary, AtlasL10n.string("application.apps.loaded.one"))
|
||
|
|
}
|
||
|
|
|
||
|
|
func testRestoreRecoveryItemReturnsFindingToWorkspace() async throws {
|
||
|
|
let repository = makeRepository()
|
||
|
|
let worker = AtlasScaffoldWorkerService(repository: repository, allowStateOnlyCleanExecution: true)
|
||
|
|
let model = AtlasAppModel(repository: repository, workerService: worker)
|
||
|
|
|
||
|
|
await model.executeCurrentPlan()
|
||
|
|
let recoveryItemID = try XCTUnwrap(model.snapshot.recoveryItems.first?.id)
|
||
|
|
let findingsCountAfterExecute = model.snapshot.findings.count
|
||
|
|
|
||
|
|
await model.restoreRecoveryItem(recoveryItemID)
|
||
|
|
|
||
|
|
XCTAssertGreaterThan(model.snapshot.findings.count, findingsCountAfterExecute)
|
||
|
|
XCTAssertFalse(model.snapshot.recoveryItems.contains(where: { $0.id == recoveryItemID }))
|
||
|
|
}
|
||
|
|
|
||
|
|
func testSettingsUpdatePersistsThroughWorker() async throws {
|
||
|
|
let repository = makeRepository()
|
||
|
|
let permissionInspector = AtlasPermissionInspector(
|
||
|
|
homeDirectoryURL: FileManager.default.temporaryDirectory,
|
||
|
|
fullDiskAccessProbeURLs: [URL(fileURLWithPath: "/tmp/fda-probe")],
|
||
|
|
protectedLocationReader: { _ in false },
|
||
|
|
accessibilityStatusProvider: { false },
|
||
|
|
notificationsAuthorizationProvider: { false }
|
||
|
|
)
|
||
|
|
let worker = AtlasScaffoldWorkerService(
|
||
|
|
repository: repository,
|
||
|
|
permissionInspector: permissionInspector,
|
||
|
|
allowStateOnlyCleanExecution: true
|
||
|
|
)
|
||
|
|
let model = AtlasAppModel(
|
||
|
|
repository: repository,
|
||
|
|
workerService: worker,
|
||
|
|
notificationPermissionRequester: { true }
|
||
|
|
)
|
||
|
|
|
||
|
|
await model.setRecoveryRetentionDays(14)
|
||
|
|
await model.setNotificationsEnabled(false)
|
||
|
|
|
||
|
|
XCTAssertEqual(model.settings.recoveryRetentionDays, 14)
|
||
|
|
XCTAssertFalse(model.settings.notificationsEnabled)
|
||
|
|
XCTAssertEqual(repository.loadSettings().recoveryRetentionDays, 14)
|
||
|
|
XCTAssertFalse(repository.loadSettings().notificationsEnabled)
|
||
|
|
}
|
||
|
|
|
||
|
|
func testRefreshCurrentRouteRefreshesAppsWhenAppsSelected() async throws {
|
||
|
|
let repository = makeRepository()
|
||
|
|
let worker = AtlasScaffoldWorkerService(
|
||
|
|
repository: repository,
|
||
|
|
appsInventoryProvider: FakeInventoryProvider()
|
||
|
|
)
|
||
|
|
let model = AtlasAppModel(repository: repository, workerService: worker)
|
||
|
|
|
||
|
|
model.navigate(to: .apps)
|
||
|
|
await model.refreshCurrentRoute()
|
||
|
|
|
||
|
|
XCTAssertEqual(model.selection, .apps)
|
||
|
|
XCTAssertEqual(model.snapshot.apps.count, 1)
|
||
|
|
XCTAssertEqual(model.snapshot.apps.first?.name, "Sample App")
|
||
|
|
XCTAssertEqual(model.latestAppsSummary, AtlasL10n.string("application.apps.loaded.one"))
|
||
|
|
}
|
||
|
|
|
||
|
|
func testSetNotificationsEnabledRequestsNotificationPermissionWhenEnabling() async {
|
||
|
|
let repository = makeRepository()
|
||
|
|
let permissionInspector = AtlasPermissionInspector(
|
||
|
|
homeDirectoryURL: FileManager.default.temporaryDirectory,
|
||
|
|
fullDiskAccessProbeURLs: [URL(fileURLWithPath: "/tmp/fda-probe")],
|
||
|
|
protectedLocationReader: { _ in false },
|
||
|
|
accessibilityStatusProvider: { false },
|
||
|
|
notificationsAuthorizationProvider: { false }
|
||
|
|
)
|
||
|
|
let worker = AtlasScaffoldWorkerService(
|
||
|
|
repository: repository,
|
||
|
|
permissionInspector: permissionInspector,
|
||
|
|
allowStateOnlyCleanExecution: true
|
||
|
|
)
|
||
|
|
let recorder = NotificationPermissionRecorder()
|
||
|
|
let model = AtlasAppModel(
|
||
|
|
repository: repository,
|
||
|
|
workerService: worker,
|
||
|
|
notificationPermissionRequester: { await recorder.request() }
|
||
|
|
)
|
||
|
|
|
||
|
|
await model.setNotificationsEnabled(false)
|
||
|
|
await model.setNotificationsEnabled(true)
|
||
|
|
|
||
|
|
let callCount = await recorder.callCount()
|
||
|
|
XCTAssertEqual(callCount, 1)
|
||
|
|
}
|
||
|
|
|
||
|
|
func testRefreshPermissionsIfNeededUpdatesSnapshotFromWorker() async {
|
||
|
|
let repository = makeRepository()
|
||
|
|
let permissionInspector = AtlasPermissionInspector(
|
||
|
|
homeDirectoryURL: FileManager.default.temporaryDirectory,
|
||
|
|
fullDiskAccessProbeURLs: [URL(fileURLWithPath: "/tmp/fda-probe")],
|
||
|
|
protectedLocationReader: { _ in true },
|
||
|
|
accessibilityStatusProvider: { true },
|
||
|
|
notificationsAuthorizationProvider: { false }
|
||
|
|
)
|
||
|
|
let worker = AtlasScaffoldWorkerService(
|
||
|
|
repository: repository,
|
||
|
|
permissionInspector: permissionInspector,
|
||
|
|
allowStateOnlyCleanExecution: true
|
||
|
|
)
|
||
|
|
let model = AtlasAppModel(repository: repository, workerService: worker)
|
||
|
|
|
||
|
|
await model.refreshPermissionsIfNeeded()
|
||
|
|
|
||
|
|
XCTAssertEqual(model.snapshot.permissions.first(where: { $0.kind == .fullDiskAccess })?.isGranted, true)
|
||
|
|
XCTAssertEqual(model.snapshot.permissions.first(where: { $0.kind == .accessibility })?.isGranted, true)
|
||
|
|
XCTAssertEqual(model.snapshot.permissions.first(where: { $0.kind == .notifications })?.isGranted, false)
|
||
|
|
}
|
||
|
|
|
||
|
|
func testToggleTaskCenterFlipsPresentationState() {
|
||
|
|
let model = AtlasAppModel(repository: makeRepository(), workerService: AtlasScaffoldWorkerService(allowStateOnlyCleanExecution: true))
|
||
|
|
|
||
|
|
XCTAssertFalse(model.isTaskCenterPresented)
|
||
|
|
model.toggleTaskCenter()
|
||
|
|
XCTAssertTrue(model.isTaskCenterPresented)
|
||
|
|
model.toggleTaskCenter()
|
||
|
|
XCTAssertFalse(model.isTaskCenterPresented)
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
func testSetLanguagePersistsThroughWorkerAndUpdatesLocalization() async throws {
|
||
|
|
let repository = makeRepository()
|
||
|
|
let worker = AtlasScaffoldWorkerService(repository: repository, allowStateOnlyCleanExecution: true)
|
||
|
|
let model = AtlasAppModel(repository: repository, workerService: worker)
|
||
|
|
|
||
|
|
await model.setLanguage(.en)
|
||
|
|
|
||
|
|
XCTAssertEqual(model.settings.language, .en)
|
||
|
|
XCTAssertEqual(repository.loadSettings().language, .en)
|
||
|
|
XCTAssertEqual(AtlasRoute.overview.title, "Overview")
|
||
|
|
}
|
||
|
|
|
||
|
|
private func makeRepository() -> AtlasWorkspaceRepository {
|
||
|
|
AtlasWorkspaceRepository(
|
||
|
|
stateFileURL: FileManager.default.temporaryDirectory
|
||
|
|
.appendingPathComponent(UUID().uuidString, isDirectory: true)
|
||
|
|
.appendingPathComponent("workspace-state.json")
|
||
|
|
)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private struct FakeSmartCleanProvider: AtlasSmartCleanScanProviding {
|
||
|
|
func collectSmartCleanScan() async throws -> AtlasSmartCleanScanResult {
|
||
|
|
AtlasSmartCleanScanResult(
|
||
|
|
findings: [
|
||
|
|
Finding(title: "Build Cache", detail: "Temporary build outputs.", bytes: 512_000_000, risk: .safe, category: "Developer", targetPaths: [FileManager.default.homeDirectoryForCurrentUser.appendingPathComponent("Library/Caches/FakeBuildCache.bin").path]),
|
||
|
|
Finding(title: "Old Runtime", detail: "Unused runtime assets.", bytes: 1_024_000_000, risk: .review, category: "Developer", targetPaths: [FileManager.default.homeDirectoryForCurrentUser.appendingPathComponent("Library/Developer/Xcode/DerivedData/FakeOldRuntime").path]),
|
||
|
|
],
|
||
|
|
summary: "Smart Clean dry run found 2 reclaimable items."
|
||
|
|
)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private struct FakeInventoryProvider: AtlasAppInventoryProviding {
|
||
|
|
func collectInstalledApps() async throws -> [AppFootprint] {
|
||
|
|
[
|
||
|
|
AppFootprint(
|
||
|
|
name: "Sample App",
|
||
|
|
bundleIdentifier: "com.example.sample",
|
||
|
|
bundlePath: "/Applications/Sample App.app",
|
||
|
|
bytes: 2_048_000_000,
|
||
|
|
leftoverItems: 3
|
||
|
|
)
|
||
|
|
]
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private struct FailingSmartCleanProvider: AtlasSmartCleanScanProviding {
|
||
|
|
func collectSmartCleanScan() async throws -> AtlasSmartCleanScanResult {
|
||
|
|
throw NSError(domain: "AtlasAppModelTests", code: 1, userInfo: [NSLocalizedDescriptionKey: "Fixture scan failed."])
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private actor NotificationPermissionRecorder {
|
||
|
|
private var calls = 0
|
||
|
|
|
||
|
|
func request() -> Bool {
|
||
|
|
calls += 1
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
func callCount() -> Int {
|
||
|
|
calls
|
||
|
|
}
|
||
|
|
}
|