|
28 | 28 |
|
29 | 29 | // TestRoundTripSNI test SNI reverseproxy using example.com domain
|
30 | 30 | func (s *SNI) TestRoundTripSNI(c *check.C) {
|
31 |
| - var receivedReq *http.Request |
32 | 31 | ts := httptest.NewTLSServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
33 |
| - receivedReq = req |
| 32 | + c.Assert(req.Host, check.Equals, "example.com") |
| 33 | + c.Assert(req.Header.Get("X-My-Header"), check.Equals, "myvalue") |
| 34 | + c.Assert(req.Header.Get("X-Host"), check.Equals, "") |
34 | 35 | rw.Header().Set("X-Some-Rsp-Header", "rspvalue")
|
35 | 36 | rw.WriteHeader(200)
|
36 | 37 | rw.Write([]byte("my result"))
|
@@ -68,18 +69,12 @@ func (s *SNI) TestRoundTripSNI(c *check.C) {
|
68 | 69 | data, err := ioutil.ReadAll(rsp.Body)
|
69 | 70 | c.Assert(err, check.IsNil)
|
70 | 71 | c.Assert(string(data), check.Equals, "my result")
|
71 |
| - c.Assert(receivedReq.Host, check.Equals, "example.com") |
72 |
| - c.Assert(receivedReq.Header.Get("X-My-Header"), check.Equals, "myvalue") |
73 |
| - c.Assert(receivedReq.Header.Get("X-Host"), check.Equals, "") |
74 |
| - c.Assert(receivedReq.Header.Get("X-Forwarded-Host"), check.Equals, "") |
75 | 72 | c.Assert(router.resultHost, check.Equals, "example.com")
|
76 | 73 | }
|
77 | 74 |
|
78 | 75 | // TestRoundTripSNIWithoutHostname test SNI reverseproxy using example.com domain without SNI
|
79 | 76 | func (s *SNI) TestRoundTripSNIWithoutHostname(c *check.C) {
|
80 |
| - var receivedReq *http.Request |
81 | 77 | ts := httptest.NewTLSServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
82 |
| - receivedReq = req |
83 | 78 | rw.Header().Set("X-Some-Rsp-Header", "rspvalue")
|
84 | 79 | rw.WriteHeader(200)
|
85 | 80 | rw.Write([]byte("my result"))
|
|
0 commit comments