synapsecns/sanguine

package http_test

import (
    "context"
    "github.com/brianvoe/gofakeit/v6"
    . "github.com/stretchr/testify/assert"
    "github.com/synapsecns/sanguine/services/omnirpc/http"
    "github.com/synapsecns/sanguine/services/omnirpc/http/mocks"
    "testing"
)

func TestCaptureClient(t *testing.T) {
    testRes := gofakeit.ImageJpeg(50, 50)
    testBody := gofakeit.ImageJpeg(50, 50)

    client := http.NewCaptureClient(func(c *http.CapturedRequest) (http.Response, error) {
        bodyRes := new(mocks.Response)
        bodyRes.On("Body").Return(testRes)

        return bodyRes, nil
    })

    testCtx := context.Background()

    byteHeaderK := []byte(gofakeit.Word())
    byteHeaderV := []byte(gofakeit.Sentence(4))

    strHeaderK := gofakeit.Word()
    strHeaderV := gofakeit.Sentence(4)

    testURL := gofakeit.URL()

    testReq := client.NewRequest()
    testReq.SetBody(testBody)
    testReq.SetContext(testCtx)
    testReq.SetHeaderBytes(byteHeaderK, byteHeaderV)
    testReq.SetHeader(strHeaderK, strHeaderV)
    testReq.SetRequestURI(testURL)

    resp, err := testReq.Do()
    Nil(t, err)

    Equal(t, resp.Body(), testRes)

    for _, request := range client.Requests() {
        Equal(t, request.Body, testBody)
        Equal(t, request.Context, testCtx)
        Equal(t, request.StringHeaders[strHeaderK], strHeaderV)
        byteHeaderRet, _ := request.ByteHeaders.Get(byteHeaderK)

        Equal(t, byteHeaderRet, byteHeaderV)
        Equal(t, request.RequestURI, testURL)
    }
}