3553cc4a5f
Before this change backend types were printing incorrectly as the name of the type, not what was defined by the Type() method. This was not working due to not calling the Type() method. However this needed to be defined on a non-pointer type due to the way the options are handled.
98 lines
2.9 KiB
Go
98 lines
2.9 KiB
Go
package fs
|
|
|
|
import (
|
|
"encoding/json"
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
// Check it satisfies the interfaces
|
|
var (
|
|
_ flagger = (*DumpFlags)(nil)
|
|
_ flaggerNP = DumpFlags(0)
|
|
)
|
|
|
|
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)
|
|
}
|
|
}
|
|
}
|
|
}
|