Inject calendar to make tests reliable

ullas-jain · Nov 21, 2023 · 112bc3a · 112bc3a
1 parent bd099f5
commit 112bc3a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/EssentialFeedTests/Helpers/SharedTestHelpers.swift b/EssentialFeedTests/Helpers/SharedTestHelpers.swift
@@ -37,11 +37,11 @@ extension Date {
         return self + seconds
     }
 
-    func adding(minutes: Int) -> Date {
-        return Calendar(identifier: .gregorian).date(byAdding: .minute, value: minutes, to: self)!
+    func adding(minutes: Int, calendar: Calendar = Calendar(identifier: .gregorian)) -> Date {
+        return calendar.date(byAdding: .minute, value: minutes, to: self)!
     }
 
-    func adding(days: Int) -> Date {
-        return Calendar(identifier: .gregorian).date(byAdding: .day, value: days, to: self)!
+    func adding(days: Int, calendar: Calendar = Calendar(identifier: .gregorian)) -> Date {
+        return calendar.date(byAdding: .day, value: days, to: self)!
     }
 }
diff --git a/EssentialFeedTests/Image Comments Presentation/ImageCommentsPresenterTests.swift b/EssentialFeedTests/Image Comments Presentation/ImageCommentsPresenterTests.swift
@@ -22,13 +22,13 @@ class ImageCommentsPresenterTests: XCTestCase {
             ImageComment(
                 id: UUID(),
                 message: "a message",
-                createdAt: now.adding(minutes: -5),
+                createdAt: now.adding(minutes: -5, calendar: calendar),
                 username: "a username"
             ),
             ImageComment(
                 id: UUID(),
                 message: "another message",
-                createdAt: now.adding(days: -1),
+                createdAt: now.adding(days: -1, calendar: calendar),
                 username: "another username"
             ),
         ]