integration/bundle/init_test.go

@@ -246,7 +246,7 @@ func TestBundleInitTelemetryForCustomTemplates(t *testing.T) {
 	require.Equal(t, 1, len(logs))
 	event := logs[0].Entry.DatabricksCliLog.BundleInitEvent
 	assert.Equal(t, event.TemplateName, "custom")
-	assert.Empty(t, event.TemplateEnumArgs)
+	assert.Nil(t, event.TemplateEnumArgs)
 
 	// Ensure that the UUID returned by the `bundle_uuid` helper is the same UUID
 	// that's logged in the telemetry event.

libs/template/builtin_test.go

@@ -11,24 +11,18 @@ import (
 func TestBuiltin(t *testing.T) {
 	out, err := Builtin()
 	require.NoError(t, err)
-	assert.GreaterOrEqual(t, len(out), 3)
+	assert.Len(t, out, 3)
 
-	// Create a map of templates by name for easier lookup
-	templates := make(map[string]*BuiltinTemplate)
-	for _, tmpl := range out {
-		templates[tmpl.Name] = &tmpl
-	}
+	// Confirm names.
+	assert.Equal(t, "dbt-sql", out[0].Name)
+	assert.Equal(t, "default-python", out[1].Name)
+	assert.Equal(t, "default-sql", out[2].Name)
 
-	// Verify all expected templates exist
-	assert.Contains(t, templates, "dbt-sql")
-	assert.Contains(t, templates, "default-python")
-	assert.Contains(t, templates, "default-sql")
-
-	// Verify the filesystems work for each template
-	_, err = fs.Stat(templates["dbt-sql"].FS, `template/{{.project_name}}/dbt_project.yml.tmpl`)
+	// Confirm that the filesystems work.
+	_, err = fs.Stat(out[0].FS, `template/{{.project_name}}/dbt_project.yml.tmpl`)
 	assert.NoError(t, err)
-	_, err = fs.Stat(templates["default-python"].FS, `template/{{.project_name}}/tests/main_test.py.tmpl`)
+	_, err = fs.Stat(out[1].FS, `template/{{.project_name}}/tests/main_test.py.tmpl`)
 	assert.NoError(t, err)
-	_, err = fs.Stat(templates["default-sql"].FS, `template/{{.project_name}}/src/orders_daily.sql.tmpl`)
+	_, err = fs.Stat(out[2].FS, `template/{{.project_name}}/src/orders_daily.sql.tmpl`)
 	assert.NoError(t, err)
 }