Expand live stream testing capabilities.

This commit is contained in:
Lance Stout 2011-09-28 17:26:29 -04:00
parent 262da78ca7
commit d10f591bf4

View file

@ -293,7 +293,7 @@ class SleekTest(unittest.TestCase):
def stream_start(self, mode='client', skip=True, header=None, def stream_start(self, mode='client', skip=True, header=None,
socket='mock', jid='tester@localhost', socket='mock', jid='tester@localhost',
password='test', server='localhost', password='test', server='localhost',
port=5222, plugins=None): port=5222, plugins=None, plugin_config={}):
""" """
Initialize an XMPP client or component using a dummy XML stream. Initialize an XMPP client or component using a dummy XML stream.
@ -317,10 +317,11 @@ class SleekTest(unittest.TestCase):
are loaded. are loaded.
""" """
if mode == 'client': if mode == 'client':
self.xmpp = ClientXMPP(jid, password) self.xmpp = ClientXMPP(jid, password, plugin_config=plugin_config)
elif mode == 'component': elif mode == 'component':
self.xmpp = ComponentXMPP(jid, password, self.xmpp = ComponentXMPP(jid, password,
server, port) server, port,
plugin_config=plugin_config)
else: else:
raise ValueError("Unknown XMPP connection mode.") raise ValueError("Unknown XMPP connection mode.")
@ -347,7 +348,10 @@ class SleekTest(unittest.TestCase):
skip_queue.put('started') skip_queue.put('started')
self.xmpp.add_event_handler('session_start', wait_for_session) self.xmpp.add_event_handler('session_start', wait_for_session)
self.xmpp.connect() if server is not None:
self.xmpp.connect((server, port))
else:
self.xmpp.connect()
else: else:
raise ValueError("Unknown socket type.") raise ValueError("Unknown socket type.")