Add Tests!

Author: thecoolwinter

…ments/CodeFileDocument+UTTypeTests.swift …eFile/CodeFileDocument+UTTypeTests.swiftCodeEditTests/Features/Documents/CodeFileDocument+UTTypeTests.swift renamed to CodeEditTests/Features/CodeFile/CodeFileDocument+UTTypeTests.swift CodeEditTests/Features/Documents/CodeFileDocument+UTTypeTests.swift renamed to CodeEditTests/Features/CodeFile/CodeFileDocument+UTTypeTests.swift

@@ -0,0 +1,103 @@
+//
+//  CodeFileDocumentTests.swift
+//  CodeEditModules/CodeFileTests
+//
+//  Created by Marco Carnevali on 18/03/22.
+//
+
+import Foundation
+import SwiftUI
+import Testing
+@testable import CodeEdit
+
+@Suite(.serialized)
+struct CodeFileDocumentTests {
+    let defaultString = "func test() { }"
+
+    private func withFile(_ operation: (URL) throws -> Void) throws {
+        try withTempDir { dir in
+            let fileURL = dir.appending(path: "file.swift")
+            try operation(fileURL)
+        }
+    }
+
+    private func withCodeFile(_ operation: (CodeFileDocument) throws -> Void) throws {
+        try withFile { fileURL in
+            try defaultString.write(to: fileURL, atomically: true, encoding: .utf8)
+            let codeFile = try CodeFileDocument(contentsOf: fileURL, ofType: "public.source-code")
+            try operation(codeFile)
+        }
+    }
+
+    @Test
+    func testLoadUTF8Encoding() throws {
+        try withFile { fileURL in
+            try defaultString.write(to: fileURL, atomically: true, encoding: .utf8)
+            let codeFile = try CodeFileDocument(
+                for: fileURL,
+                withContentsOf: fileURL,
+                ofType: "public.source-code"
+            )
+            #expect(codeFile.content?.string == defaultString)
+            #expect(codeFile.sourceEncoding == .utf8)
+        }
+    }
+
+    @Test
+    func testWriteUTF8Encoding() throws {
+        try withFile { fileURL in
+            let codeFile = CodeFileDocument()
+            codeFile.content = NSTextStorage(string: defaultString)
+            codeFile.sourceEncoding = .utf8
+            try codeFile.write(to: fileURL, ofType: "public.source-code")
+
+            let data = try Data(contentsOf: fileURL)
+            var nsString: NSString?
+            let fileEncoding = NSString.stringEncoding(
+                for: data,
+                encodingOptions: [
+                    .suggestedEncodingsKey: FileEncoding.allCases.map { $0.nsValue },
+                    .useOnlySuggestedEncodingsKey: true
+                ],
+                convertedString: &nsString,
+                usedLossyConversion: nil
+            )
+
+            #expect(codeFile.content?.string as NSString? == nsString)
+            #expect(fileEncoding == NSUTF8StringEncoding)
+        }
+    }
+
+    @Test
+    func ignoresExternalUpdatesWithOutstandingChanges() throws {
+        try withCodeFile { codeFile in
+            // Mark the file dirty
+            codeFile.updateChangeCount(.changeDone)
+
+            // Update the modification date
+            try "different contents".write(to: codeFile.fileURL!, atomically: true, encoding: .utf8)
+
+            // Tell the file the disk representation changed
+            codeFile.presentedItemDidChange()
+
+            // The file should not have reloaded
+            #expect(codeFile.content?.string == defaultString)
+            #expect(codeFile.isDocumentEdited == true)
+        }
+    }
+
+    @Test
+    func loadsExternalUpdatesWithNoOutstandingChanges() throws {
+        try withCodeFile { codeFile in
+            // Update the modification date
+            try "different contents".write(to: codeFile.fileURL!, atomically: true, encoding: .utf8)
+
+            // Tell the file the disk representation changed
+            codeFile.presentedItemDidChange()
+
+            // The file should have reloaded (it was clean)
+            #expect(codeFile.content?.string == "different contents")
+            #expect(codeFile.isDocumentEdited == false)
+        }
+    }
+}

CodeEditTests/Features/CodeFile/CodeFileDocumentTests.swift

@@ -0,0 +1,48 @@
+//
+//  withTempDir.swift
+//  CodeEditTests
+//
+//  Created by Khan Winter on 7/3/25.
+//
+
+import Foundation
+
+func withTempDir(_ test: (URL) async throws -> Void) async throws {
+    let tempDirURL = try createAndClearDir()
+    do {
+        try await test(tempDirURL)
+    } catch {
+        try clearDir(tempDirURL)
+        throw error
+    }
+    try clearDir(tempDirURL)
+}
+
+func withTempDir(_ test: (URL) throws -> Void) throws {
+    let tempDirURL = try createAndClearDir()
+    do {
+        try test(tempDirURL)
+    } catch {
+        try clearDir(tempDirURL)
+        throw error
+    }
+    try clearDir(tempDirURL)
+}
+
+private func createAndClearDir() throws -> URL {
+    let tempDirURL = FileManager.default.temporaryDirectory
+        .appending(path: "CodeEditTestDirectory", directoryHint: .isDirectory)
+
+    // If it exists, delete it before the test
+    try clearDir(tempDirURL)
+
+    try FileManager.default.createDirectory(at: tempDirURL, withIntermediateDirectories: true)
+
+    return tempDirURL
+}
+
+private func clearDir(_ url: URL) throws {
+    if FileManager.default.fileExists(atPath: url.absoluteURL.path(percentEncoded: false)) {
+        try FileManager.default.removeItem(at: url)
+    }
+}