http_request_helpers.go

package integration

import (
	"bufio"
	"context"
	"encoding/json"
	"fmt"
	"io"
	"net"
	"net/http"
	"net/textproto"

	// revive:disable:dot-imports
	. "github.com/onsi/gomega"
	// revive:enable:dot-imports
)

func routerRequest(port int, path string) *http.Response {
	return doRequest(newRequest("GET", routerURL(port, path)))
}

func routerRequestWithHeaders(port int, path string, headers map[string]string) *http.Response {
	return doRequest(newRequestWithHeaders("GET", routerURL(port, path), headers))
}

func newRequest(method, url string) *http.Request {
	req, err := http.NewRequestWithContext(context.Background(), method, url, nil)
	Expect(err).NotTo(HaveOccurred())
	return req
}

func newRequestWithHeaders(method, url string, headers map[string]string) *http.Request {
	req := newRequest(method, url)
	for k, v := range headers {
		req.Header.Set(k, v)
	}
	return req
}

func doRequest(req *http.Request) *http.Response {
	if _, ok := req.Header[textproto.CanonicalMIMEHeaderKey("User-Agent")]; !ok {
		// Setting a blank User-Agent causes the http lib not to output one, whereas if there
		// is no header, it will output a default one.
		// See: https://github.com/golang/go/blob/release-branch.go1.5/src/net/http/request.go#L419
		req.Header.Set("User-Agent", "")
	}
	resp, err := http.DefaultTransport.RoundTrip(req)
	Expect(err).NotTo(HaveOccurred())
	return resp
}

func doHTTP10Request(req *http.Request) *http.Response {
	conn, err := net.Dial("tcp", req.URL.Host)
	Expect(err).NotTo(HaveOccurred())
	defer conn.Close()

	if req.Method == "" {
		req.Method = "GET"
	}
	req.Proto = "HTTP/1.0"
	req.ProtoMinor = 0
	fmt.Fprintf(conn, "%s %s %s\r\n", req.Method, req.URL.RequestURI(), req.Proto)
	err = req.Header.Write(conn)
	Expect(err).NotTo(HaveOccurred())
	fmt.Fprintf(conn, "\r\n")

	resp, err := http.ReadResponse(bufio.NewReader(conn), req)
	Expect(err).NotTo(HaveOccurred())
	return resp
}

func readBody(resp *http.Response) string {
	bytes, err := io.ReadAll(resp.Body)
	Expect(err).NotTo(HaveOccurred())
	return string(bytes)
}

func readJSONBody(resp *http.Response, data interface{}) {
	bytes, err := io.ReadAll(resp.Body)
	Expect(err).NotTo(HaveOccurred())
	err = json.Unmarshal(bytes, data)
	Expect(err).NotTo(HaveOccurred())
}