Add derived_from to scrooge_gen scrooge generated targets

Testing Done: https://travis-ci.org/pantsbuild/pants/builds/33099414 Reviewed at https://rbcommons.com/s/twitter/r/926/
benjyw · Aug 20, 2014 · e5162aa · e5162aa
1 parent 541f13c
commit e5162aa
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/python/pants/backend/codegen/tasks/scrooge_gen.py b/src/python/pants/backend/codegen/tasks/scrooge_gen.py
@@ -243,7 +243,8 @@ def create_target(files, deps, target_type):
                                          sources=files,
                                          provides=gentarget.provides,
                                          dependencies=deps,
-                                         excludes=gentarget.excludes)
+                                         excludes=gentarget.excludes,
+                                         derived_from=gentarget)
 
     def create_geninfo(key):
       compiler = self.compiler_for_name[self.defaults.get_compiler(gentarget)]

diff --git a/tests/python/pants_test/tasks/test_scrooge_gen.py b/tests/python/pants_test/tasks/test_scrooge_gen.py
@@ -91,9 +91,10 @@ def test_smoke(self):
       )
     '''))
 
+    target = self.target('test_smoke:a')
     task = prepare_task(ScroogeGen,
                         build_graph=self.build_graph,
-                        targets=[self.target('test_smoke:a')],
+                        targets=[target],
                         build_file_parser=self.build_file_parser)
 
     with patch('pants.backend.codegen.tasks.scrooge_gen.calculate_services'):
@@ -116,6 +117,7 @@ def test_smoke(self):
                                                        sources=sources,
                                                        excludes=OrderedSet(),
                                                        dependencies=OrderedSet(),
-                                                       provides=None)
+                                                       provides=None,
+                                                       derived_from=target)
       finally:
         Context.add_new_target = saved_add_new_target