Fix tests, close test loop if needed

aio-libs · Jun 22, 2015 · 5e97e11 · 5e97e11
1 parent b20f166
commit 5e97e11
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/tests/selectors_test.py b/tests/selectors_test.py
@@ -444,7 +444,7 @@ def test_select_interrupt(self):
         s.register(rd, EVENT_READ)
         t = time()
         self.assertFalse(s.select(2))
-        self.assertLess(time() - t, 2.5)
+        self.assertLess(time() - t, 3.5)
 
     # see issue #18963 for why it's skipped on older OS X versions
     @requires_mac_ver(10, 5)

diff --git a/tests/transport_test.py b/tests/transport_test.py
@@ -25,6 +25,9 @@ def setUp(self):
         self.exc_handler = mock.Mock()
         self.loop.set_exception_handler(self.exc_handler)
 
+    def tearDown(self):
+        self.loop.close()
+
     def test_empty_write(self):
         self.tr.write([b''])
         self.assertTrue(self.sock.send_multipart.called)
@@ -571,6 +574,9 @@ def setUp(self):
         self.exc_handler = mock.Mock()
         self.loop.set_exception_handler(self.exc_handler)
 
+    def tearDown(self):
+        self.loop.close()
+
     def test_incomplete_read(self):
         self.sock.recv_multipart.side_effect = zmq.Again(errno.EAGAIN)
         self.tr._do_read()