synapsecns/sanguine

package http_test

import (
    "github.com/brianvoe/gofakeit/v6"
    "github.com/stretchr/testify/suite"
    "github.com/synapsecns/sanguine/core/testsuite"
    "github.com/synapsecns/sanguine/services/omnirpc/http"
    "testing"
)

// clientSuite defines the basic test suite.
type HTTPSuite struct {
    *testsuite.TestSuite
    clients []http.Client
}

// NewTestSuite creates a new test suite and performs some basic checks afterward.
// Every test suite in the synapse library should inherit from this suite and override where necessary.
func NewClientSuite(tb testing.TB) *HTTPSuite {
    tb.Helper()
    return &HTTPSuite{
        TestSuite: testsuite.NewTestSuite(tb),
    }
}

func (c *HTTPSuite) SetupTest() {
    c.TestSuite.SetupTest()

    for _, clientType := range http.AllClientTypes {
        c.clients = append(c.clients, http.NewClient(clientType))
    }
}

func TestCommonSuite(t *testing.T) {
    suite.Run(t, NewClientSuite(t))
}

func (c *HTTPSuite) MockHeaders(count int) (headers map[string]string) {
    headers = make(map[string]string)
    for i := 0; i < count; i++ {
        headers[gofakeit.FirstName()] = gofakeit.Sentence(10)
    }
    return headers
}