use MustPassRepeatedly

This commit is contained in:
Chotiwat Chawannakul 2024-02-16 02:24:33 -08:00
parent 0f48040783
commit bde121a429

View file

@ -220,56 +220,48 @@ var _ = framework.IngressNginxDescribe("[Flag] enable-ssl-passthrough", func() {
} }
tries := 3 tries := 3
ginkgo.It("should handle known traffic without Host header", func() { ginkgo.It("should handle known traffic without Host header", ginkgo.MustPassRepeatedly(tries), func() {
for i := 0; i < tries; i++ { f.HTTPTestClientWithTLSConfig(tlsConfig).
f.HTTPTestClientWithTLSConfig(tlsConfig). GET("/").
GET("/"). WithURL(url).
WithURL(url). ForceResolve(f.GetNginxIP(), 443).
ForceResolve(f.GetNginxIP(), 443). WithDialContextMiddleware(throttleMiddleware).
WithDialContextMiddleware(throttleMiddleware). Expect().
Expect(). Status(http.StatusOK)
Status(http.StatusOK)
}
}) })
ginkgo.It("should handle insecure traffic without Host header", func() { ginkgo.It("should handle insecure traffic without Host header", ginkgo.MustPassRepeatedly(tries), func() {
for i := 0; i < tries; i++ { //nolint:gosec // Ignore the gosec error in testing
//nolint:gosec // Ignore the gosec error in testing f.HTTPTestClientWithTLSConfig(&tls.Config{ServerName: host, InsecureSkipVerify: true}).
f.HTTPTestClientWithTLSConfig(&tls.Config{ServerName: host, InsecureSkipVerify: true}). GET("/").
GET("/"). WithURL(url).
WithURL(url). ForceResolve(f.GetNginxIP(), 443).
ForceResolve(f.GetNginxIP(), 443). WithDialContextMiddleware(throttleMiddleware).
WithDialContextMiddleware(throttleMiddleware). Expect().
Expect(). Status(http.StatusOK)
Status(http.StatusOK)
}
}) })
ginkgo.It("should handle known traffic with Host header", func() { ginkgo.It("should handle known traffic with Host header", ginkgo.MustPassRepeatedly(tries), func() {
for i := 0; i < tries; i++ { f.HTTPTestClientWithTLSConfig(tlsConfig).
f.HTTPTestClientWithTLSConfig(tlsConfig). GET("/").
GET("/"). WithURL(url).
WithURL(url). WithHeader("Host", host).
WithHeader("Host", host). ForceResolve(f.GetNginxIP(), 443).
ForceResolve(f.GetNginxIP(), 443). WithDialContextMiddleware(throttleMiddleware).
WithDialContextMiddleware(throttleMiddleware). Expect().
Expect(). Status(http.StatusOK)
Status(http.StatusOK)
}
}) })
ginkgo.It("should handle insecure traffic with Host header", func() { ginkgo.It("should handle insecure traffic with Host header", ginkgo.MustPassRepeatedly(tries), func() {
for i := 0; i < tries; i++ { //nolint:gosec // Ignore the gosec error in testing
//nolint:gosec // Ignore the gosec error in testing f.HTTPTestClientWithTLSConfig(&tls.Config{ServerName: host, InsecureSkipVerify: true}).
f.HTTPTestClientWithTLSConfig(&tls.Config{ServerName: host, InsecureSkipVerify: true}). GET("/").
GET("/"). WithURL(url).
WithURL(url). WithHeader("Host", host).
WithHeader("Host", host). ForceResolve(f.GetNginxIP(), 443).
ForceResolve(f.GetNginxIP(), 443). WithDialContextMiddleware(throttleMiddleware).
WithDialContextMiddleware(throttleMiddleware). Expect().
Expect(). Status(http.StatusOK)
Status(http.StatusOK)
}
}) })
}) })
}) })