--[[
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)