New method linda:restrict()

2 files changed, 10 insertions, 3 deletions

diff --git a/tests/cancel.lua b/tests/cancel.lua
index 1984c85..e65d794 100644
--- a/tests/cancel.lua
+++ b/tests/cancel.lua
@@ -39,10 +39,17 @@ if not next(which_tests) or which_tests.genlock then
         local lock = lanes.genlock( linda, "lock", 1)
         local atomic = lanes.genatomic( linda, "atomic")
 
+        local check_returned_cancel_error = function(_status, _err)
+                assert(_status == nil and _err == lanes.cancel_error)
+        end
         -- check that cancelled lindas give cancel_error as they should
         linda:cancel()
-        local _status, _err = linda:get( "empty")
-        assert(_status == nil and _err == lanes.cancel_error)
+        check_returned_cancel_error(linda:set( "empty", 42))
+        check_returned_cancel_error(linda:get( "empty"))
+        check_returned_cancel_error(linda:send( "empty", 42))
+        check_returned_cancel_error(linda:receive( "empty"))
+        check_returned_cancel_error(linda:limit( "empty", 5))
+        check_returned_cancel_error(linda:restrict( "empty", "set/get"))
         assert( lanes.genlock( linda, "any", 1) == lanes.cancel_error)
         assert( lanes.genatomic( linda, "any") == lanes.cancel_error)
 
diff --git a/tests/fifo.lua b/tests/fifo.lua
index e1bfeae..9efcbd9 100644
--- a/tests/fifo.lua
+++ b/tests/fifo.lua
@@ -60,7 +60,7 @@ for key,count in pairs(fifo_linda:count("unknown key", A.channel, "unknown key",
 end
 print "Dumping linda stats.. [4]" -- count everything
 for key,contents in pairs(fifo_linda:dump()) do
-    print("channel " .. key .. ": limit=".. contents.limit, " first=" .. contents.first, " count=" .. contents.count)
+    print("channel " .. key .. ": limit=".. contents.limit, " first=" .. contents.first, " count=" .. contents.count .. " restrict=" .. contents.restrict)
     for k,v in pairs(contents.fifo) do
         print("[".. k.."] = " .. v)
     end