clock test

swhitty · Sep 8, 2024 · f93f952 · f93f952
1 parent 1896b58
commit f93f952
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/Tests/TimeoutTests.swift b/Tests/TimeoutTests.swift
@@ -119,6 +119,24 @@ struct TimeoutTests {
             try await TestActor("Fish").returningValue(after: 0.1, before: .now)
         }
     }
+
+    @Test
+    func returnsValueWithClock_beforeDeadlineExpires() async throws {
+        #expect(
+            try await withThrowingTimeout(after: .now + .seconds(2), clock: ContinuousClock()) {
+                "Fish"
+            } == "Fish"
+        )
+    }
+
+    @Test
+    func throwsErrorWithClock_WhenDeadlineExpires() async {
+        await #expect(throws: TimeoutError.self) {
+            try await withThrowingTimeout(after: .now, clock: ContinuousClock()) {
+                try await Task.sleep(for: .seconds(2))
+            }
+        }
+    }
 }
 
 public struct NonSendable<T> {