synapsecns/sanguine

package db_test

import (
    "database/sql"
    "math/big"

    "github.com/brianvoe/gofakeit/v6"
    "github.com/ethereum/go-ethereum/common"
    . "github.com/stretchr/testify/assert"
    model "github.com/synapsecns/sanguine/services/explorer/db/sql"
    bridgeTypes "github.com/synapsecns/sanguine/services/explorer/types/bridge"
)

func (t *DBSuite) TestBridgeWrite() {
    defer t.cleanup()
    bridgeEvent := &model.BridgeEvent{
        InsertTime:      gofakeit.Uint64(),
        ContractAddress: common.BigToAddress(big.NewInt(gofakeit.Int64())).String(),
        ChainID:         gofakeit.Uint32(),
        EventType:       bridgeTypes.DepositEvent.Int(),
        BlockNumber:     gofakeit.Uint64(),
        TxHash:          common.BigToAddress(big.NewInt(gofakeit.Int64())).String(),

        Token:              common.BigToAddress(big.NewInt(gofakeit.Int64())).String(),
        Amount:             big.NewInt(gofakeit.Int64()),
        Recipient:          sql.NullString{String: common.BigToAddress(big.NewInt(gofakeit.Int64())).String(), Valid: true},
        DestinationChainID: big.NewInt(gofakeit.Int64()),
    }
    err := t.db.StoreEvent(t.GetTestContext(), bridgeEvent)
    Nil(t.T(), err)
}

func (t *DBSuite) TestSwapWrite() {
    defer t.cleanup()
    swapEvent := &model.SwapEvent{
        InsertTime:      gofakeit.Uint64(),
        ContractAddress: common.BigToAddress(big.NewInt(gofakeit.Int64())).String(),
        ChainID:         gofakeit.Uint32(),
        EventType:       bridgeTypes.DepositEvent.Int(),
        BlockNumber:     gofakeit.Uint64(),
        TxHash:          common.BigToAddress(big.NewInt(gofakeit.Int64())).String(),

        Buyer:        sql.NullString{String: common.BigToAddress(big.NewInt(gofakeit.Int64())).String(), Valid: true},
        TokensSold:   big.NewInt(gofakeit.Int64()),
        TokensBought: big.NewInt(gofakeit.Int64()),
        SoldID:       big.NewInt(gofakeit.Int64()),
        BoughtID:     big.NewInt(gofakeit.Int64()),
    }
    err := t.db.StoreEvent(t.GetTestContext(), swapEvent)
    Nil(t.T(), err)
}

func (t *DBSuite) TestLastBlockWrite() {
    defer t.cleanup()
    chainID := gofakeit.Uint32()
    blockNumber := gofakeit.Uint64()
    contract := common.BigToAddress(big.NewInt(gofakeit.Int64())).String()
    err := t.db.StoreLastBlock(t.GetTestContext(), chainID, blockNumber, contract)
    Nil(t.T(), err)
    blockNumber++
    err = t.db.StoreLastBlock(t.GetTestContext(), chainID, blockNumber, contract)
    Nil(t.T(), err)
    storedBlockNum, err := t.db.GetLastStoredBlock(t.GetTestContext(), chainID, contract)
    Nil(t.T(), err)
    Equal(t.T(), blockNumber, storedBlockNum)

    chainID2 := gofakeit.Uint32()
    blockNumber2 := gofakeit.Uint64()
    err = t.db.StoreLastBlock(t.GetTestContext(), chainID2, blockNumber2, contract)
    Nil(t.T(), err)
    blockNumber2--
    err = t.db.StoreLastBlock(t.GetTestContext(), chainID2, blockNumber2, contract)
    Nil(t.T(), err)
    storedBlockNum2, err := t.db.GetLastStoredBlock(t.GetTestContext(), chainID2, contract)
    Nil(t.T(), err)
    Equal(t.T(), blockNumber2+1, storedBlockNum2)

    storedBlockNumOg, err := t.db.GetLastStoredBlock(t.GetTestContext(), chainID, contract)
    Nil(t.T(), err)
    Equal(t.T(), blockNumber, storedBlockNumOg)
}