mirror of
https://github.com/scroll-tech/scroll.git
synced 2026-01-20 03:18:05 -05:00
117 lines
3.5 KiB
Go
117 lines
3.5 KiB
Go
package tests
|
|
|
|
import (
|
|
"context"
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/scroll-tech/go-ethereum/common"
|
|
gethTypes "github.com/scroll-tech/go-ethereum/core/types"
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"scroll-tech/common/database"
|
|
"scroll-tech/common/types"
|
|
|
|
"scroll-tech/rollup/internal/controller/relayer"
|
|
"scroll-tech/rollup/internal/controller/watcher"
|
|
"scroll-tech/rollup/internal/orm"
|
|
)
|
|
|
|
func testImportL1GasPrice(t *testing.T) {
|
|
db := setupDB(t)
|
|
defer database.CloseDB(db)
|
|
|
|
prepareContracts(t)
|
|
|
|
l1Cfg := rollupApp.Config.L1Config
|
|
|
|
// Create L1Relayer
|
|
l1Relayer, err := relayer.NewLayer1Relayer(context.Background(), db, l1Cfg.RelayerConfig, nil)
|
|
assert.NoError(t, err)
|
|
|
|
// Create L1Watcher
|
|
startHeight, err := l1Client.BlockNumber(context.Background())
|
|
assert.NoError(t, err)
|
|
l1Watcher := watcher.NewL1WatcherClient(context.Background(), l1Client, startHeight-1, 0, l1Cfg.L1MessageQueueAddress, l1Cfg.ScrollChainContractAddress, db, nil)
|
|
|
|
// fetch new blocks
|
|
number, err := l1Client.BlockNumber(context.Background())
|
|
assert.Greater(t, number, startHeight-1)
|
|
assert.NoError(t, err)
|
|
err = l1Watcher.FetchBlockHeader(number)
|
|
assert.NoError(t, err)
|
|
|
|
l1BlockOrm := orm.NewL1Block(db)
|
|
// check db status
|
|
latestBlockHeight, err := l1BlockOrm.GetLatestL1BlockHeight(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, number, latestBlockHeight)
|
|
blocks, err := l1BlockOrm.GetL1Blocks(context.Background(), map[string]interface{}{"number": latestBlockHeight})
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, len(blocks), 1)
|
|
assert.Empty(t, blocks[0].OracleTxHash)
|
|
assert.Equal(t, types.GasOracleStatus(blocks[0].GasOracleStatus), types.GasOraclePending)
|
|
|
|
// relay gas price
|
|
l1Relayer.ProcessGasPriceOracle()
|
|
blocks, err = l1BlockOrm.GetL1Blocks(context.Background(), map[string]interface{}{"number": latestBlockHeight})
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, len(blocks), 1)
|
|
assert.NotEmpty(t, blocks[0].OracleTxHash)
|
|
assert.Equal(t, types.GasOracleStatus(blocks[0].GasOracleStatus), types.GasOracleImporting)
|
|
}
|
|
|
|
func testImportL2GasPrice(t *testing.T) {
|
|
db := setupDB(t)
|
|
defer database.CloseDB(db)
|
|
prepareContracts(t)
|
|
|
|
l2Cfg := rollupApp.Config.L2Config
|
|
l2Relayer, err := relayer.NewLayer2Relayer(context.Background(), l2Client, db, l2Cfg.RelayerConfig, false, nil)
|
|
assert.NoError(t, err)
|
|
|
|
// add fake chunk
|
|
chunk := &types.Chunk{
|
|
Blocks: []*types.WrappedBlock{
|
|
{
|
|
Header: &gethTypes.Header{
|
|
Number: big.NewInt(1),
|
|
ParentHash: common.Hash{},
|
|
Difficulty: big.NewInt(0),
|
|
BaseFee: big.NewInt(0),
|
|
},
|
|
Transactions: nil,
|
|
WithdrawRoot: common.Hash{},
|
|
RowConsumption: &gethTypes.RowConsumption{},
|
|
},
|
|
},
|
|
}
|
|
chunkHash, err := chunk.Hash(0)
|
|
assert.NoError(t, err)
|
|
|
|
batchMeta := &types.BatchMeta{
|
|
StartChunkIndex: 0,
|
|
StartChunkHash: chunkHash.Hex(),
|
|
EndChunkIndex: 0,
|
|
EndChunkHash: chunkHash.Hex(),
|
|
}
|
|
batchOrm := orm.NewBatch(db)
|
|
_, err = batchOrm.InsertBatch(context.Background(), []*types.Chunk{chunk}, batchMeta)
|
|
assert.NoError(t, err)
|
|
|
|
// check db status
|
|
batch, err := batchOrm.GetLatestBatch(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, batch)
|
|
assert.Empty(t, batch.OracleTxHash)
|
|
assert.Equal(t, types.GasOracleStatus(batch.OracleStatus), types.GasOraclePending)
|
|
|
|
// relay gas price
|
|
l2Relayer.ProcessGasPriceOracle()
|
|
batch, err = batchOrm.GetLatestBatch(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, batch)
|
|
assert.NotEmpty(t, batch.OracleTxHash)
|
|
assert.Equal(t, types.GasOracleStatus(batch.OracleStatus), types.GasOracleImporting)
|
|
}
|