mirror of https://github.com/k3s-io/k3s
parent
6320e41b4f
commit
55b1ae857c
|
@ -64,6 +64,9 @@ func DoHTTPProbe(url *url.URL, headers http.Header, client HTTPGetInterface) (pr
|
|||
return probe.Failure, err.Error(), nil
|
||||
}
|
||||
req.Header = headers
|
||||
if headers.Get("Host") != "" {
|
||||
req.Host = headers.Get("Host")
|
||||
}
|
||||
res, err := client.Do(req)
|
||||
if err != nil {
|
||||
// Convert errors into failures to catch timeouts.
|
||||
|
|
|
@ -85,6 +85,20 @@ func TestHTTPProbeChecker(t *testing.T) {
|
|||
"X-Muffins-Or-Cupcakes: muffins",
|
||||
},
|
||||
},
|
||||
{
|
||||
// Echo handler that returns the contents of Host in the body
|
||||
func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(200)
|
||||
w.Write([]byte(r.Host))
|
||||
},
|
||||
http.Header{
|
||||
"Host": {"muffins.cupcakes.org"},
|
||||
},
|
||||
probe.Success,
|
||||
[]string{
|
||||
"muffins.cupcakes.org",
|
||||
},
|
||||
},
|
||||
{
|
||||
handleReq(FailureCode, "fail body"),
|
||||
nil,
|
||||
|
|
Loading…
Reference in New Issue