pep/slixmpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Corrected stream header tester.

Browse source 
Added test for testing stream headers.
This commit is contained in:
Lance Stout 2010-10-06 18:46:23 -04:00
parent 799645f13f
commit 88d21d210c
2 changed files with 11 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/sleektest.py
View file

@ -378,7 +378,6 @@ class SleekTest(unittest.TestCase):
self.xmpp.socket.recvData(header)
self.xmpp.connect = lambda a=None, b=None, c=None, d=None: True
self.xmpp.start_tls = lambda: True
self.xmpp.process(threaded=True)
if skip:
# Clear startup stanzas
@ -420,6 +419,7 @@ class SleekTest(unittest.TestCase):
appended before the stream header.
"""
header = '<stream:stream %s>'
parts = []
if xml_header:
header = '<?xml version="1.0"?>' + header
if sto:
@ -467,8 +467,8 @@ class SleekTest(unittest.TestCase):
# Apply closing elements so that we can construct
# XML objects for comparison.
header2 += '</stream:stream>'
sent_header2 += '</stream:stream>'
header2 = header + '</stream:stream>'
sent_header2 = sent_header + '</stream:stream>'
xml = ET.fromstring(header2)
sent_xml = ET.fromstring(sent_header2)

11
tests/test_streamtester.py
View file

@ -16,15 +16,15 @@ class TestStreamTester(SleekTest):
def echo(msg):
msg.reply('Thanks for sending: %(body)s' % msg).send()
self.xmpp.add_event_handler('message', echo)
self.streamRecv("""
<message to="tester@localhost" from="user@localhost">
<body>Hi!</body>
</message>
""")
self.streamSendMessage("""
<message to="user@localhost">
<body>Thanks for sending: Hi!</body>
@ -52,4 +52,9 @@ class TestStreamTester(SleekTest):
</message>
""")
def testSendStreamHeader(self):
"""Test that we can check a sent stream header."""
self.streamStart(mode='client', skip=False)
self.streamSendHeader(sto='localhost')
suite = unittest.TestLoader().loadTestsFromTestCase(TestStreamTester)