Fix test.

coleifer · Jan 3, 2016 · b7555db · b7555db
1 parent e526b8f
commit b7555db
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/huey/tests/test_consumer.py b/huey/tests/test_consumer.py
@@ -68,7 +68,9 @@ def test_threaded_execution(self):
                 time.sleep(.05)
 
             consumer.stop()
-            consumer.wait_finished()
+            for worker in consumer.worker_threads:
+                worker.join()
+            #consumer.wait_finished()
 
         self.assertEqual(state, {'k1': 'v1', 'k2': 'v2', 'k3': 'v3'})