fix unit tests

Author: roninopf

tests/common/CommonMocks.swift

@@ -446,16 +446,17 @@ class MockLocalStorage: LocalStorageProtocol {
         attributionInfoExpiration = expiration
     }
 
-    func getPayload(currentDate: Date) -> [AnyHashable : Any]? {
+    func getLastPushPayload(_ currentDate: Date) -> [AnyHashable : Any]? {
         guard !MockLocalStorage.isExpired(expiration: payloadExpiration, currentDate: currentDate) else {
             return nil
         }
+        
         return payload
     }
 
-    func save(payload: [AnyHashable : Any]?, withExpiration: Date?) {
+    func saveLastPushPayload(_ payload: [AnyHashable : Any]?, withExpiration expiration: Date?) {
         self.payload = payload
-        payloadExpiration = withExpiration
+        payloadExpiration = expiration
     }
 
     private var payload: [AnyHashable: Any]? = nil

tests/unit-tests/LocalStorageTests.swift

@@ -114,16 +114,21 @@ class LocalStorageTests: XCTestCase {
                 ],
             ],
         ]
+        
         let currentDate = Date()
         let expiration = Calendar.current.date(byAdding: Calendar.Component.hour, value: 24, to: currentDate)!
-        localStorage.save(payload: payload, withExpiration: expiration)
+        
+        localStorage.saveLastPushPayload(payload, withExpiration: expiration)
+        
         // 23 hours, not expired, still present
         mockDateProvider.currentDate = Calendar.current.date(byAdding: Calendar.Component.hour, value: 23, to: currentDate)!
-        let fromLocalStorage: [AnyHashable: Any] = localStorage.getPayload(currentDate: mockDateProvider.currentDate)!
+        let fromLocalStorage: [AnyHashable: Any] = localStorage.getLastPushPayload(mockDateProvider.currentDate)!
+        
         XCTAssertTrue(NSDictionary(dictionary: payload).isEqual(to: fromLocalStorage))
 
         mockDateProvider.currentDate = Calendar.current.date(byAdding: Calendar.Component.hour, value: 25, to: currentDate)!
-        let fromLocalStorage2: [AnyHashable: Any]? = localStorage.getPayload(currentDate: mockDateProvider.currentDate)
+        let fromLocalStorage2: [AnyHashable: Any]? = localStorage.getLastPushPayload(mockDateProvider.currentDate)
+        
         XCTAssertNil(fromLocalStorage2)
     }
 
@@ -150,13 +155,16 @@ class LocalStorageTests: XCTestCase {
             ],
             "someClass": A(),
         ]
+        
         let currentDate = Date()
         let expiration = Calendar.current.date(byAdding: Calendar.Component.hour, value: 24, to: currentDate)!
-        localStorage.save(payload: payload, withExpiration: expiration)
+        
+        localStorage.saveLastPushPayload(payload, withExpiration: expiration)
 
         // 23 hours, not expired, still present
         mockDateProvider.currentDate = Calendar.current.date(byAdding: Calendar.Component.hour, value: 23, to: currentDate)!
-        let fromLocalStorage = localStorage.getPayload(currentDate: mockDateProvider.currentDate)
+        let fromLocalStorage = localStorage.getLastPushPayload(mockDateProvider.currentDate)
+        
         XCTAssertNil(fromLocalStorage)
     }