From 843a32df31bc1a7f1186c41dfbcc9da5b9bfd7e8 Mon Sep 17 00:00:00 2001 From: David Cheung Date: Thu, 20 Aug 2020 15:10:17 -0400 Subject: [PATCH] create: template to have access to files as well --- internal/generate/generate_modules.go | 2 ++ internal/generate/generate_test.go | 4 ++-- tests/test_data/generate/file_to_template.txt | 1 + 3 files changed, 5 insertions(+), 2 deletions(-) diff --git a/internal/generate/generate_modules.go b/internal/generate/generate_modules.go index 29e60a2a4..e522d75cf 100644 --- a/internal/generate/generate_modules.go +++ b/internal/generate/generate_modules.go @@ -49,9 +49,11 @@ func Generate(projectConfig projectconfig.ZeroProjectConfig) error { templateData := struct { Name string Params projectconfig.Parameters + Files projectconfig.Files }{ projectConfig.Name, mod.Parameters, + mod.Files, } txtTypeFiles, binTypeFiles := sortFileType(moduleDir, outputDir, false) diff --git a/internal/generate/generate_test.go b/internal/generate/generate_test.go index a348c8243..8938cca25 100644 --- a/internal/generate/generate_test.go +++ b/internal/generate/generate_test.go @@ -29,10 +29,9 @@ func TestGenerateModules(t *testing.T) { projectConfig := projectconfig.ZeroProjectConfig{ Name: "foo", Modules: projectconfig.Modules{ - "mod1": projectconfig.NewModule(map[string]string{"test": "bar"}, tmpDir, "n/a", baseTestFixturesDir, []string{}), + "mod1": projectconfig.NewModule(map[string]string{"test": "bar"}, tmpDir, "github.com/fake-org/repo-foo", baseTestFixturesDir, []string{}), }, } - generate.Generate(projectConfig) content, err := ioutil.ReadFile(filepath.Join(tmpDir, "file_to_template.txt")) @@ -40,6 +39,7 @@ func TestGenerateModules(t *testing.T) { expectedContent := `Name is foo Params.test is bar +Files.Repository is github.com/fake-org/repo-foo ` assert.Equal(t, string(content), expectedContent) } diff --git a/tests/test_data/generate/file_to_template.txt b/tests/test_data/generate/file_to_template.txt index 7915cf819..e78404811 100644 --- a/tests/test_data/generate/file_to_template.txt +++ b/tests/test_data/generate/file_to_template.txt @@ -1,2 +1,3 @@ Name is {{.Name}} Params.test is {{.Params.test}} +Files.Repository is {{.Files.Repository}}