@@ -4,16 +4,12 @@ package headers
44import (
55 "fmt"
66 "net/http"
7- "net/url"
87 "strings"
98 "time"
109
1110 . "github.com/onsi/ginkgo"
1211 . "github.com/onsi/gomega"
1312
14- corev1 "k8s.io/api/core/v1"
15- metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
16-
1713 "github.com/jetstack/kube-oidc-proxy/test/e2e/framework"
1814 testutil "github.com/jetstack/kube-oidc-proxy/test/util"
1915)
@@ -28,10 +24,11 @@ var _ = framework.CasesDescribe("Headers", func() {
2824 })
2925
3026 It ("should not respond with any extra headers if none are set on the proxy" , func () {
31- fakeAPIServerURL , extraVolumes := deployFakeAPIServer (f )
27+ extraOIDCVolumes , fakeAPIServerURL , err := f .Helper ().DeployFakeAPIServer (f .Namespace .Name )
28+ Expect (err ).NotTo (HaveOccurred ())
3229
3330 By ("Redeploying proxy to send traffic to fake API server" )
34- f .DeployProxyWith (extraVolumes , fmt .Sprintf ("--server=%s" , fakeAPIServerURL ), "--certificate-authority=/fake-apiserver/ca.pem" )
31+ f .DeployProxyWith (extraOIDCVolumes , fmt .Sprintf ("--server=%s" , fakeAPIServerURL ), "--certificate-authority=/fake-apiserver/ca.pem" )
3532
3633 resp := sendRequestToProxy (f )
3734
@@ -44,10 +41,12 @@ var _ = framework.CasesDescribe("Headers", func() {
4441 })
4542
4643 It ("should respond with remote address and custom extra headers when they are set" , func () {
47- fakeAPIServerURL , extraVolumes := deployFakeAPIServer (f )
44+ By ("Deploying fake API Server" )
45+ extraOIDCVolumes , fakeAPIServerURL , err := f .Helper ().DeployFakeAPIServer (f .Namespace .Name )
46+ Expect (err ).NotTo (HaveOccurred ())
4847
4948 By ("Redeploying proxy to send traffic to fake API server with extra headers set" )
50- f .DeployProxyWith (extraVolumes , fmt .Sprintf ("--server=%s" , fakeAPIServerURL ), "--certificate-authority=/fake-apiserver/ca.pem" ,
49+ f .DeployProxyWith (extraOIDCVolumes , fmt .Sprintf ("--server=%s" , fakeAPIServerURL ), "--certificate-authority=/fake-apiserver/ca.pem" ,
5150 "--extra-user-header-client-ip" , "--extra-user-headers=key1=foo,key2=foo,key1=bar" )
5251
5352 resp := sendRequestToProxy (f )
@@ -86,36 +85,6 @@ var _ = framework.CasesDescribe("Headers", func() {
8685 })
8786})
8887
89- func deployFakeAPIServer (f * framework.Framework ) (* url.URL , []corev1.Volume ) {
90- By ("Deploying fake API Server" )
91- fAPIServerBundle , fakeAPIServerURL , err := f .Helper ().DeployFakeAPIServer (f .Namespace .Name )
92- Expect (err ).NotTo (HaveOccurred ())
93-
94- sec , err := f .KubeClientSet .CoreV1 ().Secrets (f .Namespace .Name ).Create (& corev1.Secret {
95- ObjectMeta : metav1.ObjectMeta {
96- GenerateName : "fake-apiserver-ca-" ,
97- Namespace : f .Namespace .Name ,
98- },
99- Data : map [string ][]byte {
100- "ca.pem" : fAPIServerBundle .CertBytes ,
101- },
102- })
103- Expect (err ).NotTo (HaveOccurred ())
104-
105- extraVolumes := []corev1.Volume {
106- {
107- Name : "fake-apiserver" ,
108- VolumeSource : corev1.VolumeSource {
109- Secret : & corev1.SecretVolumeSource {
110- SecretName : sec .Name ,
111- },
112- },
113- },
114- }
115-
116- return fakeAPIServerURL , extraVolumes
117- }
118-
11988func sendRequestToProxy (f * framework.Framework ) * http.Response {
12089 By ("Building request to proxy" )
12190 tokenPayload := f .Helper ().NewTokenPayload (
0 commit comments