mirror of
https://github.com/octoleo/restic.git
synced 2024-12-24 20:00:21 +00:00
223 lines
5.7 KiB
Go
223 lines
5.7 KiB
Go
package pflag
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func setUpIPSFlagSet(ipsp *[]net.IP) *FlagSet {
|
|
f := NewFlagSet("test", ContinueOnError)
|
|
f.IPSliceVar(ipsp, "ips", []net.IP{}, "Command separated list!")
|
|
return f
|
|
}
|
|
|
|
func setUpIPSFlagSetWithDefault(ipsp *[]net.IP) *FlagSet {
|
|
f := NewFlagSet("test", ContinueOnError)
|
|
f.IPSliceVar(ipsp, "ips",
|
|
[]net.IP{
|
|
net.ParseIP("192.168.1.1"),
|
|
net.ParseIP("0:0:0:0:0:0:0:1"),
|
|
},
|
|
"Command separated list!")
|
|
return f
|
|
}
|
|
|
|
func TestEmptyIP(t *testing.T) {
|
|
var ips []net.IP
|
|
f := setUpIPSFlagSet(&ips)
|
|
err := f.Parse([]string{})
|
|
if err != nil {
|
|
t.Fatal("expected no error; got", err)
|
|
}
|
|
|
|
getIPS, err := f.GetIPSlice("ips")
|
|
if err != nil {
|
|
t.Fatal("got an error from GetIPSlice():", err)
|
|
}
|
|
if len(getIPS) != 0 {
|
|
t.Fatalf("got ips %v with len=%d but expected length=0", getIPS, len(getIPS))
|
|
}
|
|
}
|
|
|
|
func TestIPS(t *testing.T) {
|
|
var ips []net.IP
|
|
f := setUpIPSFlagSet(&ips)
|
|
|
|
vals := []string{"192.168.1.1", "10.0.0.1", "0:0:0:0:0:0:0:2"}
|
|
arg := fmt.Sprintf("--ips=%s", strings.Join(vals, ","))
|
|
err := f.Parse([]string{arg})
|
|
if err != nil {
|
|
t.Fatal("expected no error; got", err)
|
|
}
|
|
for i, v := range ips {
|
|
if ip := net.ParseIP(vals[i]); ip == nil {
|
|
t.Fatalf("invalid string being converted to IP address: %s", vals[i])
|
|
} else if !ip.Equal(v) {
|
|
t.Fatalf("expected ips[%d] to be %s but got: %s from GetIPSlice", i, vals[i], v)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIPSDefault(t *testing.T) {
|
|
var ips []net.IP
|
|
f := setUpIPSFlagSetWithDefault(&ips)
|
|
|
|
vals := []string{"192.168.1.1", "0:0:0:0:0:0:0:1"}
|
|
err := f.Parse([]string{})
|
|
if err != nil {
|
|
t.Fatal("expected no error; got", err)
|
|
}
|
|
for i, v := range ips {
|
|
if ip := net.ParseIP(vals[i]); ip == nil {
|
|
t.Fatalf("invalid string being converted to IP address: %s", vals[i])
|
|
} else if !ip.Equal(v) {
|
|
t.Fatalf("expected ips[%d] to be %s but got: %s", i, vals[i], v)
|
|
}
|
|
}
|
|
|
|
getIPS, err := f.GetIPSlice("ips")
|
|
if err != nil {
|
|
t.Fatal("got an error from GetIPSlice")
|
|
}
|
|
for i, v := range getIPS {
|
|
if ip := net.ParseIP(vals[i]); ip == nil {
|
|
t.Fatalf("invalid string being converted to IP address: %s", vals[i])
|
|
} else if !ip.Equal(v) {
|
|
t.Fatalf("expected ips[%d] to be %s but got: %s", i, vals[i], v)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIPSWithDefault(t *testing.T) {
|
|
var ips []net.IP
|
|
f := setUpIPSFlagSetWithDefault(&ips)
|
|
|
|
vals := []string{"192.168.1.1", "0:0:0:0:0:0:0:1"}
|
|
arg := fmt.Sprintf("--ips=%s", strings.Join(vals, ","))
|
|
err := f.Parse([]string{arg})
|
|
if err != nil {
|
|
t.Fatal("expected no error; got", err)
|
|
}
|
|
for i, v := range ips {
|
|
if ip := net.ParseIP(vals[i]); ip == nil {
|
|
t.Fatalf("invalid string being converted to IP address: %s", vals[i])
|
|
} else if !ip.Equal(v) {
|
|
t.Fatalf("expected ips[%d] to be %s but got: %s", i, vals[i], v)
|
|
}
|
|
}
|
|
|
|
getIPS, err := f.GetIPSlice("ips")
|
|
if err != nil {
|
|
t.Fatal("got an error from GetIPSlice")
|
|
}
|
|
for i, v := range getIPS {
|
|
if ip := net.ParseIP(vals[i]); ip == nil {
|
|
t.Fatalf("invalid string being converted to IP address: %s", vals[i])
|
|
} else if !ip.Equal(v) {
|
|
t.Fatalf("expected ips[%d] to be %s but got: %s", i, vals[i], v)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIPSCalledTwice(t *testing.T) {
|
|
var ips []net.IP
|
|
f := setUpIPSFlagSet(&ips)
|
|
|
|
in := []string{"192.168.1.2,0:0:0:0:0:0:0:1", "10.0.0.1"}
|
|
expected := []net.IP{net.ParseIP("192.168.1.2"), net.ParseIP("0:0:0:0:0:0:0:1"), net.ParseIP("10.0.0.1")}
|
|
argfmt := "ips=%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 ips {
|
|
if !expected[i].Equal(v) {
|
|
t.Fatalf("expected ips[%d] to be %s but got: %s", i, expected[i], v)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIPSBadQuoting(t *testing.T) {
|
|
|
|
tests := []struct {
|
|
Want []net.IP
|
|
FlagArg []string
|
|
}{
|
|
{
|
|
Want: []net.IP{
|
|
net.ParseIP("a4ab:61d:f03e:5d7d:fad7:d4c2:a1a5:568"),
|
|
net.ParseIP("203.107.49.208"),
|
|
net.ParseIP("14.57.204.90"),
|
|
},
|
|
FlagArg: []string{
|
|
"a4ab:61d:f03e:5d7d:fad7:d4c2:a1a5:568",
|
|
"203.107.49.208",
|
|
"14.57.204.90",
|
|
},
|
|
},
|
|
{
|
|
Want: []net.IP{
|
|
net.ParseIP("204.228.73.195"),
|
|
net.ParseIP("86.141.15.94"),
|
|
},
|
|
FlagArg: []string{
|
|
"204.228.73.195",
|
|
"86.141.15.94",
|
|
},
|
|
},
|
|
{
|
|
Want: []net.IP{
|
|
net.ParseIP("c70c:db36:3001:890f:c6ea:3f9b:7a39:cc3f"),
|
|
net.ParseIP("4d17:1d6e:e699:bd7a:88c5:5e7e:ac6a:4472"),
|
|
},
|
|
FlagArg: []string{
|
|
"c70c:db36:3001:890f:c6ea:3f9b:7a39:cc3f",
|
|
"4d17:1d6e:e699:bd7a:88c5:5e7e:ac6a:4472",
|
|
},
|
|
},
|
|
{
|
|
Want: []net.IP{
|
|
net.ParseIP("5170:f971:cfac:7be3:512a:af37:952c:bc33"),
|
|
net.ParseIP("93.21.145.140"),
|
|
net.ParseIP("2cac:61d3:c5ff:6caf:73e0:1b1a:c336:c1ca"),
|
|
},
|
|
FlagArg: []string{
|
|
" 5170:f971:cfac:7be3:512a:af37:952c:bc33 , 93.21.145.140 ",
|
|
"2cac:61d3:c5ff:6caf:73e0:1b1a:c336:c1ca",
|
|
},
|
|
},
|
|
{
|
|
Want: []net.IP{
|
|
net.ParseIP("2e5e:66b2:6441:848:5b74:76ea:574c:3a7b"),
|
|
net.ParseIP("2e5e:66b2:6441:848:5b74:76ea:574c:3a7b"),
|
|
net.ParseIP("2e5e:66b2:6441:848:5b74:76ea:574c:3a7b"),
|
|
net.ParseIP("2e5e:66b2:6441:848:5b74:76ea:574c:3a7b"),
|
|
},
|
|
FlagArg: []string{
|
|
`"2e5e:66b2:6441:848:5b74:76ea:574c:3a7b, 2e5e:66b2:6441:848:5b74:76ea:574c:3a7b,2e5e:66b2:6441:848:5b74:76ea:574c:3a7b "`,
|
|
" 2e5e:66b2:6441:848:5b74:76ea:574c:3a7b"},
|
|
},
|
|
}
|
|
|
|
for i, test := range tests {
|
|
|
|
var ips []net.IP
|
|
f := setUpIPSFlagSet(&ips)
|
|
|
|
if err := f.Parse([]string{fmt.Sprintf("--ips=%s", strings.Join(test.FlagArg, ","))}); err != nil {
|
|
t.Fatalf("flag parsing failed with error: %s\nparsing:\t%#v\nwant:\t\t%s",
|
|
err, test.FlagArg, test.Want[i])
|
|
}
|
|
|
|
for j, b := range ips {
|
|
if !b.Equal(test.Want[j]) {
|
|
t.Fatalf("bad value parsed for test %d on net.IP %d:\nwant:\t%s\ngot:\t%s", i, j, test.Want[j], b)
|
|
}
|
|
}
|
|
}
|
|
}
|