1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
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)
|
