github
/
k3s
mirror of https://github.com/k3s-io/k3s
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Use endpoint annotation to recover container ports in e2e tests on Mesos

pull/6/head
Dr. Stefan Schimanski 2015-08-04 15:24:37 +02:00
parent c55e7bf731
commit f04f31f799
2 changed files with 27 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/e2e/kubectl.go
View File

@ -348,7 +348,7 @@ var _ = Describe("Kubectl client", func() {
endpoints, err := c.Endpoints(ns).Get(name)
Expect(err).NotTo(HaveOccurred())
uidToPort := getPortsByPodUID(endpoints.Subsets)
uidToPort := getContainerPortsByPodUID(endpoints)
if len(uidToPort) == 0 {
Logf("No endpoint found, retrying")
continue

31
test/e2e/service.go
View File

@ -22,6 +22,7 @@ import (
"math/rand"
"net/http"
"sort"
"strconv"
"strings"
"time"
@ -1023,16 +1024,36 @@ func validateUniqueOrFail(s []string) {
}
}
func getPortsByPodUID(subsets []api.EndpointSubset) PortsByPodUID {
func getContainerPortsByPodUID(endpoints *api.Endpoints) PortsByPodUID {
m := PortsByPodUID{}
for _, ss := range subsets {
for _, ss := range endpoints.Subsets {
for _, port := range ss.Ports {
for _, addr := range ss.Addresses {
Logf("Found pod %v and port %v", addr.TargetRef.UID, port.Port)
containerPort := port.Port
hostPort := port.Port
// use endpoint annotations to recover the container port in a Mesos setup
// compare contrib/mesos/pkg/service/endpoints_controller.syncService
// TODO(sttts): add ContainerPort to EndpointPort struct, defaulting to (host) Port
if providerIs("mesos/docker") {
key := fmt.Sprintf("k8s.mesosphere.io/containerPort_%s_%s_%d", port.Protocol, addr.IP, hostPort)
containerPortString := endpoints.Annotations[key]
if containerPortString == "" {
continue
}
var err error
containerPort, err = strconv.Atoi(containerPortString)
if err != nil {
continue
}
Logf("Mapped mesos host port %d to container port %d via annotation %s=%s", hostPort, containerPort, key, containerPortString)
}
Logf("Found pod %v, host port %d and container port %d", addr.TargetRef.UID, hostPort, containerPort)
if _, ok := m[addr.TargetRef.UID]; !ok {
m[addr.TargetRef.UID] = make([]int, 0)
}
m[addr.TargetRef.UID] = append(m[addr.TargetRef.UID], port.Port)
m[addr.TargetRef.UID] = append(m[addr.TargetRef.UID], containerPort)
}
}
}
@ -1089,7 +1110,7 @@ func validateEndpointsOrFail(c *client.Client, namespace, serviceName string, ex
}
Logf("Found endpoints %v", endpoints)
portsByPodUID := getPortsByPodUID(endpoints.Subsets)
portsByPodUID := getContainerPortsByPodUID(endpoints)
Logf("Found port by pod UID %v", portsByPodUID)
expectedPortsByPodUID := translatePodNameToUIDOrFail(c, namespace, expectedEndpoints)