synapsecns/sanguine

package db_test

import (
    "testing"

    "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/ethergo/backends"
    "github.com/synapsecns/sanguine/services/explorer/consumer/client"
    "github.com/synapsecns/sanguine/services/explorer/db"
    "github.com/synapsecns/sanguine/services/explorer/testutil"
    scribedb "github.com/synapsecns/sanguine/services/scribe/db"
    "github.com/synapsecns/sanguine/services/scribe/metadata"
    "go.uber.org/atomic"
)

type DBSuite struct {
    *testsuite.TestSuite
    db            db.ConsumerDB
    eventDB       scribedb.EventDB
    gqlClient     *client.Client
    logIndex      atomic.Int64
    cleanup       func()
    testBackend   backends.SimulatedTestBackend
    deployManager *testutil.DeployManager
    scribeMetrics metrics.Handler
}

// NewDBSuite creates a new ConsumerDBSuite.
func NewDBSuite(tb testing.TB) *DBSuite {
    tb.Helper()
    return &DBSuite{
        TestSuite: testsuite.NewTestSuite(tb),
        logIndex:  atomic.Int64{},
    }
}

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

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

    var err error
    t.scribeMetrics, err = metrics.NewByType(t.GetSuiteContext(), metadata.BuildInfo(), metricsHandler)
    t.Require().Nil(err)
    t.db, t.eventDB, t.gqlClient, t.logIndex, t.cleanup, t.testBackend, t.deployManager = testutil.NewTestEnvDB(t.GetTestContext(), t.T(), t.scribeMetrics)
}

// TestDBSuite tests the db suite.
func TestDBSuite(t *testing.T) {
    suite.Run(t, NewDBSuite(t))
}