浏览代码

Test events coming out of parallel_execute_stream in error case

Signed-off-by: Aanand Prasad <[email protected]>
Aanand Prasad 9 年之前
父节点
当前提交
a81b9dc6a0
共有 1 个文件被更改,包括 13 次插入0 次删除
  1. 13 0
      tests/unit/parallel_test.py

+ 13 - 0
tests/unit/parallel_test.py

@@ -5,6 +5,8 @@ import six
 from docker.errors import APIError
 
 from compose.parallel import parallel_execute
+from compose.parallel import parallel_execute_stream
+from compose.parallel import UpstreamError
 
 
 web = 'web'
@@ -75,3 +77,14 @@ def test_parallel_execute_with_upstream_errors():
     )
 
     assert log == [cache]
+
+    events = [
+        (obj, result, type(exception))
+        for obj, result, exception
+        in parallel_execute_stream(objects, process, get_deps)
+    ]
+
+    assert (cache, None, type(None)) in events
+    assert (data_volume, None, APIError) in events
+    assert (db, None, UpstreamError) in events
+    assert (web, None, UpstreamError) in events