-
Notifications
You must be signed in to change notification settings - Fork 199
/
blockProcessorMock.go
104 lines (80 loc) · 4.23 KB
/
blockProcessorMock.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
package mock
import (
"math/big"
"time"
"github.com/ElrondNetwork/elrond-go/data"
"github.com/ElrondNetwork/elrond-go/process/block/processedMb"
)
type BlockProcessorMock struct {
ProcessBlockCalled func(blockChain data.ChainHandler, header data.HeaderHandler, body data.BodyHandler, haveTime func() time.Duration) error
CommitBlockCalled func(blockChain data.ChainHandler, header data.HeaderHandler, body data.BodyHandler) error
RevertAccountStateCalled func()
CreateGenesisBlockCalled func(balances map[string]*big.Int) (data.HeaderHandler, error)
CreateBlockCalled func(initialHdrData data.HeaderHandler, haveTime func() bool) (data.BodyHandler, error)
RestoreBlockIntoPoolsCalled func(header data.HeaderHandler, body data.BodyHandler) error
noShards uint32
SetOnRequestTransactionCalled func(f func(destShardID uint32, txHash []byte))
ApplyBodyToHeaderCalled func(header data.HeaderHandler, body data.BodyHandler) (data.BodyHandler, error)
MarshalizedDataToBroadcastCalled func(header data.HeaderHandler, body data.BodyHandler) (map[uint32][]byte, map[string][][]byte, error)
DecodeBlockBodyCalled func(dta []byte) data.BodyHandler
DecodeBlockHeaderCalled func(dta []byte) data.HeaderHandler
AddLastNotarizedHdrCalled func(shardId uint32, processedHdr data.HeaderHandler)
CreateNewHeaderCalled func() data.HeaderHandler
RevertStateToBlockCalled func(header data.HeaderHandler) error
}
func (bpm *BlockProcessorMock) ApplyProcessedMiniBlocks(*processedMb.ProcessedMiniBlockTracker) {
}
func (bpm *BlockProcessorMock) RestoreLastNotarizedHrdsToGenesis() {
}
func (bpm *BlockProcessorMock) SetNumProcessedObj(numObj uint64) {
}
func (bpm *BlockProcessorMock) ProcessBlock(blockChain data.ChainHandler, header data.HeaderHandler, body data.BodyHandler, haveTime func() time.Duration) error {
return bpm.ProcessBlockCalled(blockChain, header, body, haveTime)
}
func (bpm *BlockProcessorMock) CommitBlock(blockChain data.ChainHandler, header data.HeaderHandler, body data.BodyHandler) error {
return bpm.CommitBlockCalled(blockChain, header, body)
}
func (bpm *BlockProcessorMock) RevertAccountState() {
bpm.RevertAccountStateCalled()
}
func (bpm *BlockProcessorMock) CreateNewHeader() data.HeaderHandler {
return bpm.CreateNewHeaderCalled()
}
func (bpm *BlockProcessorMock) CreateGenesisBlock(balances map[string]*big.Int) (data.HeaderHandler, error) {
return bpm.CreateGenesisBlockCalled(balances)
}
func (bpm *BlockProcessorMock) CreateBlockBody(initialHdrData data.HeaderHandler, haveTime func() bool) (data.BodyHandler, error) {
return bpm.CreateBlockCalled(initialHdrData, haveTime)
}
func (bpm *BlockProcessorMock) RestoreBlockIntoPools(header data.HeaderHandler, body data.BodyHandler) error {
return bpm.RestoreBlockIntoPoolsCalled(header, body)
}
func (bpm *BlockProcessorMock) ApplyBodyToHeader(header data.HeaderHandler, body data.BodyHandler) (data.BodyHandler, error) {
return bpm.ApplyBodyToHeaderCalled(header, body)
}
func (bpm *BlockProcessorMock) MarshalizedDataToBroadcast(header data.HeaderHandler, body data.BodyHandler) (map[uint32][]byte, map[string][][]byte, error) {
return bpm.MarshalizedDataToBroadcastCalled(header, body)
}
func (bpm *BlockProcessorMock) DecodeBlockBody(dta []byte) data.BodyHandler {
return bpm.DecodeBlockBodyCalled(dta)
}
func (bpm *BlockProcessorMock) DecodeBlockHeader(dta []byte) data.HeaderHandler {
return bpm.DecodeBlockHeaderCalled(dta)
}
func (bpm *BlockProcessorMock) AddLastNotarizedHdr(shardId uint32, processedHdr data.HeaderHandler) {
bpm.AddLastNotarizedHdrCalled(shardId, processedHdr)
}
func (bpm *BlockProcessorMock) SetConsensusData(randomness []byte, round uint64, epoch uint32, shardId uint32) {
panic("implement me")
}
// RevertStateToBlock recreates thee state tries to the root hashes indicated by the provided header
func (bpm *BlockProcessorMock) RevertStateToBlock(header data.HeaderHandler) error {
if bpm.RevertStateToBlockCalled != nil {
return bpm.RevertStateToBlockCalled(header)
}
return nil
}
// IsInterfaceNil returns true if there is no value under the interface
func (bpm *BlockProcessorMock) IsInterfaceNil() bool {
return bpm == nil
}