Merge d2c4815 into b055d1e

syrusakbary · Dec 19, 2017 · d29aaf4 · d29aaf4
2 parents b055d1e + d2c4815
commit d29aaf4
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/promise/async_.py b/promise/async_.py
@@ -108,11 +108,11 @@ def wait(self, promise, timeout=None):
         self.schedule.wait(target, timeout)
 
     def drain_queues(self):
-        assert self.is_tick_used
-        self.drain_queue(self.normal_queue)
-        self.reset()
-        self.have_drained_queues = True
-        self.drain_queue(self.late_queue)
+        if not self.is_tick_used:
+            self.drain_queue(self.normal_queue)
+            self.reset()
+            self.have_drained_queues = True
+            self.drain_queue(self.late_queue)
 
     def queue_tick(self):
         if not self.is_tick_used: