File

tests/test_util_xmppstream.lua @ 7567:495de404a8ae

ejabberdsql2prosody: rename variable 'host' to prevent shadowing upvalue [luacheck] Functions roster(), roster_pending(), roster_group(), private_storage() and offline_msg() have argument named "host", which used to shadow upvalue of this variable before this change. Instead of renaming this argument, let's rename the variable to match what the script says in usage: Usage: ejabberdsql2prosody.lua filename.txt hostname
author Anton Shestakov <av6@dwimlabs.net>
date Fri, 12 Aug 2016 13:44:47 +0800
parent 7241:5e7797822f19
line wrap: on
line source

function new(new_stream, _M)
	local function test(xml, expect_success, ex)
		local stanzas = {};
		local session = { notopen = true };
		local callbacks = {
			stream_ns = "streamns";
			stream_tag = "stream";
			default_ns = "stanzans";
			streamopened = function (_session)
				assert_equal(session, _session);
				assert_equal(session.notopen, true);
				_session.notopen = nil;
				return true;
			end;
			handlestanza = function (_session, stanza)
				assert_equal(session, _session);
				assert_equal(_session.notopen, nil);
				table.insert(stanzas, stanza);
			end;
			streamclosed = function (_session)
				assert_equal(session, _session);
				assert_equal(_session.notopen, nil);
				_session.notopen = nil;
			end;
		}
		if type(ex) == "table" then
			for k, v in pairs(ex) do
				if k ~= "_size_limit" then
					callbacks[k] = v;
				end
			end
		end
		local stream = new_stream(session, callbacks, size_limit);
		local ok, err = pcall(function ()
			assert(stream:feed(xml));
		end);

		if ok and type(expect_success) == "function" then
			expect_success(stanzas);
		end
		assert_equal(not not ok, not not expect_success, "Expected "..(expect_success and ("success ("..tostring(err)..")") or "failure"));
	end

	local function test_stanza(stanza, expect_success, ex)
		return test([[<stream:stream xmlns:stream="streamns" xmlns="stanzans">]]..stanza, expect_success, ex);
	end

	test([[<stream:stream xmlns:stream="streamns"/>]], true);
	test([[<stream xmlns="streamns"/>]], true);

	test([[<stream1 xmlns="streamns"/>]], false);
	test([[<stream xmlns="streamns1"/>]], false);
	test("<>", false);

	test_stanza("<message/>", function (stanzas)
		assert_equal(#stanzas, 1);
		assert_equal(stanzas[1].name, "message");
	end);
	test_stanza("< message>>>>/>\n", false);

	test_stanza([[<x xmlns:a="b">
		<y xmlns:a="c">
			<a:z/>
		</y>
		<a:z/>
	</x>]], function (stanzas)
		assert_equal(#stanzas, 1);
		local s = stanzas[1];
		assert_equal(s.name, "x");
		assert_equal(#s.tags, 2);

		assert_equal(s.tags[1].name, "y");
		assert_equal(s.tags[1].attr.xmlns, nil);

		assert_equal(s.tags[1].tags[1].name, "z");
		assert_equal(s.tags[1].tags[1].attr.xmlns, "c");

		assert_equal(s.tags[2].name, "z");
		assert_equal(s.tags[2].attr.xmlns, "b");

		assert_equal(s.namespaces, nil);
	end);
end