improve coverage

census-instrumentation · May 15, 2019 · 369c072 · 369c072
1 parent cfea29b
commit 369c072
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/unit/common/test_schedule.py b/tests/unit/common/test_schedule.py
@@ -63,10 +63,14 @@ def test_gets_event(self):
     def test_flush_timeout(self):
         queue = Queue(capacity=10)
         self.assertIsNone(queue.flush(timeout=TIMEOUT))
+        queue.puts(range(100), timeout=TIMEOUT)
+        self.assertIsNone(queue.flush(timeout=TIMEOUT))
+
         def proc():
             for item in queue.gets(count=1, timeout=TIMEOUT):
                 if isinstance(item, QueueEvent):
                     item.set()
+
         task = PeriodicTask(TIMEOUT / 10, proc)
         task.start()
         try: