Expand live stream testing capabilities.
This commit is contained in:
parent
262da78ca7
commit
d10f591bf4
1 changed files with 8 additions and 4 deletions
|
@ -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.")
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue