Tweak teardown procedure

aio-libs · Feb 21, 2016 · 009db41 · 009db41
1 parent 8a6b069
commit 009db41
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/test_redis_storage.py b/tests/test_redis_storage.py
@@ -20,8 +20,11 @@ def setUp(self):
         self.srv = None
 
     def tearDown(self):
-        self.loop.run_until_complete(self.handler.finish_connections())
-        self.srv.close()
+        if self.handler is not None:
+            self.loop.run_until_complete(self.handler.finish_connections())
+        if self.srv is not None:
+            self.srv.close()
+            self.loop.run_until_complete(self.srv.wait_closed())
         self.loop.stop()
         self.loop.run_forever()
         self.loop.close()