tests: Add test for iq error replies

[prosody.git] / tests / test_core_stanza_router.lua

diff --git a/tests/test_core_stanza_router.lua b/tests/test_core_stanza_router.lua
index 1193a7dbf7c87dde38ecb5569f0cae289b719409..5d00ce2afbdf1dc9fffa1002671bf535bd27e0e8 100644 (file)
--- a/tests/test_core_stanza_router.lua
+++ b/tests/test_core_stanza_router.lua
@@ -1,6 +1,6 @@
 -- Prosody IM v0.3
--- Copyright (C) 2008 Matthew Wild
--- Copyright (C) 2008 Waqas Hussain
+-- Copyright (C) 2008-2009 Matthew Wild
+-- Copyright (C) 2008-2009 Waqas Hussain
 -- 
 -- This project is MIT/X11 licensed. Please see the
 -- COPYING file in the source package for more information.
@@ -10,8 +10,8 @@
 
 function core_process_stanza(core_process_stanza)
        local s2sout_session = { to_host = "remotehost", from_host = "localhost", type = "s2sout" }
-       local s2sin_session = { from_host = "remotehost", to_host = "localhost", type = "s2sin" }
-       local local_host_session = { host = "localhost", type = "local" }
+       local s2sin_session = { from_host = "remotehost", to_host = "localhost", type = "s2sin", hosts = { ["remotehost"] = { authed = true } } }
+       local local_host_session = { host = "localhost", type = "local", s2sout = { ["remotehost"] = s2sout_session } }
        local local_user_session = { username = "user", host = "localhost", resource = "resource", full_jid = "user@localhost/resource", type = "c2s" }
        local hosts = {
                        ["localhost"] = local_host_session;
@@ -155,6 +155,28 @@ function core_process_stanza(core_process_stanza)
                assert_equal(target_routed, true, "stanza was not routed successfully");
        end
 
+       local function test_iq_to_local_bare()
+               local env = testlib_new_env();
+               local msg = stanza.stanza("iq", { to = "user@localhost", from = "user@localhost", type = "get" }):tag("ping", { xmlns = "urn:xmpp:ping:0" });
+               
+               local target_handled;
+               
+               function env.core_handle_stanza(p_origin, p_stanza)
+                       assert_equal(p_origin, local_user_session, "origin of handled stanza is not correct");
+                       assert_equal(p_stanza, msg, "handled stanza is not correct one: "..p_stanza:pretty_print());
+                       target_handled = true;          
+               end
+
+               function env.core_route_stanza(p_origin, p_stanza)
+                       
+               end
+
+               env.hosts = hosts;
+               setfenv(core_process_stanza, env);
+               assert_equal(core_process_stanza(local_user_session, msg), nil, "core_process_stanza returned incorrect value");
+               assert_equal(target_handled, true, "stanza was not handled successfully");
+       end
+
        runtest(test_message_full_jid, "Messages with full JID destinations get routed");
        runtest(test_message_bare_jid, "Messages with bare JID destinations get routed");
        runtest(test_message_no_to, "Messages with no destination are handled by the server");
@@ -162,6 +184,43 @@ function core_process_stanza(core_process_stanza)
        runtest(test_message_to_remote_server, "Messages to a remote server's JID are routed");
 
        runtest(test_iq_to_remote_server, "iq to a remote server's JID are routed");
+       runtest(test_iq_to_local_bare, "iq from a local user to a local user's bare JID are handled");
        runtest(test_iq_error_to_local_user, "iq type=error to a local user's JID are routed");
+end
+
+function core_route_stanza(core_route_stanza)
+       local s2sout_session = { to_host = "remotehost", from_host = "localhost", type = "s2sout" }
+       local s2sin_session = { from_host = "remotehost", to_host = "localhost", type = "s2sin", hosts = { ["remotehost"] = { authed = true } } }
+       local local_host_session = { host = "localhost", type = "local", s2sout = { ["remotehost"] = s2sout_session }, sessions = {} }
+       local local_user_session = { username = "user", host = "localhost", resource = "resource", full_jid = "user@localhost/resource", type = "c2s" }
+       local hosts = {
+                       ["localhost"] = local_host_session;
+                       }
+
+       local function test_iq_result_to_offline_user()
+               local env = testlib_new_env();
+               local msg = stanza.stanza("iq", { to = "user@localhost/foo", from = "user@localhost", type = "result" }):tag("ping", { xmlns = "urn:xmpp:ping:0" });
+               local msg2 = stanza.stanza("iq", { to = "user@localhost/foo", from = "user@localhost", type = "error" }):tag("ping", { xmlns = "urn:xmpp:ping:0" });
+               --package.loaded["core.usermanager"] = { user_exists = function (user, host) print("RAR!") return true or user == "user" and host == "localhost" and true; end };
+               local target_handled, target_replied;
+               
+               function env.core_handle_stanza(p_origin, p_stanza)
+                       target_handled = true;
+               end
+               
+               function local_user_session.send(data)
+                       --print("Replying with: ", tostring(data));
+                       --print(debug.traceback())
+                       target_replied = true;
+               end
+
+               env.hosts = hosts;
+               setfenv(core_route_stanza, env);
+               assert_equal(core_route_stanza(local_user_session, msg), nil, "core_route_stanza returned incorrect value");
+               assert_equal(target_handled, nil, "stanza was handled and not dropped");
+               assert_equal(target_replied, nil, "stanza was replied to and not dropped");
+               package.loaded["core.usermanager"] = nil;
+       end
 
+       runtest(test_iq_result_to_offline_user, "iq type=result|error to an offline user are not replied to");
 end