diff --git a/sleekxmpp/test/sleektest.py b/sleekxmpp/test/sleektest.py index 27a7556f..5e61cec2 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')