package test

import (
	"context"
	"io"
	"net/http"
	"net/http/httptest"
)

type RequestOption func(*http.Request) *http.Request

func Request(method, target string, options ...RequestOption) *http.Request {
	request := httptest.NewRequest(method, target, nil)
	for _, option := range options {
		request = option(request)
	}
	return request
}

func RequestResponse(method, target string, options ...RequestOption) (*http.Request, *httptest.ResponseRecorder) {
	return Request(method, target, options...), httptest.NewRecorder()
}

func WithRequestHeader(key, value string) RequestOption {
	return func(r *http.Request) *http.Request {
		r.Header.Set(key, value)
		return r
	}
}

func WithRequestBody(body io.ReadCloser) RequestOption {
	return func(r *http.Request) *http.Request {
		r.Body = body
		return r
	}
}

func WithContext(ctx context.Context) RequestOption {
	return func(r *http.Request) *http.Request {
		return r.WithContext(ctx)
	}
}

func WithCookie(cookie *http.Cookie) RequestOption {
	return func(r *http.Request) *http.Request {
		r.AddCookie(cookie)
		return r
	}
}