Add missing tests for #197.

wireservice · Aug 29, 2015 · f504d17 · f504d17
1 parent 3eee866
commit f504d17
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/tests/test_analysis.py b/tests/test_analysis.py
@@ -142,3 +142,25 @@ def test_ancestor_changed(self):
 
         self.assertEqual(self.executed_stage1, 1)
         self.assertEqual(self.executed_stage2, 2)
+
+    def test_same_function_twice_parallel(self):
+        analysis = Analysis(self.stage1, cache_dir=TEST_CACHE)
+        noop = analysis.then(self.stage_noop)
+        noop.then(self.stage2)
+        noop.then(self.stage2)
+
+        analysis.run()
+
+        self.assertEqual(self.executed_stage1, 1)
+        self.assertEqual(self.executed_stage2, 2)
+
+    def test_same_function_twice_sequence(self):
+        analysis = Analysis(self.stage1, cache_dir=TEST_CACHE)
+        analysis.then(self.stage2)
+        analysis.then(self.stage_noop)
+        analysis.then(self.stage2)
+
+        analysis.run()
+
+        self.assertEqual(self.executed_stage1, 1)
+        self.assertEqual(self.executed_stage2, 2)