synapsecns/sanguine

package httpcapture_test

import (
    "fmt"
    "github.com/gin-gonic/gin"
    "github.com/ipfs/go-log"
    "github.com/phayes/freeport"
    "github.com/stretchr/testify/suite"
    "github.com/synapsecns/sanguine/core/ginhelper"
    "github.com/synapsecns/sanguine/core/metrics"
    "github.com/synapsecns/sanguine/core/metrics/instrumentation/httpcapture"
    baseServer "github.com/synapsecns/sanguine/core/server"
    "github.com/synapsecns/sanguine/core/testsuite"
    "net/http"
    "strings"
    "testing"
)

type HTTPCaptureSuite struct {
    *testsuite.TestSuite
    server *gin.Engine
    port   int
}

// NewCaptureSuite creates a new HTTPCaptureSuite.
func NewCaptureSuite(t *testing.T) *HTTPCaptureSuite {
    t.Helper()
    return &HTTPCaptureSuite{
        TestSuite: testsuite.NewTestSuite(t),
    }
}

func (t *HTTPCaptureSuite) SetupTest() {
    t.TestSuite.SetupTest()
    var logger = log.Logger("httpcapture-test")
    t.server = ginhelper.New(logger)
    t.server.POST("/test", func(c *gin.Context) {
        c.JSON(200, gin.H{"message": "hello world"})
    })

    var err error
    t.port, err = freeport.GetFreePort()
    t.Require().NoError(err)

    go func() {
        connection := baseServer.Server{}
        _ = connection.ListenAndServe(t.GetTestContext(), fmt.Sprintf(":%d", t.port), t.server)
    }()

    err = ginhelper.WaitForStart(t.GetTestContext(), t.port)
    t.Require().NoError(err)
}

func (t *HTTPCaptureSuite) TestCaptureTransport() {
    httpClient := new(http.Client)

    mocktracer := metrics.NewTestTracer(t.GetTestContext(), t.T())

    httpClient.Transport = httpcapture.NewCaptureTransport(httpClient.Transport, mocktracer)

    const testRequestBody = "hi"
    req, err := http.NewRequestWithContext(t.GetTestContext(), http.MethodPost, fmt.Sprintf("http://localhost:%d/test", t.port), strings.NewReader(testRequestBody))
    t.Require().NoError(err)

    res, err := httpClient.Do(req)
    defer func() {
        if res != nil && res.Body != nil {
            _ = res.Body.Close()
            _ = req.Body.Close()
        }
    }()
    t.Require().NoError(err)
    t.Require().NotNil(res)

    span := mocktracer.GetSpansByName(httpcapture.RequestSpanName)[0]

    reqBody := metrics.SpanEventByName(span, httpcapture.RequestEventName)
    t.Equal(reqBody.AsString(), testRequestBody)

    respBody := metrics.SpanEventByName(span, httpcapture.ResponseEventName)
    t.Equal(respBody.AsString(), `{"message":"hello world"}`)
}

func TestCaptureSuite(t *testing.T) {
    suite.Run(t, NewCaptureSuite(t))
}