Add unit test for docker network inspect

pull/10053/head
Yanshu Zhao 2020-12-30 02:28:44 +00:00
parent 4a940d0079
commit f044b1fa91
2 changed files with 91 additions and 2 deletions

View File

@ -171,13 +171,18 @@ type networkInspect struct {
ContainerIPs []string
}
var dockerInsepctGetter = func(name string) (*RunResult, error) {
cmd := exec.Command(Docker, "network", "inspect", name, "--format", `{"Name": "{{.Name}}","Driver": "{{.Driver}}","Subnet": "{{range .IPAM.Config}}{{.Subnet}}{{end}}","Gateway": "{{range .IPAM.Config}}{{.Gateway}}{{end}}","MTU": {{if (index .Options "com.docker.network.driver.mtu")}}{{(index .Options "com.docker.network.driver.mtu")}}{{else}}0{{end}},{{$first := true}} "ContainerIPs": [{{range $k,$v := .Containers }}{{if $first}}{{$first = false}}{{else}}, {{end}}"{{$v.IPv4Address}}"{{end}}]}`)
rr, err := runCmd(cmd)
return rr, err
}
// if exists returns subnet, gateway and mtu
func dockerNetworkInspect(name string) (netInfo, error) {
var vals networkInspect
var info = netInfo{name: name}
cmd := exec.Command(Docker, "network", "inspect", name, "--format", `{"Name": "{{.Name}}","Driver": "{{.Driver}}","Subnet": "{{range .IPAM.Config}}{{.Subnet}}{{end}}","Gateway": "{{range .IPAM.Config}}{{.Gateway}}{{end}}","MTU": {{if (index .Options "com.docker.network.driver.mtu")}}{{(index .Options "com.docker.network.driver.mtu")}}{{else}}0{{end}},{{$first := true}} "ContainerIPs": [{{range $k,$v := .Containers }}{{if $first}}{{$first = false}}{{else}}, {{end}}"{{$v.IPv4Address}}"{{end}}]}`)
rr, err := runCmd(cmd)
rr, err := dockerInsepctGetter(name)
if err != nil {
logDockerNetworkInspect(Docker, name)
if strings.Contains(rr.Output(), "No such network") {

View File

@ -0,0 +1,84 @@
/*
Copyright 2019 The Kubernetes Authors All rights reserved.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package oci
import (
"bytes"
"net"
"testing"
)
var dockerResponse string
var dockerInspectGetterMock = func(name string) (*RunResult, error) {
var responseInBytes bytes.Buffer
responseInBytes.WriteString(dockerResponse)
response := &RunResult{Stdout: responseInBytes}
return response, nil
}
func TestDockerInspectWithMTU(t *testing.T) {
dockerInspectResponseWithMtu := `{"Name": "m2","Driver": "bridge","Subnet": "172.19.0.0/16","Gateway": "172.19.0.1","MTU": 9216, "ContainerIPs": []}`
// setting up mock funcs
dockerResponse = dockerInspectResponseWithMtu
dockerInsepctGetter = dockerInspectGetterMock
netInfo, err := dockerNetworkInspect("m2")
if err != nil {
t.Errorf("Expected not to have error but got %v", err)
}
if netInfo.mtu != 9216 {
t.Errorf("Expected not to have MTU as 9216 but got %v", netInfo.mtu)
}
if !netInfo.gateway.Equal(net.ParseIP("172.19.0.1")) {
t.Errorf("Expected not to have gateway as 172.19.0.1 but got %v", netInfo.gateway)
}
if !netInfo.subnet.IP.Equal(net.ParseIP("172.19.0.0")) {
t.Errorf("Expected not to have subnet as 172.19.0.0 but got %v", netInfo.gateway)
}
}
func TestDockerInspectWithoutMTU(t *testing.T) {
dockerInspectResponseWithMtu := `{"Name": "m2","Driver": "bridge","Subnet": "172.19.0.0/16","Gateway": "172.19.0.1","MTU": 0, "ContainerIPs": []}`
// setting up mock funcs
dockerResponse = dockerInspectResponseWithMtu
dockerInsepctGetter = dockerInspectGetterMock
netInfo, err := dockerNetworkInspect("m2")
if err != nil {
t.Errorf("Expected not to have error but got %v", err)
}
if netInfo.mtu != 0 {
t.Errorf("Expected not to have MTU as 0 but got %v", netInfo.mtu)
}
if !netInfo.gateway.Equal(net.ParseIP("172.19.0.1")) {
t.Errorf("Expected not to have gateway as 172.19.0.1 but got %v", netInfo.gateway)
}
if !netInfo.subnet.IP.Equal(net.ParseIP("172.19.0.0")) {
t.Errorf("Expected not to have subnet as 172.19.0.0 but got %v", netInfo.gateway)
}
}