Refactor core data classes (#671)

passKitTests/CoreData/CoreDataTestCase.swift

@@ -0,0 +1,31 @@
+//
+//  CoreDataTestCase.swift
+//  pass
+//
+//  Created by Mingshen Sun on 1/4/25.
+//  Copyright © 2025 Bob Sun. All rights reserved.
+//
+
+import CoreData
+import Foundation
+import XCTest
+
+@testable import passKit
+
+// swiftlint:disable:next final_test_case
+class CoreDataTestCase: XCTestCase {
+    // swiftlint:disable:next test_case_accessibility
+    private(set) var controller: PersistenceController!
+
+    override func setUpWithError() throws {
+        try super.setUpWithError()
+
+        controller = PersistenceController(isUnitTest: true)
+        controller.setup()
+    }
+
+    override func tearDown() {
+        super.tearDown()
+        controller = nil
+    }
+}

passKitTests/CoreData/PasswordEntityTest.swift

@@ -0,0 +1,88 @@
+//
+//  PasswordEntityTest.swift
+//  pass
+//
+//  Created by Mingshen Sun on 1/4/25.
+//  Copyright © 2025 Bob Sun. All rights reserved.
+//
+
+import CoreData
+import XCTest
+
+@testable import passKit
+
+final class PasswordEntityTest: CoreDataTestCase {
+    func testFetchAll() throws {
+        let context = controller.viewContext()
+        let expectedCount = 5
+        (0 ..< expectedCount).forEach { index in
+            let name = String(format: "Generated %05d", index)
+            let path = String(format: "/%05d", index)
+            PasswordEntity.insert(name: name, path: path, isDir: false, into: context)
+        }
+        let count = PasswordEntity.fetchAllPassword(in: context).count
+        XCTAssertEqual(expectedCount, count)
+        PasswordEntity.deleteAll(in: context)
+    }
+
+    func testTotalNumber() throws {
+        let context = controller.viewContext()
+        PasswordEntity.insert(name: "1", path: "path1", isDir: false, into: context)
+        PasswordEntity.insert(name: "2", path: "path2", isDir: false, into: context)
+        PasswordEntity.insert(name: "3", path: "path3", isDir: true, into: context)
+        XCTAssertEqual(2, PasswordEntity.totalNumber(in: context))
+        PasswordEntity.deleteAll(in: context)
+    }
+
+    func testFetchUnsynced() throws {
+        let context = controller.viewContext()
+        let syncedPasswordEntity = PasswordEntity.insert(name: "1", path: "path", isDir: false, into: context)
+        syncedPasswordEntity.isSynced = true
+
+        let expectedCount = 5
+        (0 ..< expectedCount).forEach { index in
+            let name = String(format: "Generated %05d", index)
+            let path = String(format: "/%05d", index)
+            PasswordEntity.insert(name: name, path: path, isDir: false, into: context)
+        }
+        let count = PasswordEntity.fetchUnsynced(in: context).count
+        XCTAssertEqual(expectedCount, count)
+        PasswordEntity.deleteAll(in: context)
+    }
+
+    func testFetchByPath() throws {
+        let context = controller.viewContext()
+        PasswordEntity.insert(name: "1", path: "path1", isDir: false, into: context)
+        PasswordEntity.insert(name: "2", path: "path2", isDir: true, into: context)
+        let passwordEntity = PasswordEntity.fetch(by: "path1", in: context)!
+        XCTAssertEqual(passwordEntity.path, "path1")
+        XCTAssertEqual(passwordEntity.name, "1")
+    }
+
+    func testFetchByParent() throws {
+        let context = controller.viewContext()
+        let parent = PasswordEntity.insert(name: "parent", path: "path1", isDir: true, into: context)
+        let child1 = PasswordEntity.insert(name: "child1", path: "path2", isDir: false, into: context)
+        let child2 = PasswordEntity.insert(name: "child2", path: "path3", isDir: true, into: context)
+        let child3 = PasswordEntity.insert(name: "child3", path: "path4", isDir: false, into: context)
+        parent.children = [child1, child2]
+        child2.children = [child3]
+        let childern = PasswordEntity.fetch(by: parent, in: context)
+        XCTAssertEqual(childern.count, 2)
+    }
+
+    func testDeleteRecursively() throws {
+        let context = controller.viewContext()
+
+        let parent = PasswordEntity.insert(name: "parent", path: "path1", isDir: true, into: context)
+        let child1 = PasswordEntity.insert(name: "child1", path: "path2", isDir: true, into: context)
+        let child2 = PasswordEntity.insert(name: "child2", path: "path3", isDir: false, into: context)
+        let child3 = PasswordEntity.insert(name: "child3", path: "path4", isDir: false, into: context)
+        parent.children = [child1, child2]
+        child1.children = [child3]
+        PasswordEntity.deleteRecursively(entity: child2, in: context)
+        PasswordEntity.deleteRecursively(entity: child3, in: context)
+
+        XCTAssertEqual(PasswordEntity.fetchAll(in: context).count, 0)
+    }
+}