Fixed tests in async environments

syrusakbary · Apr 12, 2017 · 7b7f68f · 7b7f68f
1 parent 2b4773b
commit 7b7f68f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/tests/test_extra.py b/tests/test_extra.py
@@ -637,7 +637,8 @@ def test_promises_with_only_then():
     promise3 = promise1.then(lambda x: None)
     context["promise1_reject"](error)
 
-    promise1._wait()
+    promise2._wait()
+    promise3._wait()
     assert promise2.reason == error
     assert promise3.reason == error
 

diff --git a/tests/test_issues.py b/tests/test_issues.py
@@ -61,9 +61,11 @@ def test_issue_26():
     context["promise1_reject"](RuntimeError("Ooops!"))
 
     promise2 = Promise(lambda resolve, reject: context.update({"promise2_resolve": resolve}))
-    promise2.then(lambda x: context.update({"success": True}))
+    promise3 = promise2.then(lambda x: context.update({"success": True}))
     context["promise2_resolve"](None)
 
+    # We wait so it works in asynchronous envs
+    promise3._wait(timeout=.1)
     assert context["success"]