Save memory by allowing Nodes to clean up their Executor's build environments after they've been built.

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__":