kevin9327
18c7de0920
SizeSuffix.Set and CountSuffix.Set accepted "nan" before a suffix, so
--max-transfer nank became -1 ("off") without an error. CountSuffix.Set
also had no overflow check, unlike SizeSuffix, so 10E or infk wrapped
to the same value.
157 lines
3.4 KiB
Go
157 lines
3.4 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},
|
|
{"9E", 9 * 1000 * 1000 * 1000 * 1000 * 1000 * 1000, false},
|
|
{"9.3E", 0, true},
|
|
{"10E", 0, true},
|
|
{"infk", 0, true},
|
|
{"nank", 0, true},
|
|
{"NaNM", 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))
|
|
}
|
|
}
|