tests: improve robustness of immediate successes in screen tests

1 files changed, 4 insertions, 4 deletions

diff --git a/test/functional/api/menu_spec.lua b/test/functional/api/menu_spec.lua
index d55b7b118a..2cfa0e3e47 100644
--- a/test/functional/api/menu_spec.lua
+++ b/test/functional/api/menu_spec.lua
@@ -20,15 +20,15 @@ describe("update_menu notification", function()
   end)
 
   local function expect_sent(expected)
-    screen:wait(function()
+    screen:expect{condition=function()
       if screen.update_menu ~= expected then
         if expected then
-          return 'update_menu was expected but not sent'
+          error('update_menu was expected but not sent')
         else
-          return 'update_menu was sent unexpectedly'
+          error('update_menu was sent unexpectedly')
         end
       end
-    end)
+    end, unchanged=(not expected)}
   end
 
   it("should be sent when adding a menu", function()