diff options
Diffstat (limited to 'spec/aio_spec.lua')
| -rw-r--r-- | spec/aio_spec.lua | 272 |
1 files changed, 136 insertions, 136 deletions
diff --git a/spec/aio_spec.lua b/spec/aio_spec.lua index 75e4a0a..093ce59 100644 --- a/spec/aio_spec.lua +++ b/spec/aio_spec.lua @@ -1,136 +1,136 @@ ---[[
-test the asyncronous parts of nnng
-]]
-local nng = require("nng")
-local lanes = require("lanes").configure()
-describe("nng.aio",function()
- it("should be able to create aio object",function()
- print("one")
- local aio = assert(nng.aio.alloc(function() end))
- end)
- it("should accept a callback with arguments",function()
- print("two")
- local aio = assert(nng.aio.alloc(function(a,b,c)
- print("hello")
- end, "one", 2, "three"))
- end)
- --it("should call the callback after sleeping", function()
- --print("three")
- --local called = false
- --local callback = function(lock)
- --called = true
- --end
- --print("about to alloc")
- --local aio = assert(nng.aio.alloc(callback))
- --print("done alloc")
- --print("about to sleep")
- --nng.aio.sleep(1,aio)
- --print("whatever:",whatever)
- ----nng.aio.sleep(1,aio)
- --print("done sleep")
- --os.execute("sleep 1")
- --lock(-1)
- --print("checking called...")
- --print("called was",called)
- --assert(called)
- --end)
- it("should call more than one socket getting a callback at once",function()
- print("checking recv_any callback")
- local s1 = assert(nng.bus0_open())
- assert(s1:listen("tcp://127.0.0.1:4000"))
-
- local s2 = assert(nng.bus0_open())
- assert(s2:listen("tcp://127.0.0.1:4001"))
-
- local s3 = assert(nng.bus0_open())
- assert(s3:dial("tcp://127.0.0.1:4000"))
- assert(s3:dial("tcp://127.0.0.1:4001"))
-
- for i = 1, 100 do --100 times to try to trigger race conditions
- --print("i:",i)
- assert(s3:send("one"))
- assert(s3:send("two"))
- local s1_got_one, s1_got_two, s2_got_one, s2_got_two = false, false, false, false
- while not (s1_got_one and s1_got_two and s2_got_one and s2_got_two) do
- --local socket, message = nng.aio.recv_any(s1, s2)
- --print("about to start recv_any")
- --local tbl = nng.aio.recv_any(s1,s2)
- for socket, message in pairs(nng.aio.recv_any(s1,s2)) do
- --print("in one recv any:",socket, message)
- if socket == s1 then
- if message == "one" then
- s1_got_one = true
- elseif message == "two" then
- s1_got_two = true
- else
- error("message socket 1:" .. message)
- end
- elseif socket == s2 then
- if message == "one" then
- s2_got_one = true
- elseif message == "two" then
- s2_got_two = true
- else
- error("message socket 2:" .. message)
- end
- else
- error("socket:" .. tostring(socket))
- end
- --print("done with recv_any",s1_got_one, s1_got_two, s2_got_one, s2_got_two)
- end
- end
- assert(s1_got_one)
- assert(s1_got_two)
- assert(s2_got_one)
- assert(s2_got_two)
- end
- print("Sucessful completion of recv_any test")
- end)
- --it("Should accept multiple sockets of dispirate types #writing",function()
- --local servers = {}
- --local clients = {}
- --local messages = {}
- --local halfs = {
- --{"rep","req",true},
- --{"bus","bus",nil},
- --{"surveyor","respondent",false},
- --{"pub","sub",false}
- --}
- --for i = 1, 10 do
- --local rng = math.random(#halfs)
-
- ----Create server
- --servers[i] = assert(nng[halfs[rng][1] .. "0_open"]())
- --local url = string.format("tcp://127.0.0.1:%d",4000 + i)
- --assert(servers[i]:listen(url))
-
- ----Create clients
- --local numclients = math.random(1,3)
- --clients[i] = {}
- --messages[i] = {}
- --for j = 1, numclients do
- --clients[i][j] = assert(nng[halfs[rng][2] .. "0_open"]())
- --assert(clients[i][j]:dial(url))
- --messages[i][j] = {}
- --if halfs[rng][3] == true then
- ----we send messages
- --local nummessages = math.random(5)
- --for k = 1, nummessages do
- --local message = string.format("ping_%d_%d_%d",i,j,k)
- --assert(clients[i][j]:send(message))
- --table.insert(messages[i][j],message)
- --end
- ----elseif halfs[rng][3] == nil then
- ----we can send or receive messages
-
- --else
- ----we receive and reply to messages
-
- --end
- --end
- --end
- --for socket, message in pairs(nng.aio.recv_any(table.unpack(servers))) do
- --print("GOT MESSAGE:",message)
- --end
- --end)
-end)
+--[[ +test the asyncronous parts of nnng +]] +local nng = require("nng") +local lanes = require("lanes").configure() +describe("nng.aio",function() + it("should be able to create aio object",function() + print("one") + local aio = assert(nng.aio.alloc(function() end)) + end) + it("should accept a callback with arguments",function() + print("two") + local aio = assert(nng.aio.alloc(function(a,b,c) + print("hello") + end, "one", 2, "three")) + end) + --it("should call the callback after sleeping", function() + --print("three") + --local called = false + --local callback = function(lock) + --called = true + --end + --print("about to alloc") + --local aio = assert(nng.aio.alloc(callback)) + --print("done alloc") + --print("about to sleep") + --nng.aio.sleep(1,aio) + --print("whatever:",whatever) + ----nng.aio.sleep(1,aio) + --print("done sleep") + --os.execute("sleep 1") + --lock(-1) + --print("checking called...") + --print("called was",called) + --assert(called) + --end) + it("should call more than one socket getting a callback at once",function() + print("checking recv_any callback") + local s1 = assert(nng.bus0_open()) + assert(s1:listen("tcp://127.0.0.1:4000")) + + local s2 = assert(nng.bus0_open()) + assert(s2:listen("tcp://127.0.0.1:4001")) + + local s3 = assert(nng.bus0_open()) + assert(s3:dial("tcp://127.0.0.1:4000")) + assert(s3:dial("tcp://127.0.0.1:4001")) + + for i = 1, 100 do --100 times to try to trigger race conditions + --print("i:",i) + assert(s3:send("one")) + assert(s3:send("two")) + local s1_got_one, s1_got_two, s2_got_one, s2_got_two = false, false, false, false + while not (s1_got_one and s1_got_two and s2_got_one and s2_got_two) do + --local socket, message = nng.aio.recv_any(s1, s2) + --print("about to start recv_any") + --local tbl = nng.aio.recv_any(s1,s2) + for socket, message in pairs(nng.aio.recv_any(s1,s2)) do + --print("in one recv any:",socket, message) + if socket == s1 then + if message == "one" then + s1_got_one = true + elseif message == "two" then + s1_got_two = true + else + error("message socket 1:" .. message) + end + elseif socket == s2 then + if message == "one" then + s2_got_one = true + elseif message == "two" then + s2_got_two = true + else + error("message socket 2:" .. message) + end + else + error("socket:" .. tostring(socket)) + end + --print("done with recv_any",s1_got_one, s1_got_two, s2_got_one, s2_got_two) + end + end + assert(s1_got_one) + assert(s1_got_two) + assert(s2_got_one) + assert(s2_got_two) + end + print("Sucessful completion of recv_any test") + end) + --it("Should accept multiple sockets of dispirate types #writing",function() + --local servers = {} + --local clients = {} + --local messages = {} + --local halfs = { + --{"rep","req",true}, + --{"bus","bus",nil}, + --{"surveyor","respondent",false}, + --{"pub","sub",false} + --} + --for i = 1, 10 do + --local rng = math.random(#halfs) + + ----Create server + --servers[i] = assert(nng[halfs[rng][1] .. "0_open"]()) + --local url = string.format("tcp://127.0.0.1:%d",4000 + i) + --assert(servers[i]:listen(url)) + + ----Create clients + --local numclients = math.random(1,3) + --clients[i] = {} + --messages[i] = {} + --for j = 1, numclients do + --clients[i][j] = assert(nng[halfs[rng][2] .. "0_open"]()) + --assert(clients[i][j]:dial(url)) + --messages[i][j] = {} + --if halfs[rng][3] == true then + ----we send messages + --local nummessages = math.random(5) + --for k = 1, nummessages do + --local message = string.format("ping_%d_%d_%d",i,j,k) + --assert(clients[i][j]:send(message)) + --table.insert(messages[i][j],message) + --end + ----elseif halfs[rng][3] == nil then + ----we can send or receive messages + + --else + ----we receive and reply to messages + + --end + --end + --end + --for socket, message in pairs(nng.aio.recv_any(table.unpack(servers))) do + --print("GOT MESSAGE:",message) + --end + --end) +end) |