1 files changed, 7 insertions, 2 deletions

diff --git a/tests/protect_allocator.lua b/tests/protect_allocator.lua
index 5cbb1d8..994cb39 100644
--- a/tests/protect_allocator.lua
+++ b/tests/protect_allocator.lua
@@ -2,6 +2,11 @@ local print = print
 
 local lanes = require "lanes".configure{ with_timers = false, allocator="protected"}
 
+local SLEEP = function(...)
+        local k, v = lanes.sleep(...)
+        assert(k == nil and v == "timeout")
+end
+
 local linda = lanes.linda()
 
 local body = function( id_)
@@ -38,7 +43,7 @@ end
 
 -- wait a bit
 print "waiting a bit ..."
-linda:receive( 2, "foo")
+SLEEP(2)
 -- tell lanes to start running
 print "GO!"
 for i = 1, COUNT do
@@ -49,5 +54,5 @@ end
 print "wait for completion"
 linda:receive( linda.batched, "key", COUNT)
 print "waiting a bit more ..."
-linda:receive( 1, "foo")
+SLEEP(1)
 print "SUCCESS"