Defer workflow completion wake until callback unwinds

1 files changed, 6 insertions, 3 deletions

diff --git a/spec/test_luatool.lua b/spec/test_luatool.lua
index ee0a79e..013459f 100644
--- a/spec/test_luatool.lua
+++ b/spec/test_luatool.lua
@@ -27,7 +27,10 @@ local function pump(id)
     for _ = 1, 2000 do
         uv.run("nowait")
         local record = workflow.workflows[id]
-        if record and record.status ~= "running" then return record end
+        if record and record.status ~= "running" then
+            uv.run("nowait") -- deliver the deferred primary wake
+            return record
+        end
         uv.sleep(1)
     end
     error("workflow did not settle: " .. tostring(id), 0)
@@ -113,8 +116,8 @@ return {
             assert(#handle.submissions == 1, "completion wakes the primary once")
             has(handle.submissions[1], id)
             assert(table.concat(handle.emitted, ",") ==
-                "background_work_start,agent_submission,background_work_end",
-                "background work brackets the primary wake")
+                "background_work_start,background_work_end,agent_submission",
+                "the primary wake runs after the completion callback unwinds")
         end)
     end },