github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-09-12 12:48:51 +00:00
Code Issues Projects Releases Wiki Activity

Refactor test for findPort in service pkg

Browse Source
This commit is contained in:
Claire Li
2014-11-05 22:38:34 -08:00
parent ea0c6bcbfd
commit c5205870c0
2 changed files with 59 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
pkg/service/endpoints_controller.go
View File

@@ -138,15 +138,15 @@ func endpointsEqual(e *api.Endpoints, endpoints []string) bool {
// findPort locates the container port for the given manifest and portName.
func findPort(manifest *api.ContainerManifest, portName util.IntOrString) (int, error) {
firstContianerPort := -1
firstContainerPort := 0
if len(manifest.Containers[0].Ports) > 0 {
firstContianerPort = manifest.Containers[0].Ports[0].ContainerPort
firstContainerPort = manifest.Containers[0].Ports[0].ContainerPort
}
switch portName.Kind {
case util.IntstrString:
if len(portName.StrVal) == 0 && firstContianerPort != -1 {
return firstContianerPort, nil
if len(portName.StrVal) == 0 && firstContainerPort != -1 {
return firstContainerPort, nil
}
name := portName.StrVal
for _, container := range manifest.Containers {
@@ -157,11 +157,11 @@ func findPort(manifest *api.ContainerManifest, portName util.IntOrString) (int,
}
}
case util.IntstrInt:
if portName.IntVal == 0 && firstContianerPort != -1 {
return firstContianerPort, nil
if portName.IntVal == 0 && firstContainerPort != -1 {
return firstContainerPort, nil
}
return portName.IntVal, nil
}
return -1, fmt.Errorf("no suitable port for manifest: %s", manifest.ID)
return 0, fmt.Errorf("no suitable port for manifest: %s", manifest.ID)
}

90
pkg/service/endpoints_controller_test.go
View File

@@ -79,45 +79,59 @@ func TestFindPort(t *testing.T) {
},
},
}
port, err := findPort(&manifest, util.IntOrString{Kind: util.IntstrString, StrVal: "foo"})
if err != nil {
t.Errorf("unexpected error: %v", err)
tests := []struct {
portName util.IntOrString
wport int
werr bool
}{
{
util.IntOrString{Kind: util.IntstrString, StrVal: "foo"},
8080,
false,
},
{
util.IntOrString{Kind: util.IntstrString, StrVal: "bar"},
8000,
false,
},
{
util.IntOrString{Kind: util.IntstrInt, IntVal: 8000},
8000,
false,
},
{
util.IntOrString{Kind: util.IntstrInt, IntVal: 7000},
7000,
false,
},
{
util.IntOrString{Kind: util.IntstrString, StrVal: "baz"},
0,
true,
},
{
util.IntOrString{Kind: util.IntstrString, StrVal: ""},
8080,
false,
},
{
util.IntOrString{},
8080,
false,
},
}
if port != 8080 {
t.Errorf("Expected 8080, Got %d", port)
}
port, err = findPort(&manifest, util.IntOrString{Kind: util.IntstrString, StrVal: "bar"})
if err != nil {
t.Errorf("unexpected error: %v", err)
}
if port != 8000 {
t.Errorf("Expected 8000, Got %d", port)
}
port, err = findPort(&manifest, util.IntOrString{Kind: util.IntstrInt, IntVal: 8000})
if port != 8000 {
t.Errorf("Expected 8000, Got %d", port)
}
port, err = findPort(&manifest, util.IntOrString{Kind: util.IntstrInt, IntVal: 7000})
if port != 7000 {
t.Errorf("Expected 7000, Got %d", port)
}
port, err = findPort(&manifest, util.IntOrString{Kind: util.IntstrString, StrVal: "baz"})
if err == nil {
t.Error("unexpected non-error")
}
port, err = findPort(&manifest, util.IntOrString{Kind: util.IntstrString, StrVal: ""})
if err != nil {
t.Errorf("unexpected error: %v", err)
}
if port != 8080 {
t.Errorf("Expected 8080, Got %d", port)
}
port, err = findPort(&manifest, util.IntOrString{})
if err != nil {
t.Errorf("unexpected error: %v", err)
}
if port != 8080 {
t.Errorf("Expected 8080, Got %d", port)
for _, test := range tests {
port, err := findPort(&manifest, test.portName)
if port != test.wport {
t.Errorf("Expected port %d, Got %d", test.wport, port)
}
if err == nil && test.werr {
t.Errorf("unexpected non-error")
}
if err != nil && test.werr == false {
t.Errorf("unexpected error: %v", err)
}
}
}