Clenup tests

This commit is contained in:
Manuel de Brito Fontes 2018-01-23 20:37:47 -03:00 committed by Manuel Alejandro de Brito Fontes
parent 12ec0475c0
commit 62622f6516
3 changed files with 18 additions and 16 deletions

View file

@ -27,12 +27,14 @@ import (
restclient "k8s.io/client-go/rest" restclient "k8s.io/client-go/rest"
"github.com/golang/glog" "github.com/golang/glog"
. "github.com/onsi/ginkgo" . "github.com/onsi/ginkgo"
. "github.com/onsi/gomega" . "github.com/onsi/gomega"
) )
const ( const (
podName = "test-ingress-controller" podName = "test-ingress-controller"
controllerPodName = "nginx-ingress-controller"
) )
const ( const (
@ -225,12 +227,15 @@ func (f *Framework) matchNginxConditions(name string, matcher func(cfg string) b
} }
var pod *v1.Pod var pod *v1.Pod
Loop:
for _, p := range l.Items { for _, p := range l.Items {
if strings.HasPrefix(p.GetName(), "nginx-ingress-controller") && if strings.HasPrefix(p.GetName(), "nginx-ingress-controller") {
len(p.Status.ContainerStatuses) > 0 && for _, cs := range p.Status.ContainerStatuses {
p.Status.ContainerStatuses[0].State.Running != nil { if cs.State.Running != nil && cs.Name == "nginx-ingress-controller" {
pod = &p pod = &p
break break Loop
}
}
} }
} }

View file

@ -21,6 +21,7 @@ import (
"io/ioutil" "io/ioutil"
"net" "net"
"strings" "strings"
"time"
. "github.com/onsi/ginkgo" . "github.com/onsi/ginkgo"
. "github.com/onsi/gomega" . "github.com/onsi/gomega"
@ -35,20 +36,21 @@ import (
var _ = framework.IngressNginxDescribe("Proxy Protocol", func() { var _ = framework.IngressNginxDescribe("Proxy Protocol", func() {
f := framework.NewDefaultFramework("proxy-protocol") f := framework.NewDefaultFramework("proxy-protocol")
setting := "use-proxy-protocol"
BeforeEach(func() { BeforeEach(func() {
err := f.NewEchoDeployment() err := f.NewEchoDeployment()
Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
}) })
AfterEach(func() { AfterEach(func() {
updateConfigmap(setting, "false", f.KubeClientSet)
}) })
It("should respect port passed by the PROXY Protocol", func() { It("should respect port passed by the PROXY Protocol", func() {
host := "proxy-protocol" host := "proxy-protocol"
setting := "use-proxy-protocol"
updateConfigmap(setting, "true", f.KubeClientSet) updateConfigmap(setting, "true", f.KubeClientSet)
defer updateConfigmap(setting, "false", f.KubeClientSet)
ing, err := f.EnsureIngress(&v1beta1.Ingress{ ing, err := f.EnsureIngress(&v1beta1.Ingress{
ObjectMeta: metav1.ObjectMeta{ ObjectMeta: metav1.ObjectMeta{
@ -120,11 +122,8 @@ func updateConfigmap(k, v string, c kubernetes.Interface) {
config.Data = map[string]string{} config.Data = map[string]string{}
} }
if config.Data[k] == v {
return
}
config.Data[k] = v config.Data[k] = v
_, err = c.CoreV1().ConfigMaps("ingress-nginx").Update(config) _, err = c.CoreV1().ConfigMaps("ingress-nginx").Update(config)
Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
time.Sleep(1 * time.Second)
} }

View file

@ -30,6 +30,7 @@ import (
var _ = framework.IngressNginxDescribe("Server Tokens", func() { var _ = framework.IngressNginxDescribe("Server Tokens", func() {
f := framework.NewDefaultFramework("server-tokens") f := framework.NewDefaultFramework("server-tokens")
serverTokens := "server-tokens"
BeforeEach(func() { BeforeEach(func() {
err := f.NewEchoDeployment() err := f.NewEchoDeployment()
@ -37,12 +38,11 @@ var _ = framework.IngressNginxDescribe("Server Tokens", func() {
}) })
AfterEach(func() { AfterEach(func() {
updateConfigmap(serverTokens, "false", f.KubeClientSet)
}) })
It("should not exists Server header in the response", func() { It("should not exists Server header in the response", func() {
serverTokens := "server-tokens"
updateConfigmap(serverTokens, "false", f.KubeClientSet) updateConfigmap(serverTokens, "false", f.KubeClientSet)
defer updateConfigmap(serverTokens, "false", f.KubeClientSet)
ing, err := f.EnsureIngress(&v1beta1.Ingress{ ing, err := f.EnsureIngress(&v1beta1.Ingress{
ObjectMeta: metav1.ObjectMeta{ ObjectMeta: metav1.ObjectMeta{
@ -84,9 +84,7 @@ var _ = framework.IngressNginxDescribe("Server Tokens", func() {
}) })
It("should exists Server header in the response when is enabled", func() { It("should exists Server header in the response when is enabled", func() {
serverTokens := "server-tokens"
updateConfigmap(serverTokens, "true", f.KubeClientSet) updateConfigmap(serverTokens, "true", f.KubeClientSet)
defer updateConfigmap(serverTokens, "false", f.KubeClientSet)
ing, err := f.EnsureIngress(&v1beta1.Ingress{ ing, err := f.EnsureIngress(&v1beta1.Ingress{
ObjectMeta: metav1.ObjectMeta{ ObjectMeta: metav1.ObjectMeta{