example_test.go 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123
  1. // Copyright 2015 The Go Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style
  3. // license that can be found in the LICENSE file.
  4. package httputil_test
  5. import (
  6. "fmt"
  7. "io"
  8. "log"
  9. "net/http"
  10. "net/http/httptest"
  11. "net/http/httputil"
  12. "net/url"
  13. "strings"
  14. )
  15. func ExampleDumpRequest() {
  16. ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
  17. dump, err := httputil.DumpRequest(r, true)
  18. if err != nil {
  19. http.Error(w, fmt.Sprint(err), http.StatusInternalServerError)
  20. return
  21. }
  22. fmt.Fprintf(w, "%q", dump)
  23. }))
  24. defer ts.Close()
  25. const body = "Go is a general-purpose language designed with systems programming in mind."
  26. req, err := http.NewRequest("POST", ts.URL, strings.NewReader(body))
  27. if err != nil {
  28. log.Fatal(err)
  29. }
  30. req.Host = "www.example.org"
  31. resp, err := http.DefaultClient.Do(req)
  32. if err != nil {
  33. log.Fatal(err)
  34. }
  35. defer resp.Body.Close()
  36. b, err := io.ReadAll(resp.Body)
  37. if err != nil {
  38. log.Fatal(err)
  39. }
  40. fmt.Printf("%s", b)
  41. // Output:
  42. // "POST / HTTP/1.1\r\nHost: www.example.org\r\nAccept-Encoding: gzip\r\nContent-Length: 75\r\nUser-Agent: Go-http-client/1.1\r\n\r\nGo is a general-purpose language designed with systems programming in mind."
  43. }
  44. func ExampleDumpRequestOut() {
  45. const body = "Go is a general-purpose language designed with systems programming in mind."
  46. req, err := http.NewRequest("PUT", "http://www.example.org", strings.NewReader(body))
  47. if err != nil {
  48. log.Fatal(err)
  49. }
  50. dump, err := httputil.DumpRequestOut(req, true)
  51. if err != nil {
  52. log.Fatal(err)
  53. }
  54. fmt.Printf("%q", dump)
  55. // Output:
  56. // "PUT / HTTP/1.1\r\nHost: www.example.org\r\nUser-Agent: Go-http-client/1.1\r\nContent-Length: 75\r\nAccept-Encoding: gzip\r\n\r\nGo is a general-purpose language designed with systems programming in mind."
  57. }
  58. func ExampleDumpResponse() {
  59. const body = "Go is a general-purpose language designed with systems programming in mind."
  60. ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
  61. w.Header().Set("Date", "Wed, 19 Jul 1972 19:00:00 GMT")
  62. fmt.Fprintln(w, body)
  63. }))
  64. defer ts.Close()
  65. resp, err := http.Get(ts.URL)
  66. if err != nil {
  67. log.Fatal(err)
  68. }
  69. defer resp.Body.Close()
  70. dump, err := httputil.DumpResponse(resp, true)
  71. if err != nil {
  72. log.Fatal(err)
  73. }
  74. fmt.Printf("%q", dump)
  75. // Output:
  76. // "HTTP/1.1 200 OK\r\nContent-Length: 76\r\nContent-Type: text/plain; charset=utf-8\r\nDate: Wed, 19 Jul 1972 19:00:00 GMT\r\n\r\nGo is a general-purpose language designed with systems programming in mind.\n"
  77. }
  78. func ExampleReverseProxy() {
  79. backendServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
  80. fmt.Fprintln(w, "this call was relayed by the reverse proxy")
  81. }))
  82. defer backendServer.Close()
  83. rpURL, err := url.Parse(backendServer.URL)
  84. if err != nil {
  85. log.Fatal(err)
  86. }
  87. frontendProxy := httptest.NewServer(httputil.NewSingleHostReverseProxy(rpURL))
  88. defer frontendProxy.Close()
  89. resp, err := http.Get(frontendProxy.URL)
  90. if err != nil {
  91. log.Fatal(err)
  92. }
  93. b, err := io.ReadAll(resp.Body)
  94. if err != nil {
  95. log.Fatal(err)
  96. }
  97. fmt.Printf("%s", b)
  98. // Output:
  99. // this call was relayed by the reverse proxy
  100. }