mirror of
https://github.com/restic/restic.git
synced 2024-08-23 12:50:45 +02:00
162 lines
3.7 KiB
Go
162 lines
3.7 KiB
Go
|
package pflag
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"strconv"
|
||
|
"strings"
|
||
|
"testing"
|
||
|
)
|
||
|
|
||
|
func setUpUISFlagSet(uisp *[]uint) *FlagSet {
|
||
|
f := NewFlagSet("test", ContinueOnError)
|
||
|
f.UintSliceVar(uisp, "uis", []uint{}, "Command separated list!")
|
||
|
return f
|
||
|
}
|
||
|
|
||
|
func setUpUISFlagSetWithDefault(uisp *[]uint) *FlagSet {
|
||
|
f := NewFlagSet("test", ContinueOnError)
|
||
|
f.UintSliceVar(uisp, "uis", []uint{0, 1}, "Command separated list!")
|
||
|
return f
|
||
|
}
|
||
|
|
||
|
func TestEmptyUIS(t *testing.T) {
|
||
|
var uis []uint
|
||
|
f := setUpUISFlagSet(&uis)
|
||
|
err := f.Parse([]string{})
|
||
|
if err != nil {
|
||
|
t.Fatal("expected no error; got", err)
|
||
|
}
|
||
|
|
||
|
getUIS, err := f.GetUintSlice("uis")
|
||
|
if err != nil {
|
||
|
t.Fatal("got an error from GetUintSlice():", err)
|
||
|
}
|
||
|
if len(getUIS) != 0 {
|
||
|
t.Fatalf("got is %v with len=%d but expected length=0", getUIS, len(getUIS))
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestUIS(t *testing.T) {
|
||
|
var uis []uint
|
||
|
f := setUpUISFlagSet(&uis)
|
||
|
|
||
|
vals := []string{"1", "2", "4", "3"}
|
||
|
arg := fmt.Sprintf("--uis=%s", strings.Join(vals, ","))
|
||
|
err := f.Parse([]string{arg})
|
||
|
if err != nil {
|
||
|
t.Fatal("expected no error; got", err)
|
||
|
}
|
||
|
for i, v := range uis {
|
||
|
u, err := strconv.ParseUint(vals[i], 10, 0)
|
||
|
if err != nil {
|
||
|
t.Fatalf("got error: %v", err)
|
||
|
}
|
||
|
if uint(u) != v {
|
||
|
t.Fatalf("expected uis[%d] to be %s but got %d", i, vals[i], v)
|
||
|
}
|
||
|
}
|
||
|
getUIS, err := f.GetUintSlice("uis")
|
||
|
if err != nil {
|
||
|
t.Fatalf("got error: %v", err)
|
||
|
}
|
||
|
for i, v := range getUIS {
|
||
|
u, err := strconv.ParseUint(vals[i], 10, 0)
|
||
|
if err != nil {
|
||
|
t.Fatalf("got error: %v", err)
|
||
|
}
|
||
|
if uint(u) != v {
|
||
|
t.Fatalf("expected uis[%d] to be %s but got: %d from GetUintSlice", i, vals[i], v)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestUISDefault(t *testing.T) {
|
||
|
var uis []uint
|
||
|
f := setUpUISFlagSetWithDefault(&uis)
|
||
|
|
||
|
vals := []string{"0", "1"}
|
||
|
|
||
|
err := f.Parse([]string{})
|
||
|
if err != nil {
|
||
|
t.Fatal("expected no error; got", err)
|
||
|
}
|
||
|
for i, v := range uis {
|
||
|
u, err := strconv.ParseUint(vals[i], 10, 0)
|
||
|
if err != nil {
|
||
|
t.Fatalf("got error: %v", err)
|
||
|
}
|
||
|
if uint(u) != v {
|
||
|
t.Fatalf("expect uis[%d] to be %d but got: %d", i, u, v)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
getUIS, err := f.GetUintSlice("uis")
|
||
|
if err != nil {
|
||
|
t.Fatal("got an error from GetUintSlice():", err)
|
||
|
}
|
||
|
for i, v := range getUIS {
|
||
|
u, err := strconv.ParseUint(vals[i], 10, 0)
|
||
|
if err != nil {
|
||
|
t.Fatal("got an error from GetIntSlice():", err)
|
||
|
}
|
||
|
if uint(u) != v {
|
||
|
t.Fatalf("expected uis[%d] to be %d from GetUintSlice but got: %d", i, u, v)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestUISWithDefault(t *testing.T) {
|
||
|
var uis []uint
|
||
|
f := setUpUISFlagSetWithDefault(&uis)
|
||
|
|
||
|
vals := []string{"1", "2"}
|
||
|
arg := fmt.Sprintf("--uis=%s", strings.Join(vals, ","))
|
||
|
err := f.Parse([]string{arg})
|
||
|
if err != nil {
|
||
|
t.Fatal("expected no error; got", err)
|
||
|
}
|
||
|
for i, v := range uis {
|
||
|
u, err := strconv.ParseUint(vals[i], 10, 0)
|
||
|
if err != nil {
|
||
|
t.Fatalf("got error: %v", err)
|
||
|
}
|
||
|
if uint(u) != v {
|
||
|
t.Fatalf("expected uis[%d] to be %d from GetUintSlice but got: %d", i, u, v)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
getUIS, err := f.GetUintSlice("uis")
|
||
|
if err != nil {
|
||
|
t.Fatal("got an error from GetUintSlice():", err)
|
||
|
}
|
||
|
for i, v := range getUIS {
|
||
|
u, err := strconv.ParseUint(vals[i], 10, 0)
|
||
|
if err != nil {
|
||
|
t.Fatalf("got error: %v", err)
|
||
|
}
|
||
|
if uint(u) != v {
|
||
|
t.Fatalf("expected uis[%d] to be %d from GetUintSlice but got: %d", i, u, v)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestUISCalledTwice(t *testing.T) {
|
||
|
var uis []uint
|
||
|
f := setUpUISFlagSet(&uis)
|
||
|
|
||
|
in := []string{"1,2", "3"}
|
||
|
expected := []int{1, 2, 3}
|
||
|
argfmt := "--uis=%s"
|
||
|
arg1 := fmt.Sprintf(argfmt, in[0])
|
||
|
arg2 := fmt.Sprintf(argfmt, in[1])
|
||
|
err := f.Parse([]string{arg1, arg2})
|
||
|
if err != nil {
|
||
|
t.Fatal("expected no error; got", err)
|
||
|
}
|
||
|
for i, v := range uis {
|
||
|
if uint(expected[i]) != v {
|
||
|
t.Fatalf("expected uis[%d] to be %d but got: %d", i, expected[i], v)
|
||
|
}
|
||
|
}
|
||
|
}
|