No ticket: [refactor] Combined Cloisimis's TimeInterval.durationStrin…

…g extension with the TimeIntervalExtensions.
jrtibbetts · Jul 31, 2021 · e148ff6 · e148ff6
1 parent a59b2e9
commit e148ff6
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/Source/Util/TimeIntervalExtensions.swift b/Source/Util/TimeIntervalExtensions.swift
@@ -12,4 +12,16 @@ public extension TimeInterval {
     var weeks:   TimeInterval { return   7.0 * days    }
     var years:   TimeInterval { return 365.0 * days    }
 
+    var formattedString: String? {
+        return Self.durationFormatter.string(from: self)
+    }
+
+    static var durationFormatter: DateComponentsFormatter = {
+        let formatter = DateComponentsFormatter()
+        formatter.allowedUnits = [.hour, .minute, .second]
+        formatter.collapsesLargestUnit = true
+
+        return formatter
+    }()
+
 }
diff --git a/Tests/Util/TimeIntervalExtensionsTests.swift b/Tests/Util/TimeIntervalExtensionsTests.swift
@@ -13,4 +13,24 @@ class TimeIntervalExtensionsTests: XCTestCase {
         XCTAssertEqual(2.years, 104.weeks + 2.days)
     }
 
+    // MARK: - durationString
+
+    func testFormattedStringDropsHours() {
+        let timeInterval = 165.0000 // 2:45
+        let durationString = timeInterval.formattedString
+        XCTAssertEqual(durationString, "2:45")
+    }
+
+    func testFormattedStringIncludesHours() {
+        let timeInterval = (3.0 * 60.0 * 60.0) + 165.0000 // 3:02:45
+        let durationString = timeInterval.formattedString
+        XCTAssertEqual(durationString, "3:02:45")
+    }
+
+    func testFormattedStringHandlesNegativeValues() {
+        let timeInterval = -165.0000 // -2:45
+        let durationString = timeInterval.formattedString
+        XCTAssertEqual(durationString, "-2:45")
+    }
+
 }