Added test on take crash

igorcoding · Sep 13, 2017 · 7dfe7d4 · 7dfe7d4
1 parent 7ad9ade
commit 7dfe7d4
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/test_tube.py b/tests/test_tube.py
@@ -30,6 +30,11 @@ async def test__tube_take(self):
         self.assertEqual(taken_t.status, Status.TAKEN)
         self.assertDictEqual(taken_t.data, t.data)
 
+    async def test__tube_take_no_tasks(self):
+        tube = self.create_tube()
+        taken_t = await tube.take(0.5)
+        self.assertIsNone(taken_t)
+
     async def test__tube_ack(self):
         tube = self.create_tube()
         t = await tube.put(self._data_obj())