From 5824f4e6daaf6a7d4bd529ed8c2e2e46a6bedc21 Mon Sep 17 00:00:00 2001
From: Abhishek <abhishekbanthia@me.com>
Date: Fri, 2 Jul 2021 14:45:49 -0500
Subject: [PATCH] Update ClockerUnitTests.swift

---
 Clocker/ClockerUnitTests/ClockerUnitTests.swift | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/Clocker/ClockerUnitTests/ClockerUnitTests.swift b/Clocker/ClockerUnitTests/ClockerUnitTests.swift
index 903932b..09e0533 100644
--- 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"