synapsecns/sanguine

package inventory_test

import (
    "github.com/Flaque/filet"
    "github.com/ethereum/go-ethereum/params"
    "github.com/stretchr/testify/suite"
    "github.com/synapsecns/sanguine/core/metrics"
    "github.com/synapsecns/sanguine/core/testsuite"
    "github.com/synapsecns/sanguine/ethergo/backends"
    "github.com/synapsecns/sanguine/ethergo/backends/base"
    "github.com/synapsecns/sanguine/ethergo/backends/geth"
    "github.com/synapsecns/sanguine/ethergo/signer/wallet"
    "github.com/synapsecns/sanguine/services/omnirpc/testhelper"
    "github.com/synapsecns/sanguine/services/rfq/relayer/reldb"
    "github.com/synapsecns/sanguine/services/rfq/relayer/reldb/sqlite"
    "github.com/synapsecns/sanguine/services/rfq/testutil"
    "math/big"
    "sync"
    "testing"
)

// InventoryTestSuite is the test suite used for testing the inventory manager.
type InventoryTestSuite struct {
    *testsuite.TestSuite
    backends   map[int]backends.SimulatedTestBackend
    manager    *testutil.DeployManager
    relayer    wallet.Wallet
    omnirpcURL string
    db         reldb.Service
}

// NewInventorySuite creates the inventory suite.
func NewInventorySuite(tb testing.TB) *InventoryTestSuite {
    tb.Helper()

    return &InventoryTestSuite{
        TestSuite: testsuite.NewTestSuite(tb),
        backends:  map[int]backends.SimulatedTestBackend{},
    }
}

func TestInventorySuite(t *testing.T) {
    suite.Run(t, NewInventorySuite(t))
}

func (i *InventoryTestSuite) TearDownTest() {
    i.TestSuite.TearDownTest()

    i.backends = map[int]backends.SimulatedTestBackend{}
    i.manager = nil
    i.relayer = nil
}

func (i *InventoryTestSuite) SetupTest() {
    var err error
    i.TestSuite.SetupTest()
    i.manager = testutil.NewDeployManager(i.T())
    i.relayer, err = wallet.FromRandom()
    i.Require().NoError(err)

    // used for omnirpc client construction
    var allBackends []backends.SimulatedTestBackend
    var mux sync.Mutex
    var wg sync.WaitGroup

    wg.Add(1)
    go func() {
        defer wg.Done()
        i.db, err = sqlite.NewSqliteStore(i.GetTestContext(), filet.TmpDir(i.T(), ""), metrics.Get())
        i.NoError(err)
    }()

    for it := 1; it < 3; it++ {
        wg.Add(1)
        it := it // capture func literal
        go func() {
            defer wg.Done()
            backend := geth.NewEmbeddedBackendForChainID(i.GetTestContext(), i.T(), big.NewInt(int64(it)))
            mux.Lock()
            i.backends[it] = backend
            allBackends = append(allBackends, backend)
            mux.Unlock()

            backend.Store(base.WalletToKey(i.T(), i.relayer))
            backend.FundAccount(i.GetTestContext(), i.relayer.Address(), *new(big.Int).Mul(big.NewInt(params.Ether), big.NewInt(10)))
        }()
    }

    wg.Wait()

    i.omnirpcURL = testhelper.NewOmnirpcServer(i.GetTestContext(), i.T(), allBackends...)
}