synapsecns/sanguine

package proxy_test

import (
    "testing"

    "github.com/stretchr/testify/assert"
    "github.com/stretchr/testify/suite"
    "github.com/synapsecns/sanguine/core"
    "github.com/synapsecns/sanguine/core/metrics"
    "github.com/synapsecns/sanguine/core/metrics/localmetrics"
    "github.com/synapsecns/sanguine/core/testsuite"
    "github.com/synapsecns/sanguine/services/omnirpc/metadata"
)

type ProxySuite struct {
    *testsuite.TestSuite
    metrics metrics.Handler
}

// NewProxySuite creates a end-to-end test suite.
func NewProxySuite(tb testing.TB) *ProxySuite {
    tb.Helper()
    return &ProxySuite{
        TestSuite: testsuite.NewTestSuite(tb),
    }
}

func (p *ProxySuite) SetupSuite() {
    p.TestSuite.SetupSuite()

    // don't use metrics on ci for integration tests
    isCI := core.GetEnvBool("CI", false)
    useMetrics := !isCI
    metricsHandler := metrics.Null

    if useMetrics {
        localmetrics.SetupTestJaeger(p.GetSuiteContext(), p.T())
        metricsHandler = metrics.Jaeger
    }

    var err error
    p.metrics, err = metrics.NewByType(p.GetSuiteContext(), metadata.BuildInfo(), metricsHandler)
    assert.Nil(p.T(), err)
}

func TestProxySuite(t *testing.T) {
    suite.Run(t, NewProxySuite(t))
}