Add StorageTests

Author: onmyway133

Cache.xcodeproj/project.pbxproj

@@ -81,6 +81,8 @@
 		D27014AF20D12D84003B45C7 /* AsyncStorageTests.swift in Sources */ = {isa = PBXBuildFile; fileRef = D292DB031F6AA0730060F614 /* AsyncStorageTests.swift */; };
 		D27014B020D12E37003B45C7 /* StorageSupportTests.swift in Sources */ = {isa = PBXBuildFile; fileRef = D2CF98881F695F9400CE8F68 /* StorageSupportTests.swift */; };
 		D27014B120D12E38003B45C7 /* StorageSupportTests.swift in Sources */ = {isa = PBXBuildFile; fileRef = D2CF98881F695F9400CE8F68 /* StorageSupportTests.swift */; };
+		D27014B320D13E2C003B45C7 /* StorageTests.swift in Sources */ = {isa = PBXBuildFile; fileRef = D236F3191F6BEF73004EE01F /* StorageTests.swift */; };
+		D27014B420D13E2C003B45C7 /* StorageTests.swift in Sources */ = {isa = PBXBuildFile; fileRef = D236F3191F6BEF73004EE01F /* StorageTests.swift */; };
 		D28897051F8B79B300C61DEE /* JSONDecoder+Extensions.swift in Sources */ = {isa = PBXBuildFile; fileRef = D28897041F8B79B300C61DEE /* JSONDecoder+Extensions.swift */; };
 		D28897061F8B79B300C61DEE /* JSONDecoder+Extensions.swift in Sources */ = {isa = PBXBuildFile; fileRef = D28897041F8B79B300C61DEE /* JSONDecoder+Extensions.swift */; };
 		D28897071F8B79B300C61DEE /* JSONDecoder+Extensions.swift in Sources */ = {isa = PBXBuildFile; fileRef = D28897041F8B79B300C61DEE /* JSONDecoder+Extensions.swift */; };
@@ -828,6 +830,7 @@
 			files = (
 				D2CF98201F69427C00CE8F68 /* TestCase+Extensions.swift in Sources */,
 				D2CF98231F69427C00CE8F68 /* TestHelper.swift in Sources */,
+				D27014B420D13E2C003B45C7 /* StorageTests.swift in Sources */,
 				D27014A320D129A3003B45C7 /* DiskStorageTests.swift in Sources */,
 				D27014A020D12870003B45C7 /* MemoryStorageTests.swift in Sources */,
 				D2CF98261F69427C00CE8F68 /* User.swift in Sources */,
@@ -852,6 +855,7 @@
 				D2CF987F1F69513800CE8F68 /* ImageWrapperTests.swift in Sources */,
 				D2D4CC1A1FA3166900E4A2D5 /* MD5Tests.swift in Sources */,
 				D2D4CC281FA342CA00E4A2D5 /* JSONWrapperTests.swift in Sources */,
+				D27014B320D13E2C003B45C7 /* StorageTests.swift in Sources */,
 				D28C9BAF1F67EF8300C180C1 /* UIImage+ExtensionsTests.swift in Sources */,
 				D2CF987D1F69513800CE8F68 /* MemoryCapsuleTests.swift in Sources */,
 				D27014A120D129A2003B45C7 /* DiskStorageTests.swift in Sources */,

Tests/iOS/Tests/Storage/StorageTests.swift

@@ -2,13 +2,17 @@ import XCTest
 import Cache
 
 final class StorageTests: XCTestCase {
-  private var storage: Storage!
+  private var storage: Storage<User>!
   let user = User(firstName: "John", lastName: "Snow")
 
   override func setUp() {
     super.setUp()
 
-    storage = try! Storage(diskConfig: DiskConfig(name: "Thor"), memoryConfig: MemoryConfig())
+    storage = try! Storage<User>(
+      diskConfig: DiskConfig(name: "Thor"),
+      memoryConfig: MemoryConfig(),
+      transformer: TransformerFactory.forCodable(ofType: User.self)
+    )
   }
 
   override func tearDown() {
@@ -18,7 +22,7 @@ final class StorageTests: XCTestCase {
 
   func testSync() throws {
     try storage.setObject(user, forKey: "user")
-    let cachedObject = try storage.object(ofType: User.self, forKey: "user")
+    let cachedObject = try storage.object(forKey: "user")
 
     XCTAssertEqual(cachedObject, user)
   }
@@ -27,7 +31,7 @@ final class StorageTests: XCTestCase {
     let expectation = self.expectation(description: #function)
     storage.async.setObject(user, forKey: "user", expiry: nil, completion: { _ in })
 
-    storage.async.object(ofType: User.self, forKey: "user", completion: { result in
+    storage.async.object(forKey: "user", completion: { result in
       switch result {
       case .value(let cachedUser):
         XCTAssertEqual(cachedUser, self.user)
@@ -50,20 +54,23 @@ final class StorageTests: XCTestCase {
       let lastName: String
     }
 
+    let person1Storage = storage.transformCodable(ofType: Person1.self)
+    let person2Storage = storage.transformCodable(ofType: Person2.self)
+
     // Firstly, save object of type Person1
     let person = Person1(fullName: "John Snow")
 
-    try! storage.setObject(person, forKey: "person")
-    XCTAssertNil(try? storage.object(ofType: Person2.self, forKey: "person"))
+    try! person1Storage.setObject(person, forKey: "person")
+    XCTAssertNil(try? person2Storage.object(forKey: "person"))
 
     // Later, convert to Person2, do the migration, then overwrite
-    let tempPerson = try! storage.object(ofType: Person1.self, forKey: "person")
+    let tempPerson = try! person1Storage.object(forKey: "person")
     let parts = tempPerson.fullName.split(separator: " ")
     let migratedPerson = Person2(firstName: String(parts[0]), lastName: String(parts[1]))
-    try! storage.setObject(migratedPerson, forKey: "person")
+    try! person2Storage.setObject(migratedPerson, forKey: "person")
 
     XCTAssertEqual(
-      try! storage.object(ofType: Person2.self, forKey: "person").firstName,
+      try! person2Storage.object(forKey: "person").firstName,
       "John"
     )
   }
@@ -79,11 +86,14 @@ final class StorageTests: XCTestCase {
       let lastName: String
     }
 
+    let personStorage = storage.transformCodable(ofType: Person.self)
+    let alienStorage = storage.transformCodable(ofType: Alien.self)
+
     let person = Person(firstName: "John", lastName: "Snow")
-    try! storage.setObject(person, forKey: "person")
+    try! personStorage.setObject(person, forKey: "person")
 
     // As long as it has same properties, it works too
-    let cachedObject = try! storage.object(ofType: Alien.self, forKey: "person")
+    let cachedObject = try! alienStorage.object(forKey: "person")
     XCTAssertEqual(cachedObject.firstName, "John")
   }
 }