Function makeTestRoundTripper

client/client_mock_test.go:69–95

makeTestRoundTripper makes sure the response has a Body, using [http.NoBody] if none is present, and returns it as a testRoundTripper. If withDefaults is set, it also mocks the "/_ping" endpoint and sets default headers as returned by the daemon.

(f func(req *http.Request) (*http.Response, error))

// it also mocks the "/_ping" endpoint and sets default headers as returned
// by the daemon.
func makeTestRoundTripper(f func(req *http.Request) (*http.Response, error)) testRoundTripper {
	return func(req *http.Request) (*http.Response, error) {
		if req.URL.Path == "/_ping" {
			return mockPingResponse(http.StatusOK, PingResult{
				APIVersion:     MaxAPIVersion,
				OSType:         runtime.GOOS,
				Experimental:   true,
				BuilderVersion: build.BuilderBuildKit,
				SwarmStatus: &SwarmStatus{
					NodeState:        swarm.LocalNodeStateActive,
					ControlAvailable: true,
				},
			})(req)
		}
		resp, err := f(req)
		if resp != nil {
			if resp.Body == nil {
				resp.Body = http.NoBody
			}
			if resp.Request == nil {
				resp.Request = req
			}
		}
		applyDefaultHeaders(resp)
		return resp, err
	}
}

// applyDefaultHeaders mocks the headers set by the daemon's VersionMiddleware.
func applyDefaultHeaders(resp *http.Response) {