# tests for xmpppy's dispatcher_nb.py import unittest import lib lib.setup_env() from mock import Mock from common.xmpp import dispatcher_nb from common.xmpp import auth class TestDispatcherNB(unittest.TestCase): def test_unbound_namespace_prefix(self): '''tests our handling of a message with an unbound namespace prefix''' d = dispatcher_nb.Dispatcher() conn = Mock() owner = Mock() owner._caller = Mock() owner.defaultNamespace = auth.NS_CLIENT owner.debug_flags = [] owner.Connection = conn owner._component = False d._owner = owner d.plugin(owner) msgs = [] def _got_message(conn, msg): msgs.append(msg) d.RegisterHandler('message', _got_message) d.StreamInit() d.ProcessNonBlocking("") # should be able to parse a normal message d.ProcessNonBlocking('hello') self.assertEqual(1, len(msgs)) d.ProcessNonBlocking('') # we should have been disconnected after that message self.assertEqual(1, len(conn.mockGetNamedCalls('pollend'))) # we should not be able to keep parsing d.ProcessNonBlocking('still here?') self.assertEqual(1, len(msgs)) if __name__ == '__main__': unittest.main() # vim: se ts=3: