-
Notifications
You must be signed in to change notification settings - Fork 0
/
weighted.go
122 lines (103 loc) · 4.37 KB
/
weighted.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
package images
import (
"encoding/csv"
"fmt"
"net/http"
"strconv"
"strings"
"time"
g "github.com/onsi/ginkgo"
o "github.com/onsi/gomega"
kapi "k8s.io/kubernetes/pkg/api"
"k8s.io/kubernetes/pkg/util/wait"
e2e "k8s.io/kubernetes/test/e2e/framework"
exutil "github.com/openshift/origin/test/extended/util"
)
var _ = g.Describe("[networking][router] weighted openshift router", func() {
defer g.GinkgoRecover()
var (
configPath = exutil.FixturePath("testdata", "weighted-router.yaml")
oc = exutil.NewCLI("weighted-router", exutil.KubeConfigPath())
)
g.BeforeEach(func() {
// defer oc.Run("delete").Args("-f", configPath).Execute()
err := oc.AsAdmin().Run("adm").Args("policy", "add-cluster-role-to-user", "system:router", oc.Username()).Execute()
o.Expect(err).NotTo(o.HaveOccurred())
err = oc.Run("create").Args("-f", configPath).Execute()
o.Expect(err).NotTo(o.HaveOccurred())
})
g.Describe("The HAProxy router", func() {
g.It("should appropriately serve a route that points to two services", func() {
oc.SetOutputDir(exutil.TestContext.OutputDir)
ns := oc.KubeFramework().Namespace.Name
execPodName := exutil.CreateExecPodOrFail(oc.AdminKubeREST(), ns, "execpod")
defer func() { oc.AdminKubeREST().Pods(ns).Delete(execPodName, kapi.NewDeleteOptions(1)) }()
g.By(fmt.Sprintf("creating a weighted router from a config file %q", configPath))
var routerIP string
err := wait.Poll(time.Second, changeTimeoutSeconds*time.Second, func() (bool, error) {
pod, err := oc.KubeFramework().Client.Pods(oc.KubeFramework().Namespace.Name).Get("weighted-router")
if err != nil {
return false, err
}
if len(pod.Status.PodIP) == 0 {
return false, nil
}
routerIP = pod.Status.PodIP
return true, nil
})
o.Expect(err).NotTo(o.HaveOccurred())
// router expected to listen on port 80
routerURL := fmt.Sprintf("http://%s", routerIP)
g.By("waiting for the healthz endpoint to respond")
healthzURI := fmt.Sprintf("http://%s:1936/healthz", routerIP)
err = waitForRouterOKResponseExec(ns, execPodName, healthzURI, routerIP, changeTimeoutSeconds)
o.Expect(err).NotTo(o.HaveOccurred())
host := "weighted.example.com"
times := 100
g.By(fmt.Sprintf("checking that %d requests go through successfully", times))
// wait for the request to stabilize
err = waitForRouterOKResponseExec(ns, execPodName, routerURL, "weighted.example.com", changeTimeoutSeconds)
o.Expect(err).NotTo(o.HaveOccurred())
// all requests should now succeed
err = expectRouteStatusCodeRepeatedExec(ns, execPodName, routerURL, "weighted.example.com", http.StatusOK, times)
o.Expect(err).NotTo(o.HaveOccurred())
g.By(fmt.Sprintf("checking that there are two weighted backends in the router stats"))
var trafficValues []string
err = wait.PollImmediate(100*time.Millisecond, changeTimeoutSeconds*time.Second, func() (bool, error) {
statsURL := fmt.Sprintf("http://%s:1936/;csv", routerIP)
stats, err := getAuthenticatedRouteURLViaPod(ns, execPodName, statsURL, host, "admin", "password")
o.Expect(err).NotTo(o.HaveOccurred())
trafficValues, err = parseStats(stats, "weightedroute", 7)
o.Expect(err).NotTo(o.HaveOccurred())
return len(trafficValues) == 2, nil
})
o.Expect(err).NotTo(o.HaveOccurred())
trafficEP1, err := strconv.Atoi(trafficValues[0])
o.Expect(err).NotTo(o.HaveOccurred())
trafficEP2, err := strconv.Atoi(trafficValues[1])
o.Expect(err).NotTo(o.HaveOccurred())
weightedRatio := float32(trafficEP1) / float32(trafficEP2)
if weightedRatio < 5 && weightedRatio > 0.2 {
e2e.Failf("Unexpected weighted ratio for incoming traffic: %v (%d/%d)", weightedRatio, trafficEP1, trafficEP2)
}
g.By(fmt.Sprintf("checking that zero weights are also respected by the router"))
host = "zeroweight.example.com"
err = expectRouteStatusCodeExec(ns, execPodName, routerURL, host, http.StatusServiceUnavailable)
o.Expect(err).NotTo(o.HaveOccurred())
})
})
})
func parseStats(stats string, backendSubstr string, statsField int) ([]string, error) {
r := csv.NewReader(strings.NewReader(stats))
records, err := r.ReadAll()
if err != nil {
return nil, err
}
fieldValues := make([]string, 0)
for _, rec := range records {
if strings.Contains(rec[0], backendSubstr) && !strings.Contains(rec[1], "BACKEND") {
fieldValues = append(fieldValues, rec[statsField])
}
}
return fieldValues, nil
}