From: Zack Cerza <zack@cerza.org>
Date: Wed, 2 Jul 2014 16:08:11 +0000 (-0600)
Subject: Add a couple unit tests for teuthology.parallel
X-Git-Tag: 1.1.0~1357
X-Git-Url: http://git-server-git.apps.pok.os.sepia.ceph.com/?a=commitdiff_plain;h=43c4595ae26a7663538e712af37553c7a64fade7;p=teuthology.git

Add a couple unit tests for teuthology.parallel

Signed-off-by: Zack Cerza <zack.cerza@inktank.com>
---

diff --git a/teuthology/test/test_parallel.py b/teuthology/test/test_parallel.py
new file mode 100644
index 000000000..526b0a727
--- /dev/null
+++ b/teuthology/test/test_parallel.py
@@ -0,0 +1,28 @@
+from ..parallel import parallel
+
+
+def identity(item, input_set=None, remove=False):
+    if input_set is not None:
+        assert item in input_set
+        if remove:
+            input_set.remove(item)
+    return item
+
+
+class TestParallel(object):
+    def test_basic(self):
+        in_set = set(range(10))
+        with parallel() as para:
+            for i in in_set:
+                para.spawn(identity, i, in_set, remove=True)
+                assert para.any_spawned is True
+            assert para.count == len(in_set)
+
+    def test_result(self):
+        in_set = set(range(10))
+        with parallel() as para:
+            for i in in_set:
+                para.spawn(identity, i, in_set)
+            for result in para:
+                in_set.remove(result)
+