mirror of
https://github.com/scroll-tech/scroll.git
synced 2026-01-14 16:37:56 -05:00
284 lines
9.7 KiB
Go
284 lines
9.7 KiB
Go
package orm
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/scroll-tech/go-ethereum/common"
|
|
"github.com/stretchr/testify/assert"
|
|
"gorm.io/gorm"
|
|
|
|
"scroll-tech/common/docker"
|
|
"scroll-tech/common/types"
|
|
|
|
"scroll-tech/bridge/internal/config"
|
|
"scroll-tech/bridge/internal/orm/migrate"
|
|
bridgeTypes "scroll-tech/bridge/internal/types"
|
|
"scroll-tech/bridge/internal/utils"
|
|
)
|
|
|
|
var (
|
|
base *docker.App
|
|
|
|
db *gorm.DB
|
|
l2BlockOrm *L2Block
|
|
chunkOrm *Chunk
|
|
batchOrm *Batch
|
|
|
|
wrappedBlock1 *bridgeTypes.WrappedBlock
|
|
wrappedBlock2 *bridgeTypes.WrappedBlock
|
|
chunk1 *bridgeTypes.Chunk
|
|
chunk2 *bridgeTypes.Chunk
|
|
chunkHash1 common.Hash
|
|
chunkHash2 common.Hash
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
t := &testing.T{}
|
|
setupEnv(t)
|
|
defer tearDownEnv(t)
|
|
m.Run()
|
|
}
|
|
|
|
func setupEnv(t *testing.T) {
|
|
base = docker.NewDockerApp()
|
|
base.RunDBImage(t)
|
|
var err error
|
|
db, err = utils.InitDB(
|
|
&config.DBConfig{
|
|
DSN: base.DBConfig.DSN,
|
|
DriverName: base.DBConfig.DriverName,
|
|
MaxOpenNum: base.DBConfig.MaxOpenNum,
|
|
MaxIdleNum: base.DBConfig.MaxIdleNum,
|
|
},
|
|
)
|
|
assert.NoError(t, err)
|
|
sqlDB, err := db.DB()
|
|
assert.NoError(t, err)
|
|
assert.NoError(t, migrate.ResetDB(sqlDB))
|
|
|
|
batchOrm = NewBatch(db)
|
|
chunkOrm = NewChunk(db)
|
|
l2BlockOrm = NewL2Block(db)
|
|
|
|
templateBlockTrace, err := os.ReadFile("../../../common/testdata/blockTrace_02.json")
|
|
if err != nil {
|
|
t.Fatalf("failed to read file: %v", err)
|
|
}
|
|
wrappedBlock1 = &bridgeTypes.WrappedBlock{}
|
|
if err = json.Unmarshal(templateBlockTrace, wrappedBlock1); err != nil {
|
|
t.Fatalf("failed to unmarshal block trace: %v", err)
|
|
}
|
|
|
|
templateBlockTrace, err = os.ReadFile("../../../common/testdata/blockTrace_03.json")
|
|
if err != nil {
|
|
t.Fatalf("failed to read file: %v", err)
|
|
}
|
|
wrappedBlock2 = &bridgeTypes.WrappedBlock{}
|
|
if err = json.Unmarshal(templateBlockTrace, wrappedBlock2); err != nil {
|
|
t.Fatalf("failed to unmarshal block trace: %v", err)
|
|
}
|
|
|
|
chunk1 = &bridgeTypes.Chunk{Blocks: []*bridgeTypes.WrappedBlock{wrappedBlock1}}
|
|
chunkHash1, err = chunk1.Hash(0)
|
|
assert.NoError(t, err)
|
|
|
|
chunk2 = &bridgeTypes.Chunk{Blocks: []*bridgeTypes.WrappedBlock{wrappedBlock2}}
|
|
chunkHash2, err = chunk2.Hash(chunk1.NumL1Messages(0))
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func tearDownEnv(t *testing.T) {
|
|
sqlDB, err := db.DB()
|
|
assert.NoError(t, err)
|
|
sqlDB.Close()
|
|
base.Free()
|
|
}
|
|
|
|
func TestL2BlockOrm(t *testing.T) {
|
|
sqlDB, err := db.DB()
|
|
assert.NoError(t, err)
|
|
assert.NoError(t, migrate.ResetDB(sqlDB))
|
|
|
|
err = l2BlockOrm.InsertL2Blocks(context.Background(), []*bridgeTypes.WrappedBlock{wrappedBlock1, wrappedBlock2})
|
|
assert.NoError(t, err)
|
|
|
|
height, err := l2BlockOrm.GetL2BlocksLatestHeight(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, int64(3), height)
|
|
|
|
blocks, err := l2BlockOrm.GetUnchunkedBlocks(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.Len(t, blocks, 2)
|
|
assert.Equal(t, wrappedBlock1, blocks[0])
|
|
assert.Equal(t, wrappedBlock2, blocks[1])
|
|
|
|
blocks, err = l2BlockOrm.GetL2BlocksInRange(context.Background(), 2, 3)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, blocks, 2)
|
|
assert.Equal(t, wrappedBlock1, blocks[0])
|
|
assert.Equal(t, wrappedBlock2, blocks[1])
|
|
|
|
err = l2BlockOrm.UpdateChunkHashInRange(context.Background(), 2, 2, "test hash")
|
|
assert.NoError(t, err)
|
|
|
|
blocks, err = l2BlockOrm.GetUnchunkedBlocks(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.Len(t, blocks, 1)
|
|
assert.Equal(t, wrappedBlock2, blocks[0])
|
|
}
|
|
|
|
func TestChunkOrm(t *testing.T) {
|
|
sqlDB, err := db.DB()
|
|
assert.NoError(t, err)
|
|
assert.NoError(t, migrate.ResetDB(sqlDB))
|
|
|
|
err = l2BlockOrm.InsertL2Blocks(context.Background(), []*bridgeTypes.WrappedBlock{wrappedBlock1, wrappedBlock2})
|
|
assert.NoError(t, err)
|
|
|
|
dbChunk1, err := chunkOrm.InsertChunk(context.Background(), chunk1)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, dbChunk1.Hash, chunkHash1.Hex())
|
|
|
|
dbChunk2, err := chunkOrm.InsertChunk(context.Background(), chunk2)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, dbChunk2.Hash, chunkHash2.Hex())
|
|
|
|
chunks, err := chunkOrm.GetUnbatchedChunks(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.Len(t, chunks, 2)
|
|
assert.Equal(t, chunkHash1.Hex(), chunks[0].Hash)
|
|
assert.Equal(t, chunkHash2.Hex(), chunks[1].Hash)
|
|
|
|
err = chunkOrm.UpdateProvingStatus(context.Background(), chunkHash1.Hex(), types.ProvingTaskVerified)
|
|
assert.NoError(t, err)
|
|
err = chunkOrm.UpdateProvingStatus(context.Background(), chunkHash2.Hex(), types.ProvingTaskAssigned)
|
|
assert.NoError(t, err)
|
|
|
|
chunks, err = chunkOrm.GetChunksInRange(context.Background(), 0, 1)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, chunks, 2)
|
|
assert.Equal(t, chunkHash1.Hex(), chunks[0].Hash)
|
|
assert.Equal(t, chunkHash2.Hex(), chunks[1].Hash)
|
|
assert.Equal(t, types.ProvingTaskVerified, types.ProvingStatus(chunks[0].ProvingStatus))
|
|
assert.Equal(t, types.ProvingTaskAssigned, types.ProvingStatus(chunks[1].ProvingStatus))
|
|
|
|
err = chunkOrm.UpdateBatchHashInRange(context.Background(), 0, 0, "test hash")
|
|
assert.NoError(t, err)
|
|
chunks, err = chunkOrm.GetUnbatchedChunks(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.Len(t, chunks, 1)
|
|
}
|
|
|
|
func TestBatchOrm(t *testing.T) {
|
|
sqlDB, err := db.DB()
|
|
assert.NoError(t, err)
|
|
assert.NoError(t, migrate.ResetDB(sqlDB))
|
|
|
|
err = l2BlockOrm.InsertL2Blocks(context.Background(), []*bridgeTypes.WrappedBlock{wrappedBlock1, wrappedBlock2})
|
|
assert.NoError(t, err)
|
|
|
|
dbChunk1, err := chunkOrm.InsertChunk(context.Background(), chunk1)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, dbChunk1.Hash, chunkHash1.Hex())
|
|
|
|
dbChunk2, err := chunkOrm.InsertChunk(context.Background(), chunk2)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, dbChunk2.Hash, chunkHash2.Hex())
|
|
|
|
batch1, err := batchOrm.InsertBatch(context.Background(), 0, 0, chunkHash1.Hex(), chunkHash1.Hex(), []*bridgeTypes.Chunk{chunk1})
|
|
assert.NoError(t, err)
|
|
hash1 := batch1.Hash
|
|
|
|
batch1, err = batchOrm.GetBatchByIndex(context.Background(), 0)
|
|
assert.NoError(t, err)
|
|
batchHeader1, err := bridgeTypes.DecodeBatchHeader(batch1.BatchHeader)
|
|
assert.NoError(t, err)
|
|
batchHash1 := batchHeader1.Hash().Hex()
|
|
assert.Equal(t, hash1, batchHash1)
|
|
|
|
batch2, err := batchOrm.InsertBatch(context.Background(), 1, 1, chunkHash2.Hex(), chunkHash2.Hex(), []*bridgeTypes.Chunk{chunk2})
|
|
assert.NoError(t, err)
|
|
hash2 := batch2.Hash
|
|
|
|
batch2, err = batchOrm.GetBatchByIndex(context.Background(), 1)
|
|
assert.NoError(t, err)
|
|
batchHeader2, err := bridgeTypes.DecodeBatchHeader(batch2.BatchHeader)
|
|
assert.NoError(t, err)
|
|
batchHash2 := batchHeader2.Hash().Hex()
|
|
assert.Equal(t, hash2, batchHash2)
|
|
|
|
count, err := batchOrm.GetBatchCount(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, uint64(2), count)
|
|
|
|
pendingBatches, err := batchOrm.GetPendingBatches(context.Background(), 100)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 2, len(pendingBatches))
|
|
|
|
rollupStatus, err := batchOrm.GetRollupStatusByHashList(context.Background(), []string{batchHash1, batchHash2})
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 2, len(rollupStatus))
|
|
assert.Equal(t, types.RollupPending, rollupStatus[0])
|
|
assert.Equal(t, types.RollupPending, rollupStatus[1])
|
|
|
|
err = batchOrm.UpdateProvingStatus(context.Background(), batchHash1, types.ProvingTaskSkipped)
|
|
assert.NoError(t, err)
|
|
err = batchOrm.UpdateRollupStatus(context.Background(), batchHash1, types.RollupCommitted)
|
|
assert.NoError(t, err)
|
|
err = batchOrm.UpdateProvingStatus(context.Background(), batchHash2, types.ProvingTaskFailed)
|
|
assert.NoError(t, err)
|
|
err = batchOrm.UpdateRollupStatus(context.Background(), batchHash2, types.RollupCommitted)
|
|
assert.NoError(t, err)
|
|
|
|
count, err = batchOrm.UpdateSkippedBatches(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, uint64(2), count)
|
|
|
|
count, err = batchOrm.UpdateSkippedBatches(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, uint64(0), count)
|
|
|
|
batch, err := batchOrm.GetBatchByIndex(context.Background(), 1)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, types.RollupFinalizationSkipped, types.RollupStatus(batch.RollupStatus))
|
|
|
|
err = batchOrm.UpdateProvingStatus(context.Background(), batchHash2, types.ProvingTaskVerified)
|
|
assert.NoError(t, err)
|
|
|
|
dbProof, err := batchOrm.GetVerifiedProofByHash(context.Background(), batchHash1)
|
|
assert.Error(t, err, gorm.ErrRecordNotFound)
|
|
assert.Nil(t, dbProof)
|
|
|
|
err = batchOrm.UpdateProvingStatus(context.Background(), batchHash2, types.ProvingTaskVerified)
|
|
assert.NoError(t, err)
|
|
err = batchOrm.UpdateRollupStatus(context.Background(), batchHash2, types.RollupFinalized)
|
|
assert.NoError(t, err)
|
|
err = batchOrm.UpdateL2GasOracleStatusAndOracleTxHash(context.Background(), batchHash2, types.GasOracleImported, "oracleTxHash")
|
|
assert.NoError(t, err)
|
|
|
|
updatedBatch, err := batchOrm.GetLatestBatch(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, types.ProvingTaskVerified, types.ProvingStatus(updatedBatch.ProvingStatus))
|
|
assert.Equal(t, types.RollupFinalized, types.RollupStatus(updatedBatch.RollupStatus))
|
|
assert.Equal(t, types.GasOracleImported, types.GasOracleStatus(updatedBatch.OracleStatus))
|
|
assert.Equal(t, "oracleTxHash", updatedBatch.OracleTxHash)
|
|
|
|
err = batchOrm.UpdateCommitTxHashAndRollupStatus(context.Background(), batchHash2, "commitTxHash", types.RollupCommitted)
|
|
assert.NoError(t, err)
|
|
updatedBatch, err = batchOrm.GetLatestBatch(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "commitTxHash", updatedBatch.CommitTxHash)
|
|
assert.Equal(t, types.RollupCommitted, types.RollupStatus(updatedBatch.RollupStatus))
|
|
|
|
err = batchOrm.UpdateFinalizeTxHashAndRollupStatus(context.Background(), batchHash2, "finalizeTxHash", types.RollupFinalizeFailed)
|
|
assert.NoError(t, err)
|
|
|
|
updatedBatch, err = batchOrm.GetLatestBatch(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "finalizeTxHash", updatedBatch.FinalizeTxHash)
|
|
assert.Equal(t, types.RollupFinalizeFailed, types.RollupStatus(updatedBatch.RollupStatus))
|
|
}
|