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.
150 lines
3.2 KiB
Go
150 lines
3.2 KiB
Go
package fs
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// Check it satisfies the interfaces
|
|
var (
|
|
_ flagger = (*CountSuffix)(nil)
|
|
_ flaggerNP = CountSuffix(0)
|
|
)
|
|
|
|
func TestCountSuffixString(t *testing.T) {
|
|
for _, test := range []struct {
|
|
in float64
|
|
want string
|
|
}{
|
|
{0, "0"},
|
|
{102, "102"},
|
|
{1000, "1k"},
|
|
{1000 * 1000, "1M"},
|
|
{1000 * 1000 * 1000, "1G"},
|
|
{10 * 1000 * 1000 * 1000, "10G"},
|
|
{10.1 * 1000 * 1000 * 1000, "10.100G"},
|
|
{-1, "off"},
|
|
{-100, "off"},
|
|
} {
|
|
ss := CountSuffix(test.in)
|
|
got := ss.String()
|
|
assert.Equal(t, test.want, got)
|
|
}
|
|
}
|
|
|
|
func TestCountSuffixUnit(t *testing.T) {
|
|
for _, test := range []struct {
|
|
in float64
|
|
want string
|
|
}{
|
|
{0, "0 Byte"},
|
|
{102, "102 Byte"},
|
|
{1000, "1 kByte"},
|
|
{1000 * 1000, "1 MByte"},
|
|
{1000 * 1000 * 1000, "1 GByte"},
|
|
{10 * 1000 * 1000 * 1000, "10 GByte"},
|
|
{10.1 * 1000 * 1000 * 1000, "10.100 GByte"},
|
|
{10 * 1000 * 1000 * 1000 * 1000, "10 TByte"},
|
|
{10 * 1000 * 1000 * 1000 * 1000 * 1000, "10 PByte"},
|
|
{1 * 1000 * 1000 * 1000 * 1000 * 1000 * 1000, "1 EByte"},
|
|
{-1, "off"},
|
|
{-100, "off"},
|
|
} {
|
|
ss := CountSuffix(test.in)
|
|
got := ss.Unit("Byte")
|
|
assert.Equal(t, test.want, got)
|
|
}
|
|
}
|
|
|
|
func TestCountSuffixSet(t *testing.T) {
|
|
for _, test := range []struct {
|
|
in string
|
|
want int64
|
|
err bool
|
|
}{
|
|
{"0", 0, false},
|
|
{"1b", 1, false},
|
|
{"100B", 100, false},
|
|
{"0.1k", 100, false},
|
|
{"0.1", 100, false},
|
|
{"1K", 1000, false},
|
|
{"1k", 1000, false},
|
|
{"1KB", 1000, false},
|
|
{"1kB", 1000, false},
|
|
{"1kb", 1000, false},
|
|
{"1", 1000, false},
|
|
{"2.5", 1000 * 2.5, false},
|
|
{"1M", 1000 * 1000, false},
|
|
{"1MB", 1000 * 1000, false},
|
|
{"1.g", 1000 * 1000 * 1000, false},
|
|
{"10G", 10 * 1000 * 1000 * 1000, false},
|
|
{"10T", 10 * 1000 * 1000 * 1000 * 1000, false},
|
|
{"10P", 10 * 1000 * 1000 * 1000 * 1000 * 1000, false},
|
|
{"off", -1, false},
|
|
{"OFF", -1, false},
|
|
{"", 0, true},
|
|
{"1q", 0, true},
|
|
{"1.q", 0, true},
|
|
{"1q", 0, true},
|
|
{"-1K", 0, true},
|
|
{"1i", 0, true},
|
|
{"1iB", 0, true},
|
|
{"1Ki", 0, true},
|
|
{"1KiB", 0, true},
|
|
} {
|
|
ss := CountSuffix(0)
|
|
err := ss.Set(test.in)
|
|
if test.err {
|
|
require.Error(t, err, test.in)
|
|
} else {
|
|
require.NoError(t, err, test.in)
|
|
}
|
|
assert.Equal(t, test.want, int64(ss))
|
|
}
|
|
}
|
|
|
|
func TestCountSuffixScan(t *testing.T) {
|
|
var v CountSuffix
|
|
n, err := fmt.Sscan(" 17M ", &v)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 1, n)
|
|
assert.Equal(t, CountSuffix(17000000), v)
|
|
}
|
|
|
|
func TestCountSuffixUnmarshalJSON(t *testing.T) {
|
|
for _, test := range []struct {
|
|
in string
|
|
want int64
|
|
err bool
|
|
}{
|
|
{`"0"`, 0, false},
|
|
{`"102B"`, 102, false},
|
|
{`"1K"`, 1000, false},
|
|
{`"2.5"`, 1000 * 2.5, false},
|
|
{`"1M"`, 1000 * 1000, false},
|
|
{`"1.g"`, 1000 * 1000 * 1000, false},
|
|
{`"10G"`, 10 * 1000 * 1000 * 1000, false},
|
|
{`"off"`, -1, false},
|
|
{`""`, 0, true},
|
|
{`"1q"`, 0, true},
|
|
{`"-1K"`, 0, true},
|
|
{`0`, 0, false},
|
|
{`102`, 102, false},
|
|
{`1000`, 1000, false},
|
|
{`1000000000`, 1000000000, false},
|
|
{`1.1.1`, 0, true},
|
|
} {
|
|
var ss CountSuffix
|
|
err := json.Unmarshal([]byte(test.in), &ss)
|
|
if test.err {
|
|
require.Error(t, err, test.in)
|
|
} else {
|
|
require.NoError(t, err, test.in)
|
|
}
|
|
assert.Equal(t, test.want, int64(ss))
|
|
}
|
|
}
|