1 files changed, 41 insertions, 0 deletions
diff --git a/client/go/script-utils/logfmt/regexflag_test.go b/client/go/script-utils/logfmt/regexflag_test.go
new file mode 100644
index 00000000000..489439863a2
--- /dev/null
+++ b/client/go/script-utils/logfmt/regexflag_test.go
@@ -0,0 +1,41 @@
+// 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"
+	"testing"
+)
+
+func assertMatch(t *testing.T, re string, flag *regexFlag, texts ...string) {
+	t.Helper()
+	err := flag.Set(re)
+	require.Nil(t, err, "unexpected error with flag.Set('%s'): %v", re, err)
+	assert.Equal(t, re, flag.String(), "set flag displays as '%s', expected '%s'", flag.String(), re)
+	for _, text := range texts {
+		assert.False(t, flag.unmatched(text), "flag '%s' claims a non-match for '%s'", flag.String(), text)
+	}
+}
+
+func assertUnmatch(t *testing.T, re string, flag *regexFlag, texts ...string) {
+	t.Helper()
+	err := flag.Set(re)
+	require.Nil(t, err, "unexpected error with flag.Set('%s'): %v", re, err)
+	assert.Equal(t, re, flag.String())
+	for _, text := range texts {
+		assert.True(t, flag.unmatched(text), "flag '%s' should claim a non-match for '%s'", flag.String(), text)
+	}
+}
+
+func TestRegexFlag(t *testing.T) {
+	var flag regexFlag
+	assert.Equal(t, "<none>", flag.String())
+	assert.Equal(t, "regular expression", flag.Type())
+	assert.False(t, flag.unmatched("foobar"), "unset flag claims a non-match")
+	assert.EqualError(t, flag.Set("*"), "error parsing regexp: missing argument to repetition operator: `*`")
+	assertMatch(t, "foo.*bar", new(regexFlag), "foobar", "foo bar", "x foobar y", "xfoobary", "xfooybarz")
+	assertUnmatch(t, "foo.*bar", new(regexFlag), "Foobar", "foo Bar", "fxoobar", "whatever")
+}