passforios/passKitTests/Controllers/PersistenceControllerTest.swift at 55b682b4b0cf7e836be1cd099e4010178440d439 · mssun/passforios · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
//
//  PersistenceControllerTest.swift
//  passKitTests
//
//  Created by Lysann Tranvouez on 9/3/26.
//  Copyright © 2026 Bob Sun. All rights reserved.
//

import CoreData
import XCTest

@testable import passKit

final class PersistenceControllerTest: XCTestCase {
    func testModelLoads() {
        let controller = PersistenceController.forUnitTests()
        let context = controller.viewContext()

        let entityNames = context.persistentStoreCoordinator!.managedObjectModel.entities.map(\.name)
        XCTAssertEqual(entityNames, ["PasswordEntity"])
    }

    func testInsertAndFetch() {
        let controller = PersistenceController.forUnitTests()
        let context = controller.viewContext()
        XCTAssertEqual(PasswordEntity.fetchAll(in: context).count, 0)

        PasswordEntity.insert(name: "test", path: "test.gpg", isDir: false, into: context)
        try? context.save()

        XCTAssertEqual(PasswordEntity.fetchAll(in: context).count, 1)
    }

    func testReinitializePersistentStoreClearsData() {
        let controller = PersistenceController.forUnitTests()
        let context = controller.viewContext()

        PasswordEntity.insert(name: "test1", path: "test1.gpg", isDir: false, into: context)
        PasswordEntity.insert(name: "test2", path: "test2.gpg", isDir: false, into: context)
        try? context.save()
        XCTAssertEqual(PasswordEntity.fetchAll(in: context).count, 2)

        controller.reinitializePersistentStore()

        // After reinitialize, old data should be gone
        // (reinitializePersistentStore calls initPasswordEntityCoreData with the default repo URL,
        // which won't exist in tests, so the result should be an empty store)
        let remaining = PasswordEntity.fetchAll(in: context)
        XCTAssertEqual(remaining.count, 0)
    }

    func testMultipleControllersAreIndependent() {
        let controller1 = PersistenceController.forUnitTests()
        let controller2 = PersistenceController.forUnitTests()

        let context1 = controller1.viewContext()
        let context2 = controller2.viewContext()

        PasswordEntity.insert(name: "only-in-1", path: "only-in-1.gpg", isDir: false, into: context1)
        try? context1.save()

        XCTAssertEqual(PasswordEntity.fetchAll(in: context1).count, 1)
        XCTAssertEqual(PasswordEntity.fetchAll(in: context2).count, 0)
    }

    func testSaveAndLoadFromFile() throws {
        let tempDir = FileManager.default.temporaryDirectory.appendingPathComponent(UUID().uuidString)
        try FileManager.default.createDirectory(at: tempDir, withIntermediateDirectories: true)
        defer { try? FileManager.default.removeItem(at: tempDir) }
        let storeURL = tempDir.appendingPathComponent("test.sqlite")

        // Write
        let controller1 = PersistenceController(storeURL: storeURL)
        let context1 = controller1.viewContext()
        PasswordEntity.insert(name: "saved", path: "saved.gpg", isDir: false, into: context1)
        PasswordEntity.insert(name: "dir", path: "dir", isDir: true, into: context1)
        controller1.save()

        // Load in a fresh controller from the same file
        let controller2 = PersistenceController(storeURL: storeURL)
        let context2 = controller2.viewContext()
        let allEntities = PasswordEntity.fetchAll(in: context2)

        XCTAssertEqual(allEntities.count, 2)
        XCTAssertNotNil(allEntities.first { $0.name == "saved" && !$0.isDir })
        XCTAssertNotNil(allEntities.first { $0.name == "dir" && $0.isDir })
    }

    func testSaveError() throws {
        // NOTE: save() calls fatalError on Core Data save failures, so error propagation
        // cannot be tested without refactoring save() to throw...
    }
}