From 39dfe05a7345b82b095de46ea5937c121ea2ba71 Mon Sep 17 00:00:00 2001
From: Shreyas Goenka <shreyas.goenka@databricks.com>
Date: Mon, 6 Jan 2025 18:03:11 +0530
Subject: [PATCH] add test for the get method

---
 libs/template/template_test.go | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

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)
+	}
+}