ae3963e4b4
Before this change options were read and set in native format. This means for example nanoseconds for durations or an integer for enumerated types, which isn't very convenient for humans. This change enables these types to be set with a string with the syntax as used in the command line instead, so `"10s"` rather than `10000000000` or `"DEBUG"` rather than `8` for log level.
95 lines
2.9 KiB
Go
95 lines
2.9 KiB
Go
package fs
|
|
|
|
import (
|
|
"encoding/json"
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
// Check it satisfies the interface
|
|
var _ flagger = (*DumpFlags)(nil)
|
|
|
|
func TestDumpFlagsString(t *testing.T) {
|
|
assert.Equal(t, "", DumpFlags(0).String())
|
|
assert.Equal(t, "headers", (DumpHeaders).String())
|
|
assert.Equal(t, "headers,bodies", (DumpHeaders | DumpBodies).String())
|
|
assert.Equal(t, "headers,bodies,requests,responses,auth,filters", (DumpHeaders | DumpBodies | DumpRequests | DumpResponses | DumpAuth | DumpFilters).String())
|
|
assert.Equal(t, "headers,Unknown-0x8000", (DumpHeaders | DumpFlags(0x8000)).String())
|
|
}
|
|
|
|
func TestDumpFlagsSet(t *testing.T) {
|
|
for _, test := range []struct {
|
|
in string
|
|
want DumpFlags
|
|
wantErr string
|
|
}{
|
|
{"", DumpFlags(0), ""},
|
|
{"bodies", DumpBodies, ""},
|
|
{"bodies,headers,auth", DumpBodies | DumpHeaders | DumpAuth, ""},
|
|
{"bodies,headers,auth", DumpBodies | DumpHeaders | DumpAuth, ""},
|
|
{"headers,bodies,requests,responses,auth,filters", DumpHeaders | DumpBodies | DumpRequests | DumpResponses | DumpAuth | DumpFilters, ""},
|
|
{"headers,bodies,unknown,auth", 0, "Unknown dump flag \"unknown\""},
|
|
} {
|
|
f := DumpFlags(-1)
|
|
initial := f
|
|
err := f.Set(test.in)
|
|
if err != nil {
|
|
if test.wantErr == "" {
|
|
t.Errorf("Got an error when not expecting one on %q: %v", test.in, err)
|
|
} else {
|
|
assert.Contains(t, err.Error(), test.wantErr)
|
|
}
|
|
assert.Equal(t, initial, f, test.want)
|
|
} else {
|
|
if test.wantErr != "" {
|
|
t.Errorf("Got no error when expecting one on %q", test.in)
|
|
} else {
|
|
assert.Equal(t, test.want, f)
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
func TestDumpFlagsType(t *testing.T) {
|
|
f := DumpFlags(0)
|
|
assert.Equal(t, "DumpFlags", f.Type())
|
|
}
|
|
|
|
func TestDumpFlagsUnmarshallJSON(t *testing.T) {
|
|
for _, test := range []struct {
|
|
in string
|
|
want DumpFlags
|
|
wantErr string
|
|
}{
|
|
{`""`, DumpFlags(0), ""},
|
|
{`"bodies"`, DumpBodies, ""},
|
|
{`"bodies,headers,auth"`, DumpBodies | DumpHeaders | DumpAuth, ""},
|
|
{`"bodies,headers,auth"`, DumpBodies | DumpHeaders | DumpAuth, ""},
|
|
{`"headers,bodies,requests,responses,auth,filters"`, DumpHeaders | DumpBodies | DumpRequests | DumpResponses | DumpAuth | DumpFilters, ""},
|
|
{`"headers,bodies,unknown,auth"`, 0, "Unknown dump flag \"unknown\""},
|
|
{`0`, DumpFlags(0), ""},
|
|
{strconv.Itoa(int(DumpBodies)), DumpBodies, ""},
|
|
{strconv.Itoa(int(DumpBodies | DumpHeaders | DumpAuth)), DumpBodies | DumpHeaders | DumpAuth, ""},
|
|
} {
|
|
f := DumpFlags(-1)
|
|
initial := f
|
|
err := json.Unmarshal([]byte(test.in), &f)
|
|
if err != nil {
|
|
if test.wantErr == "" {
|
|
t.Errorf("Got an error when not expecting one on %q: %v", test.in, err)
|
|
} else {
|
|
assert.Contains(t, err.Error(), test.wantErr)
|
|
}
|
|
assert.Equal(t, initial, f, test.want)
|
|
} else {
|
|
if test.wantErr != "" {
|
|
t.Errorf("Got no error when expecting one on %q", test.in)
|
|
} else {
|
|
assert.Equal(t, test.want, f)
|
|
}
|
|
}
|
|
}
|
|
}
|