@@ -19,3 +19,33 @@ func TestFillDefault(t *testing.T) {
1919 assert .DeepEqual (t , userNet .Gateway , net .ParseIP ("192.168.104.1" ))
2020 assert .DeepEqual (t , userNet .DHCPEnd , net.IP {})
2121}
22+
23+ func TestFillDefaultWithV2 (t * testing.T ) {
24+ nwYaml := YAML {Networks : map [string ]Network {
25+ "user-v2" : {Mode : ModeUserV2 },
26+ }}
27+ newYaml , err := fillDefaults (nwYaml )
28+ assert .NilError (t , err )
29+
30+ userNet := newYaml .Networks [ModeUserV2 ]
31+ assert .Equal (t , userNet .Mode , ModeUserV2 )
32+ assert .Equal (t , userNet .Interface , "" )
33+ assert .DeepEqual (t , userNet .NetMask , net .ParseIP ("255.255.255.0" ))
34+ assert .DeepEqual (t , userNet .Gateway , net .ParseIP ("192.168.104.1" ))
35+ assert .DeepEqual (t , userNet .DHCPEnd , net.IP {})
36+ }
37+
38+ func TestFillDefaultWithV2AndGateway (t * testing.T ) {
39+ nwYaml := YAML {Networks : map [string ]Network {
40+ "user-v2" : {Mode : ModeUserV2 , Gateway : net .ParseIP ("192.168.105.1" )},
41+ }}
42+ newYaml , err := fillDefaults (nwYaml )
43+ assert .NilError (t , err )
44+
45+ userNet := newYaml .Networks [ModeUserV2 ]
46+ assert .Equal (t , userNet .Mode , ModeUserV2 )
47+ assert .Equal (t , userNet .Interface , "" )
48+ assert .DeepEqual (t , userNet .NetMask , net.IP {})
49+ assert .DeepEqual (t , userNet .Gateway , net .ParseIP ("192.168.105.1" ))
50+ assert .DeepEqual (t , userNet .DHCPEnd , net.IP {})
51+ }
0 commit comments