1 files changed, 61 insertions, 0 deletions
diff --git a/client/go/script-utils/logfmt/showflags_test.go b/client/go/script-utils/logfmt/showflags_test.go
new file mode 100644
index 00000000000..d1b66118afd
--- /dev/null
+++ b/client/go/script-utils/logfmt/showflags_test.go
@@ -0,0 +1,61 @@
+// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+// vespa logfmt command
+// Author: arnej
+
+package logfmt
+
+import (
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"strings"
+	"testing"
+)
+
+func TestShowFlags(t *testing.T) {
+	none := " -time -fmttime -msecs -usecs -host -level -pid -service -component -message"
+	var flag flagValueForShow
+	assert.Equal(t, none, flag.String(), "unset flag displays as '%s', expected '%s'", flag.String(), none)
+	assert.Equal(t, "show flags", flag.Type())
+	check := func(expected string, texts ...string) {
+		var target flagValueForShow
+		// target.levels = defaultLevelFlags()
+		target.shown = defaultShowFlags()
+		for _, text := range texts {
+			err := target.Set(text)
+			require.Nil(t, err, "unexpected error with show flags Set('%s'): %v", text, err)
+		}
+		assert.Equal(t, expected, target.String())
+	}
+	check(" +time +fmttime +msecs -usecs -host +level -pid +service +component +message")
+	check(" -time -fmttime -msecs -usecs -host -level -pid -service -component -message", "-all")
+	check(" +time +fmttime +msecs +usecs +host +level +pid +service +component +message", "all")
+	check(" +time +fmttime +msecs +usecs +host +level +pid +service +component +message", "+all")
+	check(" -time -fmttime -msecs -usecs -host -level +pid -service -component -message", "pid")
+	check(" +time +fmttime +msecs +usecs -host +level +pid +service +component +message", "all-host")
+	check(" +time +fmttime +msecs +usecs -host +level +pid +service +component +message", "all", "-host")
+	check(" +time +fmttime -msecs -usecs -host +level +pid +service +component +message", "+pid", "-msecs")
+	check(" +time -fmttime +msecs -usecs +host +level -pid -service +component +message", "+host,-fmttime,-service,pid")
+	check(" +time -fmttime +msecs -usecs +host +level +pid -service +component +message", "+host,-fmttime,-service,+pid")
+	check(" +time -fmttime +msecs -usecs +host +level +pid -service +component +message", "+host,-fmttime", "-service,+pid")
+	check(" -time -fmttime -msecs -usecs -host -level +pid -service -component -message", "+host", "-fmttime", "-service", "pid")
+	check = func(expectErr string, texts ...string) {
+		var target flagValueForShow
+		target.shown = defaultShowFlags()
+		for _, text := range texts {
+			err := target.Set(text)
+			if err != nil {
+				require.Equal(t, expectErr, err.Error())
+				return
+			}
+		}
+		t.Logf("Did not get expected error [%s] from %s", expectErr, strings.Join(texts, " "))
+		t.Fail()
+	}
+	check("not a valid show flag: 'foo'", "foo")
+	check("not a valid show flag: 'foo'", "level,foo,message")
+	check("not a valid show flag: 'foo'", "-level,-foo,-message")
+	check("not a valid show flag: 'foo'", "+level,+foo,+message")
+	check("not a valid show flag: 'foo'", "level", "foo", "message")
+	check("not a valid show flag: 'foo'", "-level", "-foo", "-message")
+	check("not a valid show flag: 'foo'", "+level", "+foo", "+message")
+}