fmt

bundle/run/job_test.go

@@ -229,7 +229,7 @@ func TestJobRunnerRestartForContinuousUnpausedJobs(t *testing.T) {
 
 	m := mocks.NewMockWorkspaceClient(t)
 	b.SetWorkpaceClient(m.WorkspaceClient)
-	
+
 	ctx := cmdio.MockDiscard(context.Background())
 	ctx = cmdio.NewContext(ctx, cmdio.NewLogger(flags.ModeAppend))
 

bundle/run/pipeline_test.go

@@ -74,7 +74,7 @@ func TestPipelineRunnerRestart(t *testing.T) {
 		Host: "https://test.com",
 	}
 	b.SetWorkpaceClient(m.WorkspaceClient)
-	
+
 	ctx := cmdio.MockDiscard(context.Background())
 	ctx = cmdio.NewContext(ctx, cmdio.NewLogger(flags.ModeAppend))
 

libs/template/template_test.go

@@ -42,18 +42,13 @@ func TestTemplateTelemetryIsCapturedForAllDefaultTemplates(t *testing.T) {
 	for _, tmpl := range databricksTemplates {
 		w := tmpl.Writer
 
-		if tmpl.name == Custom {
-			// Assert telemetry is not captured for user templates.
-			assert.IsType(t, &defaultWriter{}, w)
-		} else {
-			// Assert telemetry is captured for all other templates, i.e. templates
-			// owned by databricks.
-			assert.IsType(t, &writerWithFullTelemetry{}, w)
-		}
+		// Assert telemetry is captured for all databricks templates, i.e. templates
+		// owned by databricks.
+		assert.IsType(t, &writerWithFullTelemetry{}, w)
 	}
 }
 
-func TestTemplateGet(t *testing.T) {
+func TestTemplateGetDatabricksTemplate(t *testing.T) {
 	names := []TemplateName{
 		DefaultPython,
 		DefaultSql,

libs/template/writer.go

@@ -166,6 +166,6 @@ type writerWithFullTelemetry struct {
 }
 
 func (tmpl *writerWithFullTelemetry) LogTelemetry(ctx context.Context) error {
-	// TODO, log template name, uuid and enum args as well.``
+	// TODO, log template name, uuid and enum args as well.
 	return nil
 }