github/SleekXMPP
1
0
Fork 0
mirror of https://github.com/correl/SleekXMPP.git synced 2024-11-27 19:19:54 +00:00
Code Issues 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.socket.recvData(header)
self.xmpp.connect = lambda a=None, b=None, c=None, d=None: True self.xmpp.connect = lambda a=None, b=None, c=None, d=None: True
self.xmpp.start_tls = lambda: True
self.xmpp.process(threaded=True) self.xmpp.process(threaded=True)
if skip: if skip:
# Clear startup stanzas # Clear startup stanzas
@ -420,6 +419,7 @@ class SleekTest(unittest.TestCase):
appended before the stream header. appended before the stream header.
""" """
header = '<stream:stream %s>' header = '<stream:stream %s>'
parts = []
if xml_header: if xml_header:
header = '<?xml version="1.0"?>' + header header = '<?xml version="1.0"?>' + header
if sto: if sto:
@ -467,8 +467,8 @@ class SleekTest(unittest.TestCase):
# Apply closing elements so that we can construct # Apply closing elements so that we can construct
# XML objects for comparison. # XML objects for comparison.
header2 += '</stream:stream>' header2 = header + '</stream:stream>'
sent_header2 += '</stream:stream>' sent_header2 = sent_header + '</stream:stream>'
xml = ET.fromstring(header2) xml = ET.fromstring(header2)
sent_xml = ET.fromstring(sent_header2) sent_xml = ET.fromstring(sent_header2)

5
tests/test_streamtester.py
View file

@ -52,4 +52,9 @@ class TestStreamTester(SleekTest):
</message> </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) suite = unittest.TestLoader().loadTestsFromTestCase(TestStreamTester)