Skip to content

Commit

Permalink
unit tests for conv/abci (cosmos#1175)
Browse files Browse the repository at this point in the history
Closes cosmos#283
  • Loading branch information
chandiniv1 authored Sep 22, 2023
1 parent 667cec7 commit 35d90fa
Showing 1 changed file with 182 additions and 0 deletions.
182 changes: 182 additions & 0 deletions conv/abci/block_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,182 @@
package abci

import (
cmbytes "github.com/cometbft/cometbft/libs/bytes"

cmproto "github.com/cometbft/cometbft/proto/tendermint/types"

"testing"
"time"

cmversion "github.com/cometbft/cometbft/proto/tendermint/version"
cmtypes "github.com/cometbft/cometbft/types"

"github.com/rollkit/rollkit/types"

"github.com/stretchr/testify/assert"
)

func getRandomHeader() *types.Header {
return &types.Header{
BaseHeader: types.BaseHeader{
Height: 12,
Time: uint64(time.Now().Local().Day()),
ChainID: "test",
},
Version: types.Version{
Block: 1,
App: 2,
},
LastHeaderHash: types.GetRandomBytes(32),
LastCommitHash: types.GetRandomBytes(32),
DataHash: types.GetRandomBytes(32),
ConsensusHash: types.GetRandomBytes(32),
AppHash: types.GetRandomBytes(32),
LastResultsHash: types.GetRandomBytes(32),
ProposerAddress: types.GetRandomBytes(32),
AggregatorsHash: types.GetRandomBytes(32),
}
}

func getRandomBlock() *types.Block {
randomHeader := getRandomHeader()
return &types.Block{
SignedHeader: types.SignedHeader{
Header: *randomHeader,
},
Data: types.Data{
Txs: make(types.Txs, 1),
IntermediateStateRoots: types.IntermediateStateRoots{
RawRootsList: make([][]byte, 1),
},
},
}
}

func TestToABCIHeaderPB(t *testing.T) {
header := getRandomHeader()
expected := cmproto.Header{
Version: cmversion.Consensus{
Block: header.Version.Block,
App: header.Version.App,
},
Height: int64(header.Height()),
Time: header.Time(),
LastBlockId: cmproto.BlockID{
Hash: header.LastHeaderHash[:],
PartSetHeader: cmproto.PartSetHeader{
Total: 0,
Hash: nil,
},
},
LastCommitHash: header.LastHeaderHash[:],
DataHash: header.DataHash[:],
ValidatorsHash: header.AggregatorsHash[:],
NextValidatorsHash: nil,
ConsensusHash: header.ConsensusHash[:],
AppHash: header.AppHash[:],
LastResultsHash: header.LastResultsHash[:],
EvidenceHash: new(cmtypes.EvidenceData).Hash(),
ProposerAddress: header.ProposerAddress,
ChainID: header.ChainID(),
}

actual, err := ToABCIHeaderPB(header)
if err != nil {
t.Fatalf("ToABCIHeaderPB returned an error: %v", err)
}

assert.Equal(t, expected, actual)
}

func TestToABCIHeader(t *testing.T) {
header := getRandomHeader()
expected := cmtypes.Header{
Version: cmversion.Consensus{
Block: header.Version.Block,
App: header.Version.App,
},
Height: int64(header.Height()),
Time: header.Time(),
LastBlockID: cmtypes.BlockID{
Hash: cmbytes.HexBytes(header.LastHeaderHash[:]),
PartSetHeader: cmtypes.PartSetHeader{
Total: 0,
Hash: nil,
},
},
LastCommitHash: cmbytes.HexBytes(header.LastCommitHash),
DataHash: cmbytes.HexBytes(header.DataHash),
ValidatorsHash: cmbytes.HexBytes(header.AggregatorsHash),
NextValidatorsHash: nil,
ConsensusHash: cmbytes.HexBytes(header.ConsensusHash),
AppHash: cmbytes.HexBytes(header.AppHash),
LastResultsHash: cmbytes.HexBytes(header.LastResultsHash),
EvidenceHash: new(cmtypes.EvidenceData).Hash(),
ProposerAddress: header.ProposerAddress,
ChainID: header.ChainID(),
}

actual, err := ToABCIHeader(header)
if err != nil {
t.Fatalf("ToABCIHeaderPB returned an error: %v", err)
}

assert.Equal(t, expected, actual)
}

func TestToABCIBlock(t *testing.T) {
block := getRandomBlock()
abciHeader, err := ToABCIHeader(&block.SignedHeader.Header)
if err != nil {
t.Fatal(err)
}

abciCommit := block.SignedHeader.Commit.ToABCICommit(block.Height(), block.Hash())

if len(abciCommit.Signatures) == 1 {
abciCommit.Signatures[0].ValidatorAddress = block.SignedHeader.ProposerAddress
}

abciBlock := cmtypes.Block{
Header: abciHeader,
Evidence: cmtypes.EvidenceData{
Evidence: nil,
},
LastCommit: abciCommit,
}
abciBlock.Data.Txs = make([]cmtypes.Tx, len(block.Data.Txs))
for i := range block.Data.Txs {
abciBlock.Data.Txs[i] = cmtypes.Tx(block.Data.Txs[i])
}
abciBlock.Header.DataHash = cmbytes.HexBytes(block.SignedHeader.DataHash)

actual, err := ToABCIBlock(block)
if err != nil {
t.Fatalf("ToABCIBlock returned an error: %v", err)
}
assert.Equal(t, &abciBlock, actual)
}

func TestToABCIBlockMeta(t *testing.T) {
block := getRandomBlock()
cmblock, err := ToABCIBlock(block)
if err != nil {
t.Fatal(err)
}
blockID := cmtypes.BlockID{Hash: cmblock.Hash()}

expected := &cmtypes.BlockMeta{
BlockID: blockID,
BlockSize: cmblock.Size(),
Header: cmblock.Header,
NumTxs: len(cmblock.Txs),
}

actual, err := ToABCIBlockMeta(block)
if err != nil {
t.Fatalf("ToABCIBlock returned an error: %v", err)
}
assert.Equal(t, expected, actual)

}

0 comments on commit 35d90fa

Please sign in to comment.