synapsecns/sanguine

package db_test

import (
    "math/big"

    "github.com/brianvoe/gofakeit/v6"
    "github.com/ethereum/go-ethereum/common"
    "github.com/ethereum/go-ethereum/core/types"
    . "github.com/stretchr/testify/assert"
    "github.com/synapsecns/sanguine/services/scribe/db"
)

func (t *DBSuite) TestPagination() {
    t.RunOnAllDBs(func(testDB db.EventDB) {
        txHash := common.BigToHash(big.NewInt(gofakeit.Int64()))
        // Store 100 logs.
        for i := 101; i > 1; i-- {
            log := t.MakeRandomLog(txHash)
            log.BlockNumber = uint64(i)
            err := testDB.StoreLogs(t.GetTestContext(), 1, log)
            Nil(t.T(), err)
        }
        // Store another log that should be on the second page.
        log := t.MakeRandomLog(txHash)
        log.BlockNumber = 1
        err := testDB.StoreLogs(t.GetTestContext(), 1, log)
        Nil(t.T(), err)

        // Retrieve the log on the second page.
        retrievedLog, err := testDB.RetrieveLogsWithFilter(t.GetTestContext(), db.LogFilter{}, 2)
        Nil(t.T(), err)
        Equal(t.T(), 1, len(retrievedLog))

        // Check the value of the log.
        resA, err := log.MarshalJSON()
        Nil(t.T(), err)
        resB, err := retrievedLog[0].MarshalJSON()
        Nil(t.T(), err)
        Equal(t.T(), resA, resB)

        // Retrieve a receipt associated with all 101 logs.
        receipt := types.Receipt{
            Type:              gofakeit.Uint8(),
            PostState:         []byte(gofakeit.Sentence(10)),
            Status:            gofakeit.Uint64(),
            CumulativeGasUsed: gofakeit.Uint64(),
            Bloom:             types.BytesToBloom([]byte(gofakeit.Sentence(10))),
            TxHash:            txHash,
            ContractAddress:   common.BigToAddress(big.NewInt(gofakeit.Int64())),
            GasUsed:           gofakeit.Uint64(),
            BlockNumber:       big.NewInt(int64(gofakeit.Uint32())),
            TransactionIndex:  uint(gofakeit.Uint64()),
        }
        err = testDB.StoreReceipt(t.GetTestContext(), 1, receipt)
        Nil(t.T(), err)
        retrievedReceipt, err := testDB.RetrieveReceiptsWithFilter(t.GetTestContext(), db.ReceiptFilter{}, 1)
        Nil(t.T(), err)
        // Ensure the receipt has 101 logs.
        Equal(t.T(), 101, len(retrievedReceipt[0].Logs))
    })
}