Fix TestClock.sleep hang when sleeping until deadline that is equal t…

…o now (#22)
pointfreeco · Sep 26, 2023 · 2f8a1b0 · 2f8a1b0
1 parent d1fd837
commit 2f8a1b0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/Sources/Clocks/TestClock.swift b/Sources/Clocks/TestClock.swift
@@ -105,7 +105,7 @@
       let id = UUID()
       do {
         let stream: AsyncThrowingStream<Never, Error>? = self.lock.sync {
-          guard deadline >= self.now
+          guard deadline > self.now
           else {
             return nil
           }

diff --git a/Tests/ClocksTests/TestClocksTests.swift b/Tests/ClocksTests/TestClocksTests.swift
@@ -26,6 +26,13 @@ final class TestClockTests: XCTestCase, @unchecked Sendable {
     XCTAssertEqual(checkIsFinished, true)
   }
 
+  func testSleepUntilExactlyNow() async throws {
+    let duration: Duration = .seconds(1)
+    let deadline = clock.now.advanced(by: duration)
+    await clock.advance(by: duration)
+    try await clock.sleep(until: deadline)
+  }
+
   func testAdvanceWithReentrantUnitsOfWork() async throws {
     let task = Task {
       var count = 0