diff options
Diffstat (limited to 'src/engine/SCons/TaskmasterTests.py')
| -rw-r--r-- | src/engine/SCons/TaskmasterTests.py | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/src/engine/SCons/TaskmasterTests.py b/src/engine/SCons/TaskmasterTests.py index 63190a5..d20b805 100644 --- a/src/engine/SCons/TaskmasterTests.py +++ b/src/engine/SCons/TaskmasterTests.py @@ -53,6 +53,7 @@ class Node: self.side_effect = 0 self.side_effects = [] self.alttargets = [] + self.postprocessed = None for kid in kids: kid.parents.append(self) @@ -137,6 +138,9 @@ class Node: def __str__(self): return self.name + def postprocess(self): + self.postprocessed = 1 + class OtherError(Exception): pass @@ -791,6 +795,33 @@ class TaskmasterTestCase(unittest.TestCase): else: assert 0, "did not catch expected exception" + def test_postprocess(self): + """Test postprocessing targets to give them a chance to clean up + + """ + n1 = Node("n1") + tm = SCons.Taskmaster.Taskmaster([n1]) + + t = tm.next_task() + assert not n1.postprocessed + t.postprocess() + assert n1.postprocessed + + n2 = Node("n2") + n3 = Node("n3") + tm = SCons.Taskmaster.Taskmaster([n2, n3]) + + assert not n2.postprocessed + assert not n3.postprocessed + t = tm.next_task() + t.postprocess() + assert n2.postprocessed + assert not n3.postprocessed + t = tm.next_task() + t.postprocess() + assert n2.postprocessed + assert n3.postprocessed + if __name__ == "__main__": |