chore(kv) fix flaky queue test (#19495)

1 files changed, 10 insertions, 2 deletions

diff --git a/cli/tests/unit/kv_test.ts b/cli/tests/unit/kv_test.ts
index edb269e6e..439dcfe2c 100644
--- a/cli/tests/unit/kv_test.ts
+++ b/cli/tests/unit/kv_test.ts
@@ -1698,9 +1698,9 @@ Deno.test({
 
 Deno.test({
   name: "queue persistence with delay messages",
-  sanitizeOps: false,
-  sanitizeResources: false,
   async fn() {
+    const dispatchedPre = Deno.metrics().opsDispatchedAsync;
+    const completedPre = Deno.metrics().opsCompletedAsync;
     const filename = "cli/tests/testdata/queue.db";
     try {
       await Deno.remove(filename);
@@ -1745,6 +1745,14 @@ Deno.test({
       db.close();
       await listener;
     } finally {
+      // Wait until callbacks are drained before deleting the db.
+      let dispatched = Deno.metrics().opsDispatchedAsync - dispatchedPre;
+      let completed = Deno.metrics().opsCompletedAsync - completedPre;
+      while (dispatched !== completed) {
+        dispatched = Deno.metrics().opsDispatchedAsync - dispatchedPre;
+        completed = Deno.metrics().opsCompletedAsync - completedPre;
+        await sleep(100);
+      }
       await Deno.remove(filename);
     }
   },