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

Addressed reviewers comments.

k3s-v1.15.3
Abdullah Gharaibeh 2019-05-17 10:48:43 -04:00
parent 71b1565f2e
commit ef2e5bb8ed
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pkg/probe/http/http_test.go
View File

@ -347,14 +347,16 @@ func TestHTTPProbeChecker_HostHeaderPreservedAfterRedirect(t *testing.T) {
headers := http.Header{} headers := http.Header{}
headers.Add("Host", test.hostHeader) headers.Add("Host", test.hostHeader)
t.Run(desc+"local", func(t *testing.T) { t.Run(desc+"local", func(t *testing.T) {
prober := New(false) followNonLocalRedirects := false
prober := New(followNonLocalRedirects)
target, err := url.Parse(server.URL + "/redirect") target, err := url.Parse(server.URL + "/redirect")
require.NoError(t, err) require.NoError(t, err)
result, _, _ := prober.Probe(target, headers, wait.ForeverTestTimeout) result, _, _ := prober.Probe(target, headers, wait.ForeverTestTimeout)
assert.Equal(t, test.expectedResult, result) assert.Equal(t, test.expectedResult, result)
}) })
t.Run(desc+"nonlocal", func(t *testing.T) { t.Run(desc+"nonlocal", func(t *testing.T) {
prober := New(true) followNonLocalRedirects := true
prober := New(followNonLocalRedirects)
target, err := url.Parse(server.URL + "/redirect") target, err := url.Parse(server.URL + "/redirect")
require.NoError(t, err) require.NoError(t, err)
result, _, _ := prober.Probe(target, headers, wait.ForeverTestTimeout) result, _, _ := prober.Probe(target, headers, wait.ForeverTestTimeout)