From e648f08badce77bce3842108340a4c257b619bc1 Mon Sep 17 00:00:00 2001
From: Lance Stout <lancestout@gmail.com>
Date: Wed, 17 Nov 2010 16:07:27 -0500
Subject: [PATCH] Fix stream test errors.

---
 sleekxmpp/test/sleektest.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/sleekxmpp/test/sleektest.py b/sleekxmpp/test/sleektest.py
index 27a7556..5e61cec 100644
--- a/sleekxmpp/test/sleektest.py
+++ b/sleekxmpp/test/sleektest.py
@@ -197,7 +197,10 @@ class SleekTest(unittest.TestCase):
                     "Stanza:\n%s" % str(stanza))
         else:
             stanza_class = stanza.__class__
-            xml = self.parse_xml(criteria)
+            if isinstance(criteria, str):
+                xml = self.parse_xml(criteria)
+            else:
+                xml = criteria.xml
 
             # Ensure that top level namespaces are used, even if they
             # were not provided.
@@ -590,7 +593,6 @@ class SleekTest(unittest.TestCase):
         sent = self.xmpp.socket.next_sent(timeout)
         if sent is None:
             return False
-        print sent
         xml = self.parse_xml(sent)
         self.fix_namespaces(xml, 'jabber:client')
         sent = self.xmpp._build_stanza(xml, 'jabber:client')