87 lines
2.1 KiB
Go
87 lines
2.1 KiB
Go
package fs
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// Check it satisfies the interface
|
|
var _ flagger = (*Tristate)(nil)
|
|
|
|
func TestTristateString(t *testing.T) {
|
|
for _, test := range []struct {
|
|
in Tristate
|
|
want string
|
|
}{
|
|
{Tristate{}, "unset"},
|
|
{Tristate{Valid: false, Value: false}, "unset"},
|
|
{Tristate{Valid: false, Value: true}, "unset"},
|
|
{Tristate{Valid: true, Value: false}, "false"},
|
|
{Tristate{Valid: true, Value: true}, "true"},
|
|
} {
|
|
got := test.in.String()
|
|
assert.Equal(t, test.want, got)
|
|
}
|
|
}
|
|
|
|
func TestTristateSet(t *testing.T) {
|
|
for _, test := range []struct {
|
|
in string
|
|
want Tristate
|
|
err bool
|
|
}{
|
|
{"", Tristate{Valid: false, Value: false}, false},
|
|
{"nil", Tristate{Valid: false, Value: false}, false},
|
|
{"null", Tristate{Valid: false, Value: false}, false},
|
|
{"UNSET", Tristate{Valid: false, Value: false}, false},
|
|
{"true", Tristate{Valid: true, Value: true}, false},
|
|
{"1", Tristate{Valid: true, Value: true}, false},
|
|
{"false", Tristate{Valid: true, Value: false}, false},
|
|
{"0", Tristate{Valid: true, Value: false}, false},
|
|
{"potato", Tristate{Valid: false, Value: false}, true},
|
|
} {
|
|
var got Tristate
|
|
err := got.Set(test.in)
|
|
if test.err {
|
|
require.Error(t, err)
|
|
} else {
|
|
require.NoError(t, err)
|
|
assert.Equal(t, test.want, got)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestTristateScan(t *testing.T) {
|
|
var v Tristate
|
|
n, err := fmt.Sscan(" true ", &v)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 1, n)
|
|
assert.Equal(t, Tristate{Valid: true, Value: true}, v)
|
|
}
|
|
|
|
func TestTristateUnmarshalJSON(t *testing.T) {
|
|
for _, test := range []struct {
|
|
in string
|
|
want Tristate
|
|
err bool
|
|
}{
|
|
{`null`, Tristate{}, false},
|
|
{`true`, Tristate{Valid: true, Value: true}, false},
|
|
{`false`, Tristate{Valid: true, Value: false}, false},
|
|
{`potato`, Tristate{}, true},
|
|
{``, Tristate{}, true},
|
|
} {
|
|
var got Tristate
|
|
err := json.Unmarshal([]byte(test.in), &got)
|
|
if test.err {
|
|
require.Error(t, err, test.in)
|
|
} else {
|
|
require.NoError(t, err, test.in)
|
|
}
|
|
assert.Equal(t, test.want, got, test.in)
|
|
}
|
|
}
|