summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/sleektest.py6
-rw-r--r--tests/test_streamtester.py11
2 files changed, 11 insertions, 6 deletions
diff --git a/tests/sleektest.py b/tests/sleektest.py
index e7951892..e08a8deb 100644
--- a/tests/sleektest.py
+++ b/tests/sleektest.py
@@ -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)
diff --git a/tests/test_streamtester.py b/tests/test_streamtester.py
index 8512f401..8a4f82dd 100644
--- a/tests/test_streamtester.py
+++ b/tests/test_streamtester.py
@@ -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)