From 3722a56ab73a8c3c7b1f21b4be2015f3cad1b023 Mon Sep 17 00:00:00 2001
From: ecrupper <easton.crupper12@gmail.com>
Date: Thu, 9 May 2024 15:23:51 -0500
Subject: [PATCH] fix(compiler): aggregate templates for nested pipelines

---
 compiler/native/expand.go      |  4 ++++
 compiler/native/expand_test.go | 19 ++++++++++++++++++-
 2 files changed, 22 insertions(+), 1 deletion(-)

diff --git a/compiler/native/expand.go b/compiler/native/expand.go
index e69066ef7..335ca5007 100644
--- a/compiler/native/expand.go
+++ b/compiler/native/expand.go
@@ -60,6 +60,7 @@ func (c *client) ExpandSteps(s *yaml.Build, tmpls map[string]*yaml.Template, r *
 	secrets := s.Secrets
 	services := s.Services
 	environment := s.Environment
+	templates := s.Templates
 
 	if len(environment) == 0 {
 		environment = make(raw.StringSliceMap)
@@ -139,6 +140,8 @@ func (c *client) ExpandSteps(s *yaml.Build, tmpls map[string]*yaml.Template, r *
 				return s, fmt.Errorf("cannot use render_inline inside a called template (%s)", step.Template.Name)
 			}
 
+			templates = append(templates, tmplBuild.Templates...)
+
 			tmplBuild, err = c.ExpandSteps(tmplBuild, mapFromTemplates(tmplBuild.Templates), r, depth-1)
 			if err != nil {
 				return s, err
@@ -202,6 +205,7 @@ func (c *client) ExpandSteps(s *yaml.Build, tmpls map[string]*yaml.Template, r *
 	s.Secrets = secrets
 	s.Services = services
 	s.Environment = environment
+	s.Templates = templates
 
 	return s, nil
 }
diff --git a/compiler/native/expand_test.go b/compiler/native/expand_test.go
index cf6f3927f..0e69eed65 100644
--- a/compiler/native/expand_test.go
+++ b/compiler/native/expand_test.go
@@ -869,6 +869,19 @@ func TestNative_ExpandSteps_TemplateCallTemplate(t *testing.T) {
 		"star": "test3",
 	}
 
+	wantTemplates := yaml.TemplateSlice{
+		{
+			Name:   "chain",
+			Source: "github.example.com/faz/baz/template_calls_template.yml",
+			Type:   "github",
+		},
+		{
+			Name:   "test",
+			Source: "github.example.com/foo/bar/long_template.yml",
+			Type:   "github",
+		},
+	}
+
 	// run test
 	compiler, err := New(c)
 	if err != nil {
@@ -879,7 +892,7 @@ func TestNative_ExpandSteps_TemplateCallTemplate(t *testing.T) {
 
 	for _, test := range tests {
 		t.Run(test.name, func(t *testing.T) {
-			build, err := compiler.ExpandSteps(&yaml.Build{Steps: steps, Services: yaml.ServiceSlice{}, Environment: globalEnvironment}, test.tmpls, new(pipeline.RuleData), compiler.TemplateDepth)
+			build, err := compiler.ExpandSteps(&yaml.Build{Steps: steps, Services: yaml.ServiceSlice{}, Environment: globalEnvironment, Templates: yaml.TemplateSlice{test.tmpls["chain"]}}, test.tmpls, new(pipeline.RuleData), compiler.TemplateDepth)
 			if err != nil {
 				t.Errorf("ExpandSteps_Type%s returned err: %v", test.name, err)
 			}
@@ -899,6 +912,10 @@ func TestNative_ExpandSteps_TemplateCallTemplate(t *testing.T) {
 			if diff := cmp.Diff(build.Environment, wantEnvironment); diff != "" {
 				t.Errorf("ExpandSteps()_Type%s mismatch (-want +got):\n%s", test.name, diff)
 			}
+
+			if diff := cmp.Diff(build.Templates, wantTemplates); diff != "" {
+				t.Errorf("ExpandSteps()_Type%s mismatch (-want +got):\n%s", test.name, diff)
+			}
 		})
 	}
 }