Implement clear method for all storage types and add clear calls in tests

Author: rizwankce

Storage/Classes/Storage.swift

@@ -73,7 +73,12 @@ public final class Storage<T> where T: Codable {
         try? fileManager.createDirectory(at: folder, withIntermediateDirectories: false, attributes: nil)
     }
 
-    private func clearStorage() {
-        try? FileManager.default.removeItem(at: type.folder)
+    public func clear() {
+        switch type {
+            case .cache, .document:
+                try? FileManager.default.removeItem(at: type.folder)
+            case .userDefaults:
+                UserDefaults.standard.removeObject(forKey: type.userDefaultsKey)
+        }
     }
 }

Storage/Classes/StorageType.swift

@@ -7,13 +7,14 @@ public enum StorageType {
 
     public var searchPathDirectory: FileManager.SearchPathDirectory {
         switch self {
-        case .cache: return .cachesDirectory
-        case .document: return .documentDirectory
-        case .userDefaults: fatalError("UserDefaults does not have a search path directory")
+            case .cache: return .cachesDirectory
+            case .document: return .documentDirectory
+            case .userDefaults: return .cachesDirectory
         }
     }
 
     public var folder: URL {
+
         guard let path = NSSearchPathForDirectoriesInDomains(searchPathDirectory, .userDomainMask, true).first else {
             fatalError("Cannot find the path directory for storage")
         }

Tests/StorageTests.swift

@@ -11,6 +11,7 @@ class StorageTests: XCTestCase {
 
         let retrievedData = storage.storedValue
         XCTAssertEqual(retrievedData, testData, "Stored and retrieved data should be equal")
+        storage.clear()
     }
 
     func testRetrieveNonExistentFile() {
@@ -29,6 +30,7 @@ class StorageTests: XCTestCase {
 
         let retrievedData = storage.storedValue
         XCTAssertEqual(retrievedData, newData, "Stored and retrieved data should be equal after overwrite")
+        storage.clear()
     }
 
     func testSaveAndRetrieveUserDefaults() {
@@ -39,12 +41,14 @@ class StorageTests: XCTestCase {
 
         let retrievedData = storage.storedValue
         XCTAssertEqual(retrievedData, testData, "Stored and retrieved data should be equal for user defaults")
+        storage.clear()
     }
 
     func testRetrieveNonExistentUserDefaults() {
         let storage = Storage<[String]>(storageType: .userDefaults, filename: "nonexistentUserDefaults")
         let retrievedData = storage.storedValue
         XCTAssertNil(retrievedData, "Retrieving non-existent data from user defaults should return nil")
+        storage.clear()
     }
 
     func testOverwriteUserDefaultsData() {
@@ -57,5 +61,6 @@ class StorageTests: XCTestCase {
 
         let retrievedData = storage.storedValue
         XCTAssertEqual(retrievedData, newData, "Stored and retrieved data should be equal after overwrite in user defaults")
+        storage.clear()
     }
 }

Tests/StorageTypeTests.swift

@@ -13,13 +13,4 @@ class StorageTypeTests: XCTestCase {
         let folder = storageType.folder
         XCTAssertTrue(folder.path.contains("Documents"), "Document directory path should contain 'Documents'")
     }
-
-    func testFolderCreation() {
-        let storageType = StorageType.document
-        let folder = storageType.folder
-        let fileManager = FileManager.default
-        var isDir: ObjCBool = false
-        let exists = fileManager.fileExists(atPath: folder.path, isDirectory: &isDir)
-        XCTAssertTrue(exists && isDir.boolValue, "Folder should be created and be a directory")
-    }
 }