SleepIfNeeded unit test

tact · Aug 10, 2024 · db4c537 · db4c537
1 parent ced84f7
commit db4c537
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/Targets/Canopy/Tests/ReplayingMockDatabaseTests.swift b/Targets/Canopy/Tests/ReplayingMockDatabaseTests.swift
@@ -395,4 +395,21 @@ final class ReplayingMockDatabaseTests: XCTestCase {
     let result2 = try! await db.fetchRecords(with: []).get()
     XCTAssertEqual(result2.notFoundRecordIDs[0].recordName, "notFound1")
   }
+
+  func test_sleep_if_needed() async {
+    let db = ReplayingMockDatabase(
+      operationResults: [
+        .queryRecords(
+          .init(
+            result: .success([
+              .mock(.init(recordID: .init(recordName: "mockName"), recordType: "MockType"))
+            ])
+          )
+        )
+      ],
+      sleepBeforeEachOperation: 0.01
+    )
+    let result = try! await db.queryRecords(with: .init(recordType: "MockType", predicate: NSPredicate(value: true)), in: nil).get()
+    XCTAssertEqual(result.first!.recordID.recordName, "mockName")
+  }
 }