1 files changed, 115 insertions, 0 deletions
diff --git a/cmd/template_test.go b/cmd/template_test.go
new file mode 100644
index 0000000..3d9e127
--- /dev/null
+++ b/cmd/template_test.go
@@ -0,0 +1,115 @@
+package main
+
+import (
+	"bytes"
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestNewTemplateCommandWithEmptyArgs(t *testing.T) {
+
+	defaultExpectedOutput := `build: automatic update of example-app
+
+updates image example/example tag '1.0.0' to '1.0.1'
+updates image example/updater tag 'sha256:01d09d19c2139a46aebfb577780d123d7396e97201bc7ead210a2ebff8239dee' to 'sha256:7aa7a5359173d05b63cfd682e3c38487f3cb4f7f1d60659fe59fab1505977d4c'
+
+`
+	cmd := newTemplateCommand()
+	buf := new(bytes.Buffer)
+	args := []string{}
+	cmd.SetOut(buf)
+	cmd.SetArgs(args)
+	err := cmd.Execute()
+	if err != nil {
+		t.Fatalf("could not execute command: %v", err)
+	}
+	assert.NoError(t, err)
+	assert.Equal(t, defaultExpectedOutput, buf.String())
+
+}
+
+func TestNewTemplateCommandWithCustomTemplate(t *testing.T) {
+
+	testTemplate := `Custom Commit Message:
+App: {{.AppName}}
+{{- range .AppChanges }}
+- {{.Image}}: {{.OldTag}} -> {{.NewTag}}
+{{- end }}`
+
+	expectedOutput := `Custom Commit Message:
+App: example-app
+- example/example: 1.0.0 -> 1.0.1
+- example/updater: sha256:01d09d19c2139a46aebfb577780d123d7396e97201bc7ead210a2ebff8239dee -> sha256:7aa7a5359173d05b63cfd682e3c38487f3cb4f7f1d60659fe59fab1505977d4c
+`
+
+	// Create a temporary file to hold the test template
+	tempFile, err := os.CreateTemp("", "test-template.tmpl")
+	if err != nil {
+		t.Fatalf("could not create temp file: %v", err)
+	}
+	defer os.Remove(tempFile.Name())
+	_, err = tempFile.WriteString(testTemplate)
+	if err != nil {
+		t.Fatalf("could not write to temp file: %v", err)
+	}
+	tempFile.Close()
+
+	cmd := newTemplateCommand()
+	buf := new(bytes.Buffer)
+	args := []string{tempFile.Name()}
+	cmd.SetOut(buf)
+	cmd.SetArgs(args)
+	err = cmd.Execute()
+	if err != nil {
+		t.Fatalf("could not execute command: %v", err)
+	}
+	assert.Equal(t, expectedOutput, buf.String())
+}
+
+func TestNewTemplateCommandWithInvalidTemplate(t *testing.T) {
+
+	testTemplate := `Custom Commit Message:
+App: {{.AppName}}
+{{- range .AppChanges }}
+- {{.Image}}: {{.OldTag}} -> {{.NewTag}}
+{{- end`
+
+	// Create a temporary file to hold the test template
+	tempFile, err := os.CreateTemp("", "test-template.tmpl")
+	if err != nil {
+		t.Fatalf("could not create temp file: %v", err)
+	}
+	defer os.Remove(tempFile.Name())
+	_, err = tempFile.WriteString(testTemplate)
+	if err != nil {
+		t.Fatalf("could not write to temp file: %v", err)
+	}
+	tempFile.Close()
+
+	cmd := newTemplateCommand()
+	buf := new(bytes.Buffer)
+	args := []string{tempFile.Name()}
+	cmd.SetErr(buf)
+	cmd.SetArgs(args)
+	err = cmd.Execute()
+	if err != nil {
+		t.Fatalf("could not execute command: %v", err)
+	}
+	assert.Contains(t, buf.String(), "could not parse commit message template")
+}
+
+func TestNewTemplateCommandWithInvalidPath(t *testing.T) {
+
+	cmd := newTemplateCommand()
+	buf := new(bytes.Buffer)
+	args := []string{"test-template.tmpl"}
+	cmd.SetErr(buf)
+	cmd.SetArgs(args)
+	err := cmd.Execute()
+	if err != nil {
+		t.Fatalf("could not execute command: %v", err)
+	}
+	assert.Contains(t, buf.String(), "no such file or directory")
+}