diff --git a/libs/template/template_test.go b/libs/template/template_test.go index cfcdd6251..a51da4973 100644 --- a/libs/template/template_test.go +++ b/libs/template/template_test.go @@ -60,3 +60,30 @@ func TestTemplateTelemetryIsCapturedForAllDefaultTemplates(t *testing.T) { } } } + +func TestTemplateGet(t *testing.T) { + names := []TemplateName{ + DefaultPython, + DefaultSql, + DbtSql, + MlopsStacks, + DefaultPydabs, + Custom, + } + + for _, name := range names { + tmpl := Get(name) + assert.Equal(t, tmpl.name, name) + } + + notExist := []string{ + "/some/path", + "doesnotexist", + "https://www.someurl.com", + } + + for _, name := range notExist { + tmpl := Get(TemplateName(name)) + assert.Nil(t, tmpl) + } +}