From 88d21d210c15d76000b6a3c69e47a55b1582d054 Mon Sep 17 00:00:00 2001
From: Lance Stout <lancestout@gmail.com>
Date: Wed, 6 Oct 2010 18:46:23 -0400
Subject: Corrected stream header tester.

Added test for testing stream headers.
---
 tests/sleektest.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'tests/sleektest.py')

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)
-- 
cgit v1.2.3