2018-01-13 00:30:54 +08:00
|
|
|
package fs
|
|
|
|
|
|
|
|
import (
|
|
|
|
"testing"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/spf13/pflag"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
)
|
|
|
|
|
|
|
|
// Check it satisfies the interface
|
|
|
|
var _ pflag.Value = (*Duration)(nil)
|
|
|
|
|
|
|
|
func TestParseDuration(t *testing.T) {
|
|
|
|
for _, test := range []struct {
|
|
|
|
in string
|
2018-03-13 04:52:42 +08:00
|
|
|
want time.Duration
|
2018-01-13 00:30:54 +08:00
|
|
|
err bool
|
|
|
|
}{
|
|
|
|
{"0", 0, false},
|
|
|
|
{"", 0, true},
|
2018-03-13 04:52:42 +08:00
|
|
|
{"1ms", time.Millisecond, false},
|
|
|
|
{"1s", time.Second, false},
|
|
|
|
{"1m", time.Minute, false},
|
|
|
|
{"1h", time.Hour, false},
|
|
|
|
{"1d", time.Hour * 24, false},
|
|
|
|
{"1w", time.Hour * 24 * 7, false},
|
|
|
|
{"1M", time.Hour * 24 * 30, false},
|
|
|
|
{"1y", time.Hour * 24 * 365, false},
|
|
|
|
{"1.5y", time.Hour * 24 * 365 * 3 / 2, false},
|
|
|
|
{"-1s", -time.Second, false},
|
|
|
|
{"1.s", time.Second, false},
|
2018-01-13 00:30:54 +08:00
|
|
|
{"1x", 0, true},
|
2018-03-13 04:52:42 +08:00
|
|
|
{"off", time.Duration(DurationOff), false},
|
2018-01-13 00:30:54 +08:00
|
|
|
} {
|
|
|
|
duration, err := ParseDuration(test.in)
|
|
|
|
if test.err {
|
|
|
|
require.Error(t, err)
|
|
|
|
} else {
|
|
|
|
require.NoError(t, err)
|
|
|
|
}
|
2018-03-13 04:52:42 +08:00
|
|
|
assert.Equal(t, test.want, duration)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestDurationString(t *testing.T) {
|
|
|
|
for _, test := range []struct {
|
|
|
|
in time.Duration
|
|
|
|
want string
|
|
|
|
}{
|
2018-04-07 02:04:31 +08:00
|
|
|
{time.Duration(0), "0s"},
|
2018-03-13 04:52:42 +08:00
|
|
|
{time.Second, "1s"},
|
|
|
|
{time.Minute, "1m0s"},
|
|
|
|
{time.Duration(DurationOff), "off"},
|
|
|
|
} {
|
|
|
|
got := Duration(test.in).String()
|
|
|
|
assert.Equal(t, test.want, got)
|
2018-01-13 00:30:54 +08:00
|
|
|
}
|
|
|
|
}
|