82 lines
1.9 KiB
Go
82 lines
1.9 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestReadDurationArgs(t *testing.T) {
|
||
|
|
cases := []struct {
|
||
|
|
name string
|
||
|
|
args []string
|
||
|
|
expectedDuration time.Duration
|
||
|
|
resultArgs []string
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "No args",
|
||
|
|
args: []string{},
|
||
|
|
expectedDuration: 0,
|
||
|
|
resultArgs: []string{},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "No durations",
|
||
|
|
args: []string{"foo", "bar"},
|
||
|
|
expectedDuration: 0,
|
||
|
|
resultArgs: []string{"foo", "bar"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "duration at beginning",
|
||
|
|
args: []string{"5m", "bar"},
|
||
|
|
expectedDuration: time.Minute * 5,
|
||
|
|
resultArgs: []string{"bar"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "duration in the middle",
|
||
|
|
args: []string{"foo", "5m", "bar"},
|
||
|
|
expectedDuration: 0,
|
||
|
|
resultArgs: []string{"foo", "5m", "bar"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "duration at end without prefix",
|
||
|
|
args: []string{"foo", "bar", "5m"},
|
||
|
|
expectedDuration: 0,
|
||
|
|
resultArgs: []string{"foo", "bar", "5m"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "duration at end with prefix",
|
||
|
|
args: []string{"foo", "for", "5m"},
|
||
|
|
expectedDuration: time.Minute * 5,
|
||
|
|
resultArgs: []string{"foo"},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, c := range cases {
|
||
|
|
c := c
|
||
|
|
t.Run(c.name, func(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
|
||
|
|
actualArgs, actualDuration := readDurationArgs(c.args)
|
||
|
|
|
||
|
|
if len(actualArgs) != len(c.resultArgs) {
|
||
|
|
t.Errorf("expected args %v actual %v", c.resultArgs, actualArgs)
|
||
|
|
}
|
||
|
|
|
||
|
|
for i, expected := range c.resultArgs {
|
||
|
|
if expected != actualArgs[i] {
|
||
|
|
t.Errorf("expected arg[%d] %v actual %v", i, expected, actualArgs[i])
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if actualDuration == nil {
|
||
|
|
if c.expectedDuration != 0 {
|
||
|
|
t.Errorf("expected %v actual %v", c.expectedDuration, actualDuration)
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
if *actualDuration != c.expectedDuration {
|
||
|
|
t.Errorf("expected %v actual %v", c.expectedDuration, actualDuration)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|