synapsecns/sanguine

package http

import (
    "context"

    "github.com/synapsecns/sanguine/core/bytemap"
)

// CaptureClient is a mock client used for checking response values.
type CaptureClient struct {
    requests     []*CapturedRequest
    responseFunc MakeResponseFunc
}

// MakeResponseFunc is used for mocking responses.
type MakeResponseFunc func(c *CapturedRequest) (Response, error)

// NewCaptureClient creates  anew client for testing.
func NewCaptureClient(responseFunc MakeResponseFunc) *CaptureClient {
    return &CaptureClient{requests: []*CapturedRequest{}, responseFunc: responseFunc}
}

// Requests turns a list of sent requests. These are not mutation safe.
func (c *CaptureClient) Requests() []*CapturedRequest {
    return c.requests
}

// NewRequest creates a new request.
func (c *CaptureClient) NewRequest() Request {
    request := CapturedRequest{
        Client:        c,
        StringHeaders: make(map[string]string),
    }
    c.requests = append(c.requests, &request)
    return &request
}

// CapturedRequest stores all request data for testing.
type CapturedRequest struct {
    // ClientContains the capture client object
    Client *CaptureClient
    // Body is the request body
    Body []byte
    // Context is the request set by the client
    //nolint:containedctx
    Context context.Context
    // StringHeaders are headers set by SetHeader. Notably, this will not
    // include headers set by SetHeaderBytes
    StringHeaders map[string]string
    // StringHeaders are headers set by SetHeaderBytes. Notably, this will not
    // include headers set by SetHeader
    ByteHeaders bytemap.ByteSliceMap[[]byte]
    // RequestURI is the request uri bytes. Notably, this will not include
    // RequestURI's set by SetRequestURIBytes
    RequestURI string
    // RequestURIBytes is the request uri bytes. Notably, this will not include
    // RequestURI's set by SetRequestURI
    RequestURIBytes []byte
}

var _ Client = &CaptureClient{}

// SetBody stores the body for testing.
func (c *CapturedRequest) SetBody(body []byte) Request {
    c.Body = body
    return c
}

// SetContext stores the context for testing.
func (c *CapturedRequest) SetContext(ctx context.Context) Request {
    c.Context = ctx
    return c
}

// SetHeader sets the header for testing.
func (c *CapturedRequest) SetHeader(key, value string) Request {
    c.StringHeaders[key] = value
    return c
}

// SetHeaderBytes sets header bytes for testing.
func (c *CapturedRequest) SetHeaderBytes(key, value []byte) Request {
    c.ByteHeaders.Put(key, value)
    return c
}

// SetRequestURI stores the request uri.
func (c *CapturedRequest) SetRequestURI(uri string) Request {
    c.RequestURI = uri
    return c
}

// Do calls responseFunc for testing.
func (c *CapturedRequest) Do() (Response, error) {
    //nolint: wrapcheck
    return c.Client.responseFunc(c)
}

var _ Request = &CapturedRequest{}