🔨 Added some additional tests, including write tests

Author: admkopec

LSFileWrapperTests/LSFileWrapperTests.swift

@@ -28,6 +28,51 @@ class LSFileWrapperTests: XCTestCase {
         XCTAssertEqual(fileWrapper.string(), "Hello World!")
     }
 
+    func testUpdateDataContent() {
+        // This is a test case for testing update(newContent:) method
+        let fileWrapper = LSFileWrapper(file: ())
+        fileWrapper.update(newContent: Data("Hello World!".utf8) as NSData)
+        
+        XCTAssertNotNil(fileWrapper.data())
+        XCTAssertEqual(fileWrapper.data(), Data("Hello World!".utf8))
+        XCTAssertEqual(fileWrapper.string(), "Hello World!") // This sets the type
+        XCTAssertNil(fileWrapper.data()) // So here data() should be nil
+    }
+    
+    func testUpdateDictionaryContent() {
+        // This is a test case for testing update(newContent:) method
+        let testDict = ["Name": "LSFileWrapper", "Platforms": "macOS, iOS", "Version": "2"]
+        let fileWrapper = LSFileWrapper(file: ())
+        fileWrapper.update(newContent: testDict as NSDictionary)
+        
+        XCTAssertNotNil(fileWrapper.dictionary())
+        XCTAssertEqual(fileWrapper.dictionary() as? [String: String], testDict)
+    }
+    
+    func testUpdateAnyDictionaryContent() {
+        // This is a test case for testing update(newContent:) method
+        let testDict = ["Name": "LSFileWrapper", "Platforms": ["macOS", "iOS"], "Version": 2] as [String : Any]
+        let fileWrapper = LSFileWrapper(file: ())
+        fileWrapper.update(newContent: testDict as NSDictionary)
+        
+        XCTAssertNotNil(fileWrapper.dictionary())
+        let dict = fileWrapper.dictionary() as? [String: Any]
+        XCTAssertNotNil(dict)
+        XCTAssertEqual(dict?["Name"] as? String, (testDict["Name"] as! String))
+        XCTAssertEqual(dict?["Version"] as? Int, (testDict["Version"] as! Int))
+        XCTAssertEqual(dict?["Platforms"] as? [String], (testDict["Platforms"] as! [String]))
+    }
+    
+    func testUpdateImageContent() {
+        // This is a test case for testing update(newContent:) method
+        let testImg = NSImage(named: "NSApplicationIcon") ?? NSImage()
+        let fileWrapper = LSFileWrapper(file: ())
+        fileWrapper.update(newContent: testImg)
+        
+        XCTAssertNotNil(fileWrapper.image())
+        XCTAssertEqual(fileWrapper.image()?.size, testImg.size) // More or less enough
+    }
+    
     func testAddContent() {
         // This is a test case for testing add(content: withFilename:) method
         let wrapper = LSFileWrapper(directory: ())
@@ -94,4 +139,116 @@ class LSFileWrapperTests: XCTestCase {
 
         XCTAssertEqual(wrapperNames.sorted(by: { $1 ?? "" < $0 ?? "" }), ["hello.txt", "directory"])
     }
+    
+    func testWriteToURL() throws {
+        let url = URL(fileURLWithPath: NSTemporaryDirectory()).appendingPathComponent("testWrite.package")
+        let testDict = ["Name": "LSFileWrapper", "Platforms": ["macOS", "iOS"], "Version": 2] as [String : Any]
+        let testImg = NSImage(named: "NSApplicationIcon") ?? NSImage()
+        let wrapper = LSFileWrapper(directory: ())
+        wrapper.add(content: "Hello World!" as NSString, withFilename: "hello.txt")
+        wrapper.add(content: testDict as NSDictionary, withFilename: "hello.plist")
+        wrapper.add(content: testImg, withFilename: "hello.jpeg")
+        wrapper.add(wrapper: LSFileWrapper(directory: ()), withFilename: "directory")
+
+        try wrapper.write(to: url)
+        
+        let writtenWrapper = LSFileWrapper(url: url, isDirectory: true)
+        
+        XCTAssertNotNil(writtenWrapper)
+        XCTAssert(writtenWrapper?.isDirectory == true)
+        
+        XCTAssertNotNil(writtenWrapper?.wrapper(with: "hello.txt"))
+        XCTAssert(writtenWrapper?.wrapper(with: "hello.txt")?.isDirectory == false)
+        XCTAssertNotNil(writtenWrapper?.wrapper(with: "hello.plist"))
+        XCTAssert(writtenWrapper?.wrapper(with: "hello.plist")?.isDirectory == false)
+        XCTAssertNotNil(writtenWrapper?.wrapper(with: "hello.jpeg"))
+        XCTAssert(writtenWrapper?.wrapper(with: "hello.jpeg")?.isDirectory == false)
+        XCTAssertNotNil(writtenWrapper?.wrapper(with: "directory"))
+        XCTAssert(writtenWrapper?.wrapper(with: "directory")?.isDirectory == true)
+        
+        XCTAssertEqual(writtenWrapper?.wrapper(with: "hello.txt")?.string(), "Hello World!")
+        XCTAssertEqual(writtenWrapper?.wrapper(with: "hello.jpeg")?.image()?.size, testImg.size) // More or less enough
+        
+        let dict = writtenWrapper?.wrapper(with: "hello.plist")?.dictionary() as? [String: Any]
+        XCTAssertNotNil(dict)
+        XCTAssertEqual(dict?["Name"] as? String, (testDict["Name"] as! String))
+        XCTAssertEqual(dict?["Version"] as? Int, (testDict["Version"] as! Int))
+        XCTAssertEqual(dict?["Platforms"] as? [String], (testDict["Platforms"] as! [String]))
+        
+        try FileManager.default.removeItem(at: url)
+    }
+    
+    func testWriteFileToURL() throws {
+        let url = URL(fileURLWithPath: NSTemporaryDirectory()).appendingPathComponent("hello.txt")
+        let fileWrapper = LSFileWrapper(file: ())
+        fileWrapper.update(newContent: "Hello World!" as NSString)
+
+        try fileWrapper.write(to: url)
+        
+        let writtenFileWrapper = LSFileWrapper(url: url, isDirectory: false)
+        
+        XCTAssertNotNil(writtenFileWrapper)
+        
+        XCTAssert(writtenFileWrapper?.isDirectory == false)
+        
+        XCTAssertEqual(writtenFileWrapper?.data(), Data("Hello World!".utf8))
+        XCTAssertEqual(writtenFileWrapper?.string(), "Hello World!") // This sets the type
+        XCTAssertNil(writtenFileWrapper?.data()) // So here data() should be nil
+        
+        try FileManager.default.removeItem(at: url)
+    }
+    
+    func testInitWithContentsAutomaticDirDiscovery() throws {
+        let url = URL(fileURLWithPath: NSTemporaryDirectory()).appendingPathComponent("testAutoDir.package")
+        let wrapper = LSFileWrapper(directory: ())
+        wrapper.add(content: "Hello World!" as NSString, withFilename: "hello.txt")
+
+        try wrapper.write(to: url)
+        
+        let writtenWrapper = LSFileWrapper(url: url, isDirectory: false)
+        
+        XCTAssertNotNil(writtenWrapper)
+        
+        XCTAssert(writtenWrapper?.isDirectory == true)
+        
+        XCTAssertNotNil(writtenWrapper?.wrapper(with: "hello.txt"))
+        XCTAssert(writtenWrapper?.wrapper(with: "hello.txt")?.isDirectory == false)
+        
+        try FileManager.default.removeItem(at: url)
+    }
+    
+    func testWriteUpdatesToURL() throws {
+        let url = URL(fileURLWithPath: NSTemporaryDirectory()).appendingPathComponent("testWriteUpdates.package")
+        let wrapper = LSFileWrapper(directory: ())
+        wrapper.add(content: "Hello World!" as NSString, withFilename: "hello.txt")
+
+        try wrapper.write(to: url)
+        
+        let writtenWrapper = LSFileWrapper(url: url, isDirectory: true)
+        
+        XCTAssertNotNil(writtenWrapper)
+        XCTAssert(writtenWrapper?.isDirectory == true)
+        
+        XCTAssertNotNil(writtenWrapper?.wrapper(with: "hello.txt"))
+        XCTAssert(writtenWrapper?.wrapper(with: "hello.txt")?.isDirectory == false)
+        
+        XCTAssertEqual(writtenWrapper?.wrapper(with: "hello.txt")?.string(), "Hello World!")
+        
+        wrapper.wrapper(with: "hello.txt")?.update(newContent: "Hello Updated World!" as NSString)
+        XCTAssertEqual(wrapper.wrapper(with: "hello.txt")?.string(), "Hello Updated World!")
+        
+        try wrapper.writeUpdates(to: url)
+        
+        let updatedWrapper = LSFileWrapper(url: url, isDirectory: true)
+        
+        XCTAssertNotNil(updatedWrapper)
+        XCTAssert(updatedWrapper?.isDirectory == true)
+        
+        XCTAssertNotNil(updatedWrapper?.wrapper(with: "hello.txt"))
+        XCTAssert(updatedWrapper?.wrapper(with: "hello.txt")?.isDirectory == false)
+        
+        XCTAssertEqual(updatedWrapper?.wrapper(with: "hello.txt")?.string(), "Hello Updated World!")
+        
+        try FileManager.default.removeItem(at: url)
+    }
 }