local st = require "util.stanza";

local errors = require "util.error";

describe("util.stanza", function()

	describe("#preserialize()", function()

		it("should work", function()

			local stanza = st.stanza("message", { type = "chat" }):text_tag("body", "Hello");

			local stanza2 = st.preserialize(stanza);

			assert.is_table(stanza2, "Preserialized stanza is a table");

			assert.is_nil(getmetatable(stanza2), "Preserialized stanza has no metatable");

			assert.is_string(stanza2.name, "Preserialized stanza has a name field");

			assert.equal(stanza.name, stanza2.name, "Preserialized stanza has same name as the input stanza");

			assert.same(stanza.attr, stanza2.attr, "Preserialized stanza same attr table as input stanza");

			assert.is_nil(stanza2.tags, "Preserialized stanza has no tag list");

			assert.is_nil(stanza2.last_add, "Preserialized stanza has no last_add marker");

			assert.is_table(stanza2[1], "Preserialized child element preserved");

			assert.equal("body", stanza2[1].name, "Preserialized child element name preserved");

		end);

	end);

	describe("#deserialize()", function()

		it("should work", function()

			local stanza = { name = "message", attr = { type = "chat" }, { name = "body", attr = { }, "Hello" } };

			local stanza2 = st.deserialize(st.preserialize(stanza));

			assert.is_table(stanza2, "Deserialized stanza is a table");

			assert.equal(st.stanza_mt, getmetatable(stanza2), "Deserialized stanza has stanza metatable");

			assert.is_string(stanza2.name, "Deserialized stanza has a name field");

			assert.equal(stanza.name, stanza2.name, "Deserialized stanza has same name as the input table");

			assert.same(stanza.attr, stanza2.attr, "Deserialized stanza same attr table as input table");

			assert.is_table(stanza2.tags, "Deserialized stanza has tag list");

			assert.is_table(stanza2[1], "Deserialized child element preserved");

			assert.equal("body", stanza2[1].name, "Deserialized child element name preserved");

		end);

	end);

	describe("#stanza()", function()

		it("should work", function()

			local s = st.stanza("foo", { xmlns = "myxmlns", a = "attr-a" });

			assert.are.equal(s.name, "foo");

			assert.are.equal(s.attr.xmlns, "myxmlns");

			assert.are.equal(s.attr.a, "attr-a");

			local s1 = st.stanza("s1");

			assert.are.equal(s1.name, "s1");

			assert.are.equal(s1.attr.xmlns, nil);

			assert.are.equal(#s1, 0);

			assert.are.equal(#s1.tags, 0);

			s1:tag("child1");

			assert.are.equal(#s1.tags, 1);

			assert.are.equal(s1.tags[1].name, "child1");

			s1:tag("grandchild1"):up();

			assert.are.equal(#s1.tags, 1);

			assert.are.equal(s1.tags[1].name, "child1");

			assert.are.equal(#s1.tags[1], 1);

			assert.are.equal(s1.tags[1][1].name, "grandchild1");

			s1:up():tag("child2");

			assert.are.equal(#s1.tags, 2, tostring(s1));

			assert.are.equal(s1.tags[1].name, "child1");

			assert.are.equal(s1.tags[2].name, "child2");

			assert.are.equal(#s1.tags[1], 1);

			assert.are.equal(s1.tags[1][1].name, "grandchild1");

			s1:up():text("Hello world");

			assert.are.equal(#s1.tags, 2);

			assert.are.equal(#s1, 3);

			assert.are.equal(s1.tags[1].name, "child1");

			assert.are.equal(s1.tags[2].name, "child2");

			assert.are.equal(#s1.tags[1], 1);

			assert.are.equal(s1.tags[1][1].name, "grandchild1");

		end);

		it("should work with unicode values", function ()

			local s = st.stanza("Объект", { xmlns = "myxmlns", ["Объект"] = "&" });

			assert.are.equal(s.name, "Объект");

			assert.are.equal(s.attr.xmlns, "myxmlns");

			assert.are.equal(s.attr["Объект"], "&");

		end);

		it("should allow :text() with nil and empty strings", function ()

			local s_control = st.stanza("foo");

			assert.same(st.stanza("foo"):text(), s_control);

			assert.same(st.stanza("foo"):text(nil), s_control);

			assert.same(st.stanza("foo"):text(""), s_control);

		end);

		it("validates names", function ()

			assert.has_error_match(function ()

				st.stanza("invalid\0name");

			end, "invalid tag name:")

			assert.has_error_match(function ()

				st.stanza("name", { ["foo\1\2\3bar"] = "baz" });

			end, "invalid attribute name: contains control characters")

			assert.has_error_match(function ()

				st.stanza("name", { ["foo"] = "baz\1\2\3\255moo" });

			end, "invalid attribute value: contains control characters")

		end)

		it("validates types", function ()

			assert.has_error_match(function ()

				st.stanza(1);

			end, "invalid tag name: expected string, got number")

			assert.has_error_match(function ()

				st.stanza("name", "string");

			end, "invalid attributes: expected table, got string")

			assert.has_error_match(function ()

				st.stanza("name",{1});

			end, "invalid attribute name: expected string, got number")

			assert.has_error_match(function ()

				st.stanza("name",{foo=1});

			end, "invalid attribute value: expected string, got number")

		end)

	end);

	describe("#message()", function()

		it("should work", function()

			local m = st.message();

			assert.are.equal(m.name, "message");

		end);

	end);

	describe("#iq()", function()

		it("should create an iq stanza", function()

			local i = st.iq({ type = "get", id = "foo" });

			assert.are.equal("iq", i.name);

			assert.are.equal("foo", i.attr.id);

			assert.are.equal("get", i.attr.type);

		end);

		it("should reject stanzas with no attributes", function ()

			assert.has.error_match(function ()

				st.iq();

			end, "attributes");

		end);

		it("should reject stanzas with no id", function ()

			assert.has.error_match(function ()

				st.iq({ type = "get" });

			end, "id attribute");

		end);

		it("should reject stanzas with no type", function ()

			assert.has.error_match(function ()

				st.iq({ id = "foo" });

			end, "type attribute");

		end);

	end);

	describe("#presence()", function ()

		it("should work", function()

			local p = st.presence();

			assert.are.equal(p.name, "presence");

		end);

	end);

	describe("#reply()", function()

		it("should work for <s>", function()

			-- Test stanza

			local s = st.stanza("s", { to = "touser", from = "fromuser", id = "123" })

				:tag("child1");

			-- Make reply stanza

			local r = st.reply(s);

			assert.are.equal(r.name, s.name);

			assert.are.equal(r.id, s.id);

			assert.are.equal(r.attr.to, s.attr.from);

			assert.are.equal(r.attr.from, s.attr.to);

			assert.are.equal(#r.tags, 0, "A reply should not include children of the original stanza");

		end);

		it("should work for <iq get>", function()

			-- Test stanza

			local s = st.stanza("iq", { to = "touser", from = "fromuser", id = "123", type = "get" })

				:tag("child1");

			-- Make reply stanza

			local r = st.reply(s);

			assert.are.equal(r.name, s.name);

			assert.are.equal(r.id, s.id);

			assert.are.equal(r.attr.to, s.attr.from);

			assert.are.equal(r.attr.from, s.attr.to);

			assert.are.equal(r.attr.type, "result");

			assert.are.equal(#r.tags, 0, "A reply should not include children of the original stanza");

		end);

		it("should work for <iq set>", function()

			-- Test stanza

			local s = st.stanza("iq", { to = "touser", from = "fromuser", id = "123", type = "set" })

				:tag("child1");

			-- Make reply stanza

			local r = st.reply(s);

			assert.are.equal(r.name, s.name);

			assert.are.equal(r.id, s.id);

			assert.are.equal(r.attr.to, s.attr.from);

			assert.are.equal(r.attr.from, s.attr.to);

			assert.are.equal(r.attr.type, "result");

			assert.are.equal(#r.tags, 0, "A reply should not include children of the original stanza");

		end);

		it("should reject not-stanzas", function ()

			assert.has.error_match(function ()

				st.reply(not "a stanza");

			end, "expected stanza");

		end);

		it("should reject not-stanzas", function ()

			assert.has.error_match(function ()

				st.reply({name="x"});

			end, "expected stanza");

		end);

	end);

	describe("#error_reply()", function()

		it("should work for <s>", function()

			-- Test stanza

			local s = st.stanza("s", { to = "touser", from = "fromuser", id = "123" })

				:tag("child1");

			-- Make reply stanza

			local r = st.error_reply(s, "cancel", "service-unavailable", nil, "host");

			assert.are.equal(r.name, s.name);

			assert.are.equal(r.id, s.id);

			assert.are.equal(r.attr.to, s.attr.from);

			assert.are.equal(r.attr.from, s.attr.to);

			assert.are.equal(#r.tags, 1);

			assert.are.equal(r.tags[1].tags[1].name, "service-unavailable");

			assert.are.equal(r.tags[1].attr.by, "host");

		end);

		it("should work for <iq get>", function()

			-- Test stanza

			local s = st.stanza("iq", { to = "touser", from = "fromuser", id = "123", type = "get" })

				:tag("child1");

			-- Make reply stanza

			local r = st.error_reply(s, "cancel", "service-unavailable");

			assert.are.equal(r.name, s.name);

			assert.are.equal(r.id, s.id);

			assert.are.equal(r.attr.to, s.attr.from);

			assert.are.equal(r.attr.from, s.attr.to);

			assert.are.equal(r.attr.type, "error");

			assert.are.equal(#r.tags, 1);

			assert.are.equal(r.tags[1].tags[1].name, "service-unavailable");

		end);

		it("should reject not-stanzas", function ()

			assert.has.error_match(function ()

				st.error_reply(not "a stanza", "modify", "bad-request");

			end, "expected stanza");

		end);

		it("should reject stanzas of type error", function ()

			assert.has.error_match(function ()

				st.error_reply(st.message({type="error"}), "cancel", "conflict");

			end, "got stanza of type error");

			assert.has.error_match(function ()

				st.error_reply(st.error_reply(st.message({type="chat"}), "modify", "forbidden"), "cancel", "service-unavailable");

			end, "got stanza of type error");

		end);

		describe("util.error integration", function ()

		it("should accept util.error objects", function ()

			local s = st.message({ to = "touser", from = "fromuser", id = "123", type = "chat" }, "Hello");

			local e = errors.new({ type = "modify", condition = "not-acceptable", text = "Bork bork bork" }, { by = "this.test" });

			local r = st.error_reply(s, e);

			assert.are.equal(r.name, s.name);

			assert.are.equal(r.id, s.id);

			assert.are.equal(r.attr.to, s.attr.from);

			assert.are.equal(r.attr.from, s.attr.to);

			assert.are.equal(r.attr.type, "error");

			assert.are.equal(r.tags[1].name, "error");

			assert.are.equal(r.tags[1].attr.type, e.type);

			assert.are.equal(r.tags[1].tags[1].name, e.condition);

			assert.are.equal(r.tags[1].tags[2]:get_text(), e.text);

			assert.are.equal("this.test", r.tags[1].attr.by);

		end);

		it("should accept util.error objects with an URI", function ()

			local s = st.message({ to = "touser", from = "fromuser", id = "123", type = "chat" }, "Hello");

			local gone = errors.new({ condition = "gone", extra = { uri = "file:///dev/null" } })

			local gonner = st.error_reply(s, gone);

			assert.are.equal("gone", gonner.tags[1].tags[1].name);

			assert.are.equal("file:///dev/null", gonner.tags[1].tags[1][1]);

		end);

		it("should accept util.error objects with application specific error", function ()

			local s = st.message({ to = "touser", from = "fromuser", id = "123", type = "chat" }, "Hello");

			local e = errors.new({ condition = "internal-server-error", text = "Namespaced thing happened",

				extra = {namespace="xmpp:example.test", condition="this-happened"} })

			local r = st.error_reply(s, e);

			assert.are.equal("xmpp:example.test", r.tags[1].tags[3].attr.xmlns);

			assert.are.equal("this-happened", r.tags[1].tags[3].name);

			local e2 = errors.new({ condition = "internal-server-error", text = "Namespaced thing happened",

				extra = {tag=st.stanza("that-happened", { xmlns = "xmpp:example.test", ["another-attribute"] = "here" })} })

			local r2 = st.error_reply(s, e2);

			assert.are.equal("xmpp:example.test", r2.tags[1].tags[3].attr.xmlns);

			assert.are.equal("that-happened", r2.tags[1].tags[3].name);

			assert.are.equal("here", r2.tags[1].tags[3].attr["another-attribute"]);

		end);

		end);

	end);

	describe("#get_error()", function ()

		describe("basics", function ()

			local s = st.message();

			local e = st.error_reply(s, "cancel", "not-acceptable", "UNACCEPTABLE!!!! ONE MILLION YEARS DUNGEON!")

				:tag("dungeon", { xmlns = "urn:uuid:c9026187-5b05-4e70-b265-c3b6338a7d0f", period="1000000years"});

			local typ, cond, text, extra = e:get_error();

			assert.equal("cancel", typ);

			assert.equal("not-acceptable", cond);

			assert.equal("UNACCEPTABLE!!!! ONE MILLION YEARS DUNGEON!", text);

			assert.not_nil(extra)

		end)

	end)

	describe("#add_error()", function ()

		describe("basics", function ()

			local s = st.stanza("custom", { xmlns = "urn:example:foo" });

			local e = s:add_error("cancel", "not-acceptable", "UNACCEPTABLE!!!! ONE MILLION YEARS DUNGEON!")