Merge branch 'master' of https://github.com/n0shake/Clocker

n0shake · Jul 2, 2021 · df4cff7 · df4cff7
2 parents 2d21de0 + 5824f4e
commit df4cff7
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/Clocker/ClockerUnitTests/ClockerUnitTests.swift b/Clocker/ClockerUnitTests/ClockerUnitTests.swift
@@ -115,8 +115,21 @@ class ClockerUnitTests: XCTestCase {
             XCTFail("Default preferences aren't in the correct format")
             return
         }
-        let oldCount = currentFavourites.count
-
+        // Check if timezone with test identifier is present.
+        let filteredCount = currentFavourites.filter {
+            let timezone = TimezoneData.customObject(from: $0)
+            return timezone?.placeID == "TestIdentifier"
+        }
+
+        // California is absent. Add it!
+        if filteredCount.count == 0 {
+            let timezoneData = TimezoneData(with: california)
+            let operationsObject = TimezoneDataOperations(with: timezoneData)
+            operationsObject.saveObject()
+        }
+
+        let oldCount = (defaults.object(forKey: CLDefaultPreferenceKey) as? [Data])?.count ?? 0
+
         currentFavourites = currentFavourites.filter {
             let timezone = TimezoneData.customObject(from: $0)
             return timezone?.placeID != "TestIdentifier"