github.com/containers/libpod@v1.9.4-0.20220419124438-4284fd425507/cmd/podmanV2/parse/net_test.go (about)

     1  //nolint
     2  // most of these validate and parse functions have been taken from projectatomic/docker
     3  // and modified for cri-o
     4  package parse
     5  
     6  import (
     7  	"io/ioutil"
     8  	"os"
     9  	"testing"
    10  
    11  	"github.com/stretchr/testify/assert"
    12  )
    13  
    14  var (
    15  	Var1 = []string{"ONE=1", "TWO=2"}
    16  )
    17  
    18  func createTmpFile(content []byte) (string, error) {
    19  	tmpfile, err := ioutil.TempFile(os.TempDir(), "unittest")
    20  	if err != nil {
    21  		return "", err
    22  	}
    23  
    24  	if _, err := tmpfile.Write(content); err != nil {
    25  		return "", err
    26  
    27  	}
    28  	if err := tmpfile.Close(); err != nil {
    29  		return "", err
    30  	}
    31  	return tmpfile.Name(), nil
    32  }
    33  
    34  func TestValidateExtraHost(t *testing.T) {
    35  	type args struct {
    36  		val string
    37  	}
    38  	tests := []struct {
    39  		name    string
    40  		args    args
    41  		want    string
    42  		wantErr bool
    43  	}{
    44  		//2001:0db8:85a3:0000:0000:8a2e:0370:7334
    45  		{name: "good-ipv4", args: args{val: "foobar:192.168.1.1"}, want: "foobar:192.168.1.1", wantErr: false},
    46  		{name: "bad-ipv4", args: args{val: "foobar:999.999.999.99"}, want: "", wantErr: true},
    47  		{name: "bad-ipv4", args: args{val: "foobar:999.999.999"}, want: "", wantErr: true},
    48  		{name: "noname-ipv4", args: args{val: "192.168.1.1"}, want: "", wantErr: true},
    49  		{name: "noname-ipv4", args: args{val: ":192.168.1.1"}, want: "", wantErr: true},
    50  		{name: "noip", args: args{val: "foobar:"}, want: "", wantErr: true},
    51  		{name: "noip", args: args{val: "foobar"}, want: "", wantErr: true},
    52  		{name: "good-ipv6", args: args{val: "foobar:2001:0db8:85a3:0000:0000:8a2e:0370:7334"}, want: "foobar:2001:0db8:85a3:0000:0000:8a2e:0370:7334", wantErr: false},
    53  		{name: "bad-ipv6", args: args{val: "foobar:0db8:85a3:0000:0000:8a2e:0370:7334"}, want: "", wantErr: true},
    54  		{name: "bad-ipv6", args: args{val: "foobar:0db8:85a3:0000:0000:8a2e:0370:7334.0000.0000.000"}, want: "", wantErr: true},
    55  		{name: "noname-ipv6", args: args{val: "2001:0db8:85a3:0000:0000:8a2e:0370:7334"}, want: "", wantErr: true},
    56  		{name: "noname-ipv6", args: args{val: ":2001:0db8:85a3:0000:0000:8a2e:0370:7334"}, want: "", wantErr: true},
    57  	}
    58  	for _, tt := range tests {
    59  		t.Run(tt.name, func(t *testing.T) {
    60  			got, err := ValidateExtraHost(tt.args.val)
    61  			if (err != nil) != tt.wantErr {
    62  				t.Errorf("ValidateExtraHost() error = %v, wantErr %v", err, tt.wantErr)
    63  				return
    64  			}
    65  			if got != tt.want {
    66  				t.Errorf("ValidateExtraHost() = %v, want %v", got, tt.want)
    67  			}
    68  		})
    69  	}
    70  }
    71  
    72  func Test_validateIPAddress(t *testing.T) {
    73  	type args struct {
    74  		val string
    75  	}
    76  	tests := []struct {
    77  		name    string
    78  		args    args
    79  		want    string
    80  		wantErr bool
    81  	}{
    82  		{name: "ipv4-good", args: args{val: "192.168.1.1"}, want: "192.168.1.1", wantErr: false},
    83  		{name: "ipv4-bad", args: args{val: "192.168.1.1.1"}, want: "", wantErr: true},
    84  		{name: "ipv4-bad", args: args{val: "192."}, want: "", wantErr: true},
    85  	}
    86  	for _, tt := range tests {
    87  		t.Run(tt.name, func(t *testing.T) {
    88  			got, err := validateIPAddress(tt.args.val)
    89  			if (err != nil) != tt.wantErr {
    90  				t.Errorf("validateIPAddress() error = %v, wantErr %v", err, tt.wantErr)
    91  				return
    92  			}
    93  			if got != tt.want {
    94  				t.Errorf("validateIPAddress() = %v, want %v", got, tt.want)
    95  			}
    96  		})
    97  	}
    98  }
    99  
   100  func TestValidateFileName(t *testing.T) {
   101  	type args struct {
   102  		filename string
   103  	}
   104  	tests := []struct {
   105  		name    string
   106  		args    args
   107  		wantErr bool
   108  	}{
   109  		{name: "good", args: args{filename: "/some/rand/path"}, wantErr: false},
   110  		{name: "good", args: args{filename: "some/rand/path"}, wantErr: false},
   111  		{name: "good", args: args{filename: "/"}, wantErr: false},
   112  		{name: "bad", args: args{filename: "/:"}, wantErr: true},
   113  		{name: "bad", args: args{filename: ":/"}, wantErr: true},
   114  		{name: "bad", args: args{filename: "/some/rand:/path"}, wantErr: true},
   115  	}
   116  	for _, tt := range tests {
   117  		t.Run(tt.name, func(t *testing.T) {
   118  			if err := ValidateFileName(tt.args.filename); (err != nil) != tt.wantErr {
   119  				t.Errorf("ValidateFileName() error = %v, wantErr %v", err, tt.wantErr)
   120  			}
   121  		})
   122  	}
   123  }
   124  
   125  func TestGetAllLabels(t *testing.T) {
   126  	fileLabels := []string{}
   127  	labels, _ := GetAllLabels(fileLabels, Var1)
   128  	assert.Equal(t, len(labels), 2)
   129  }
   130  
   131  func TestGetAllLabelsBadKeyValue(t *testing.T) {
   132  	inLabels := []string{"=badValue", "="}
   133  	fileLabels := []string{}
   134  	_, err := GetAllLabels(fileLabels, inLabels)
   135  	assert.Error(t, err, assert.AnError)
   136  }
   137  
   138  func TestGetAllLabelsBadLabelFile(t *testing.T) {
   139  	fileLabels := []string{"/foobar5001/be"}
   140  	_, err := GetAllLabels(fileLabels, Var1)
   141  	assert.Error(t, err, assert.AnError)
   142  }
   143  
   144  func TestGetAllLabelsFile(t *testing.T) {
   145  	content := []byte("THREE=3")
   146  	tFile, err := createTmpFile(content)
   147  	defer os.Remove(tFile)
   148  	assert.NoError(t, err)
   149  	fileLabels := []string{tFile}
   150  	result, _ := GetAllLabels(fileLabels, Var1)
   151  	assert.Equal(t, len(result), 3)
   152  }