synapsecns/sanguine

package node_test

import (
    "fmt"
    "math/big"
    "testing"

    "github.com/ethereum/go-ethereum/common"
    "github.com/synapsecns/sanguine/core"
    "github.com/synapsecns/sanguine/core/metrics"
    "github.com/synapsecns/sanguine/core/metrics/localmetrics"
    "github.com/synapsecns/sanguine/ethergo/backends"
    "github.com/synapsecns/sanguine/ethergo/backends/geth"
    "github.com/synapsecns/sanguine/services/explorer/contracts/bridgeconfig"
    "github.com/synapsecns/sanguine/services/explorer/metadata"
    "github.com/synapsecns/sanguine/services/explorer/testutil/testcontracts"
    scribedb "github.com/synapsecns/sanguine/services/scribe/db"
    scribeMetadata "github.com/synapsecns/sanguine/services/scribe/metadata"

    "github.com/ethereum/go-ethereum/core/types"
    "github.com/stretchr/testify/suite"
    "github.com/synapsecns/sanguine/core/testsuite"

    "github.com/synapsecns/sanguine/services/explorer/consumer/client"
    "github.com/synapsecns/sanguine/services/explorer/db"
    "github.com/synapsecns/sanguine/services/explorer/testutil"
    "go.uber.org/atomic"
)

// NodeSuite is the config test suite.
type NodeSuite struct {
    *testsuite.TestSuite
    db                db.ConsumerDB
    eventDB           scribedb.EventDB
    gqlClient         *client.Client
    logIndex          atomic.Int64
    cleanup           func()
    testBackends      map[uint32]backends.SimulatedTestBackend
    deployManager     *testutil.DeployManager
    testDeployManager *testcontracts.DeployManager
    // blockConfigChainID is the chain ID of the block config.
    blockConfigChainID uint32
    scribeMetrics      metrics.Handler
    explorerMetrics    metrics.Handler
}

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

type TestToken struct {
    tokenID string
    bridgeconfig.BridgeConfigV3Token
}

func (c *TestToken) SetTokenConfig(bridgeConfigContract *bridgeconfig.BridgeConfigRef, opts backends.AuthType) (*types.Transaction, error) {
    tx, err := bridgeConfigContract.SetTokenConfig(opts.TransactOpts, c.tokenID, c.ChainId, common.HexToAddress(c.TokenAddress),
        c.TokenDecimals, c.MaxSwap, c.MinSwap, c.SwapFee, c.MaxSwapFee, c.MinSwapFee, c.HasUnderlying, c.IsUnderlying)
    if err != nil {
        return nil, fmt.Errorf("could not set token config: %w", err)
    }
    return tx, nil
}

func (c *NodeSuite) SetupTest() {
    c.TestSuite.SetupTest()
    backends := make(map[uint32]backends.SimulatedTestBackend)
    c.blockConfigChainID = uint32(10)
    c.db, c.eventDB, c.gqlClient, c.logIndex, c.cleanup, _, c.deployManager = testutil.NewTestEnvDB(c.GetTestContext(), c.T(), c.scribeMetrics)
    backend1 := geth.NewEmbeddedBackendForChainID(c.GetTestContext(), c.T(), big.NewInt(int64(1)))
    backend2 := geth.NewEmbeddedBackendForChainID(c.GetTestContext(), c.T(), big.NewInt(int64(2)))
    backend3 := geth.NewEmbeddedBackendForChainID(c.GetTestContext(), c.T(), big.NewInt(int64(3)))
    backend4 := geth.NewEmbeddedBackendForChainID(c.GetTestContext(), c.T(), big.NewInt(int64(4))) // for the bridge config contract

    backends[uint32(1)] = backend1
    backends[uint32(2)] = backend2
    backends[uint32(3)] = backend3
    backends[c.blockConfigChainID] = backend4

    c.testBackends = backends
    c.testDeployManager = testcontracts.NewDeployManager(c.T())
}

func (c *NodeSuite) SetupSuite() {
    c.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(c.GetSuiteContext(), c.T())
        metricsHandler = metrics.Jaeger
    }

    var err error
    c.scribeMetrics, err = metrics.NewByType(c.GetSuiteContext(), scribeMetadata.BuildInfo(), metricsHandler)
    c.Require().Nil(err)
    c.explorerMetrics, err = metrics.NewByType(c.GetSuiteContext(), metadata.BuildInfo(), metricsHandler)
    c.Require().Nil(err)
}

// TestConsumerSuite runs the integration test suite.
func TestConsumerSuite(t *testing.T) {
    suite.Run(t, NewConsumerSuite(t))
}