1 files changed, 30 insertions, 0 deletions
diff --git a/client/go/script-utils/logfmt/formatflags_test.go b/client/go/script-utils/logfmt/formatflags_test.go
new file mode 100644
index 00000000000..53c47d24208
--- /dev/null
+++ b/client/go/script-utils/logfmt/formatflags_test.go
@@ -0,0 +1,30 @@
+package logfmt
+
+import (
+	"fmt"
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestOutputFormat(t *testing.T) {
+	type args struct {
+		val string
+	}
+	tests := []struct {
+		expected OutputFormat
+		arg      string
+		wantErr  assert.ErrorAssertionFunc
+	}{
+		{FormatVespa, "vespa", assert.NoError},
+		{FormatRaw, "raw", assert.NoError},
+		{FormatJSON, "json", assert.NoError},
+		{-1, "foo", assert.Error},
+	}
+	for _, tt := range tests {
+		t.Run(tt.arg, func(t *testing.T) {
+			var v OutputFormat = -1
+			tt.wantErr(t, v.Set(tt.arg), fmt.Sprintf("Set(%v)", tt.arg))
+			assert.Equal(t, v, tt.expected)
+		})
+	}
+}