dofile("wow_api.lua")

RegisterAddonMessagePrefix = nil	-- make sure we look like a pre-4.1 client (in case someone puts something in wow_api)

dofile("../LibStub/LibStub.lua")
dofile("../CallbackHandler-1.0/CallbackHandler-1.0.lua")
dofile("../AceComm-3.0/ChatThrottleLib.lua")
dofile("../AceComm-3.0/AceComm-3.0.lua")

switches = arg[1] or ""

local VERBOSE,n = strfind(switches, "vv*")	-- "v" anywhere in the first arg
if VERBOSE then VERBOSE = n-VERBOSE+1 end -- "v"=1, "vv"=2, ...




assert(RegisterAddonMessagePrefix == nil)



-----------------------------------------------------------------------
-----------------------------------------------------------------------
-----------------------------------------------------------------------
--
-- Basic AceComm splitting/queueing tests
--

local NOCTL = strfind(switches, "t")

if NOCTL then	-- "t" anywhere in the first arg
	print("NOTE: Testing mode without ChatThrottleLib!")
	-- Replace ChatThrottleLib with a dummy passthrough for testing purposes
	function ChatThrottleLib:SendAddonMessage(prio, prefix, text, chattype, target, queueName)
		self.ORIG_SendAddonMessage(prefix, text, chattype, target)
	end
	function ChatThrottleLib:SendChatMessage(prio, prefix,   text, chattype, language, destination, queueName)
		self.ORIG_SendChatMessage(text, chattype, language, destination)
	end
end


local AceComm = LibStub("AceComm-3.0")

local function printf(format, ...)
	print(format:format(...))
end

local addon1 = {}
local prefix1 = "Test"

local addon2 = {}
local prefix2 = "TestTest"


local data = ""
local received = {}
local chartot = 0

AceComm:Embed(addon1)

local received1 = {}
function addon1:OnCommReceived(prefix, message, distribution, sender)
	assert(self==addon1)
	assert(prefix == prefix1)
	assert(distribution == "RAID", dump(distribution))
	tinsert(received, message)
	tinsert(received1, message)
	assert(#message == #received1)
	assert(message == strsub(data,1,#message))
	chartot=chartot+#message
end

AceComm:Embed(addon2)

local received2 = {}
function addon2:OnCommReceived(prefix, message, distribution, sender)
	assert(self==addon2)
	assert(prefix == prefix2)
	assert(distribution == "GROUP", dump(distribution))
	tinsert(received, message)
	tinsert(received2, message)
	assert(#message == #received2+9)
	assert(message == "OogaBooga"..strsub(data,1,#message-9))
	chartot=chartot+#message
end

addon1:RegisterComm(prefix1)
addon2:RegisterComm(prefix2)


local MSGS=255*4  -- length 1..1000, covers all of: Single, First+Last, First+Next+Last, First+Next+Next+Last
for i = 1,MSGS do
	data = data .. string.char(math.random(32, 255))
end

-- First send a boatload of data without pumping OnUpdates to CTL

for i = 1,MSGS do
	if VERBOSE and VERBOSE>=2 then print("Sending len "..i) end
	addon1:SendCommMessage(prefix1, strsub(data,1,i), "RAID", nil)
	addon2:SendCommMessage(prefix2, "OogaBooga"..strsub(data,1,i), "GROUP", nil)
end

-- Now start pumping OnUpdates; there should be plenty of stuff queued in CTL, and it should all be sent in the right order!
if not NOCTL then
	local sampledmid
	local esttime = ( (MSGS+20)*MSGS*2/1.62 ) / ChatThrottleLib.MAX_CPS
	local midpos = esttime * 0.5
	local latepos = esttime * 0.9

	local lastchartot=0
	local dispinterval=20

	local t=0
	if VERBOSE then
		print("time","rcvtot","rcv1","rcv2","cps","CTL choke")
	end
	while #received < MSGS*2 do
		WoWAPI_FireUpdate(t)
		if t%dispinterval==0 then
			local cps = floor((chartot-lastchartot)/dispinterval)
			if VERBOSE then print(t..": ",#received, #received1, #received2, cps, ChatThrottleLib.bChoking) end
			lastchartot=chartot
			if t>midpos then	-- when our bandwidth isn't mostly eaten by headers and stuff
				assert(cps>=ChatThrottleLib.MAX_CPS*0.6 and cps<ChatThrottleLib.MAX_CPS*1.1, cps)
			end
		end
		if t>midpos and not sampledmid then
			sampledmid=true
			assert(#received > MSGS*1.33 and #received < MSGS*1.5, dump(#received, MSGS*1.33, MSGS*1.5))	-- would be around 1 if we sent the same amount of data in each message, but we send less around the start
		end
		if t>midpos and t<latepos then
			-- prefix2 should have slightly less data transferred since the prefix name itself is longer and uses more bandwidth compared to useful data
			assert(#received2 >= #received1*0.975 and #received2 < #received1*0.99, #received1.." : "..#received2)
		end
		t=t+1
	end

	assert(t>=esttime*0.9 and t<=esttime*1.1, dump(t, esttime))

	assert(sampledmid)

end

assert(#received==MSGS*2)
assert(#received1==MSGS and #received2==MSGS)



-----------------------------------------------------------------------
print "OK"