From 7a5ef2849218e122b04e244aeedd67844a0690b2 Mon Sep 17 00:00:00 2001
From: Lance Stout <lancestout@gmail.com>
Date: Wed, 11 Aug 2010 18:41:57 -0400
Subject: [PATCH] Updated SleekTest.streamClose to check that the stream was
 actually started before closing it.

Updated tests for Iq stanzas to not start a stream for every test; tests now run a lot faster.
The call to streamClose must still be in the tearDown method to ensure it is called in the
case of an error.
---
 tests/sleektest.py      | 2 +-
 tests/test_iqstanzas.py | 5 +----
 2 files changed, 2 insertions(+), 5 deletions(-)

diff --git a/tests/sleektest.py b/tests/sleektest.py
index a711654..0adab2b 100644
--- a/tests/sleektest.py
+++ b/tests/sleektest.py
@@ -258,7 +258,7 @@ class SleekTest(unittest.TestCase):
         self.checkPresence(data, sent, use_values)
 
     def streamClose(self):
-        if self.xmpp is not None:
+        if hasattr(self, 'xmpp') and self.xmpp is not None:
             self.xmpp.disconnect()
             self.xmpp.socket.recvData(self.xmpp.stream_footer)
 
diff --git a/tests/test_iqstanzas.py b/tests/test_iqstanzas.py
index 98a01a2..2dabc5e 100644
--- a/tests/test_iqstanzas.py
+++ b/tests/test_iqstanzas.py
@@ -4,10 +4,6 @@ from sleekxmpp.xmlstream.stanzabase import ET
 
 class TestIqStanzas(SleekTest):
 
-    def setUp(self):
-        """Start XML stream for testing."""
-        self.streamStart()
-
     def tearDown(self):
         """Shutdown the XML stream after testing."""
         self.streamClose()
@@ -32,6 +28,7 @@ class TestIqStanzas(SleekTest):
 
     def testUnhandled(self):
         """Test behavior for Iq.unhandled."""
+        self.streamStart()
         self.streamRecv("""
           <iq id="test" type="get">
             <query xmlns="test" />