github.1git.de/docker/cli@v26.1.3+incompatible/cli/command/swarm/ipnet_slice_test.go (about)

     1  package swarm
     2  
     3  import (
     4  	"fmt"
     5  	"net"
     6  	"strings"
     7  	"testing"
     8  
     9  	"github.com/spf13/pflag"
    10  )
    11  
    12  // Helper function to set static slices
    13  func getCIDR(_ net.IP, cidr *net.IPNet, _ error) net.IPNet {
    14  	return *cidr
    15  }
    16  
    17  func equalCIDR(c1 net.IPNet, c2 net.IPNet) bool {
    18  	return c1.String() == c2.String()
    19  }
    20  
    21  func setUpIPNetFlagSet(ipsp *[]net.IPNet) *pflag.FlagSet {
    22  	f := pflag.NewFlagSet("test", pflag.ContinueOnError)
    23  	f.Var(newIPNetSliceValue([]net.IPNet{}, ipsp), "cidrs", "Command separated list!")
    24  	return f
    25  }
    26  
    27  func TestIPNets(t *testing.T) {
    28  	var ips []net.IPNet
    29  	f := setUpIPNetFlagSet(&ips)
    30  
    31  	vals := []string{"192.168.1.1/24", "10.0.0.1/16", "fd00:0:0:0:0:0:0:2/64"}
    32  	arg := fmt.Sprintf("--cidrs=%s", strings.Join(vals, ","))
    33  	err := f.Parse([]string{arg})
    34  	if err != nil {
    35  		t.Fatal("expected no error; got", err)
    36  	}
    37  	for i, v := range ips {
    38  		if _, cidr, _ := net.ParseCIDR(vals[i]); cidr == nil {
    39  			t.Fatalf("invalid string being converted to CIDR: %s", vals[i])
    40  		} else if !equalCIDR(*cidr, v) {
    41  			t.Fatalf("expected ips[%d] to be %s but got: %s from GetIPSlice", i, vals[i], v)
    42  		}
    43  	}
    44  }
    45  
    46  func TestIPNetCalledTwice(t *testing.T) {
    47  	var cidrs []net.IPNet
    48  	f := setUpIPNetFlagSet(&cidrs)
    49  
    50  	in := []string{"192.168.1.2/16,fd00::/64", "10.0.0.1/24"}
    51  
    52  	expected := []net.IPNet{
    53  		getCIDR(net.ParseCIDR("192.168.1.2/16")),
    54  		getCIDR(net.ParseCIDR("fd00::/64")),
    55  		getCIDR(net.ParseCIDR("10.0.0.1/24")),
    56  	}
    57  	argfmt := "--cidrs=%s"
    58  	arg1 := fmt.Sprintf(argfmt, in[0])
    59  	arg2 := fmt.Sprintf(argfmt, in[1])
    60  	err := f.Parse([]string{arg1, arg2})
    61  	if err != nil {
    62  		t.Fatal("expected no error; got", err)
    63  	}
    64  	for i, v := range cidrs {
    65  		if !equalCIDR(expected[i], v) {
    66  			t.Fatalf("expected cidrs[%d] to be %s but got: %s", i, expected[i], v)
    67  		}
    68  	}
    69  }
    70  
    71  func TestIPNetBadQuoting(t *testing.T) {
    72  	tests := []struct {
    73  		Want    []net.IPNet
    74  		FlagArg []string
    75  	}{
    76  		{
    77  			Want: []net.IPNet{
    78  				getCIDR(net.ParseCIDR("a4ab:61d:f03e:5d7d:fad7:d4c2:a1a5:568/128")),
    79  				getCIDR(net.ParseCIDR("203.107.49.208/32")),
    80  				getCIDR(net.ParseCIDR("14.57.204.90/32")),
    81  			},
    82  			FlagArg: []string{
    83  				"a4ab:61d:f03e:5d7d:fad7:d4c2:a1a5:568/128",
    84  				"203.107.49.208/32",
    85  				"14.57.204.90/32",
    86  			},
    87  		},
    88  		{
    89  			Want: []net.IPNet{
    90  				getCIDR(net.ParseCIDR("204.228.73.195/32")),
    91  				getCIDR(net.ParseCIDR("86.141.15.94/32")),
    92  			},
    93  			FlagArg: []string{
    94  				"204.228.73.195/32",
    95  				"86.141.15.94/32",
    96  			},
    97  		},
    98  		{
    99  			Want: []net.IPNet{
   100  				getCIDR(net.ParseCIDR("c70c:db36:3001:890f:c6ea:3f9b:7a39:cc3f/128")),
   101  				getCIDR(net.ParseCIDR("4d17:1d6e:e699:bd7a:88c5:5e7e:ac6a:4472/128")),
   102  			},
   103  			FlagArg: []string{
   104  				"c70c:db36:3001:890f:c6ea:3f9b:7a39:cc3f/128",
   105  				"4d17:1d6e:e699:bd7a:88c5:5e7e:ac6a:4472/128",
   106  			},
   107  		},
   108  		{
   109  			Want: []net.IPNet{
   110  				getCIDR(net.ParseCIDR("5170:f971:cfac:7be3:512a:af37:952c:bc33/128")),
   111  				getCIDR(net.ParseCIDR("93.21.145.140/32")),
   112  				getCIDR(net.ParseCIDR("2cac:61d3:c5ff:6caf:73e0:1b1a:c336:c1ca/128")),
   113  			},
   114  			FlagArg: []string{
   115  				" 5170:f971:cfac:7be3:512a:af37:952c:bc33/128  , 93.21.145.140/32     ",
   116  				"2cac:61d3:c5ff:6caf:73e0:1b1a:c336:c1ca/128",
   117  			},
   118  		},
   119  		{
   120  			Want: []net.IPNet{
   121  				getCIDR(net.ParseCIDR("2e5e:66b2:6441:848:5b74:76ea:574c:3a7b/128")),
   122  				getCIDR(net.ParseCIDR("2e5e:66b2:6441:848:5b74:76ea:574c:3a7b/128")),
   123  				getCIDR(net.ParseCIDR("2e5e:66b2:6441:848:5b74:76ea:574c:3a7b/128")),
   124  				getCIDR(net.ParseCIDR("2e5e:66b2:6441:848:5b74:76ea:574c:3a7b/128")),
   125  			},
   126  			FlagArg: []string{
   127  				`"2e5e:66b2:6441:848:5b74:76ea:574c:3a7b/128,        2e5e:66b2:6441:848:5b74:76ea:574c:3a7b/128,2e5e:66b2:6441:848:5b74:76ea:574c:3a7b/128     "`,
   128  				" 2e5e:66b2:6441:848:5b74:76ea:574c:3a7b/128",
   129  			},
   130  		},
   131  	}
   132  
   133  	for i, test := range tests {
   134  		var cidrs []net.IPNet
   135  		f := setUpIPNetFlagSet(&cidrs)
   136  
   137  		if err := f.Parse([]string{fmt.Sprintf("--cidrs=%s", strings.Join(test.FlagArg, ","))}); err != nil {
   138  			t.Fatalf("flag parsing failed with error: %s\nparsing:\t%#v\nwant:\t\t%s",
   139  				err, test.FlagArg, test.Want[i])
   140  		}
   141  
   142  		for j, b := range cidrs {
   143  			if !equalCIDR(b, test.Want[j]) {
   144  				t.Fatalf("bad value parsed for test %d on net.IP %d:\nwant:\t%s\ngot:\t%s", i, j, test.Want[j], b)
   145  			}
   146  		}
   147  	}
   148  }