-
Notifications
You must be signed in to change notification settings - Fork 199
/
headersCacherStub.go
105 lines (89 loc) · 2.91 KB
/
headersCacherStub.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
105
package mock
import (
"errors"
"github.com/ElrondNetwork/elrond-go-core/data"
)
// HeadersCacherStub -
type HeadersCacherStub struct {
AddCalled func(headerHash []byte, header data.HeaderHandler)
RemoveHeaderByHashCalled func(headerHash []byte)
RemoveHeaderByNonceAndShardIdCalled func(hdrNonce uint64, shardId uint32)
GetHeaderByNonceAndShardIdCalled func(hdrNonce uint64, shardId uint32) ([]data.HeaderHandler, [][]byte, error)
GetHeaderByHashCalled func(hash []byte) (data.HeaderHandler, error)
ClearCalled func()
RegisterHandlerCalled func(handler func(header data.HeaderHandler, shardHeaderHash []byte))
NoncesCalled func(shardId uint32) []uint64
LenCalled func() int
MaxSizeCalled func() int
GetNumHeadersCalled func(shardId uint32) int
}
// AddHeader -
func (hcs *HeadersCacherStub) AddHeader(headerHash []byte, header data.HeaderHandler) {
if hcs.AddCalled != nil {
hcs.AddCalled(headerHash, header)
}
}
// RemoveHeaderByHash -
func (hcs *HeadersCacherStub) RemoveHeaderByHash(headerHash []byte) {
if hcs.RemoveHeaderByHashCalled != nil {
hcs.RemoveHeaderByHashCalled(headerHash)
}
}
// RemoveHeaderByNonceAndShardId -
func (hcs *HeadersCacherStub) RemoveHeaderByNonceAndShardId(hdrNonce uint64, shardId uint32) {
if hcs.RemoveHeaderByNonceAndShardIdCalled != nil {
hcs.RemoveHeaderByNonceAndShardIdCalled(hdrNonce, shardId)
}
}
// GetHeadersByNonceAndShardId -
func (hcs *HeadersCacherStub) GetHeadersByNonceAndShardId(hdrNonce uint64, shardId uint32) ([]data.HeaderHandler, [][]byte, error) {
if hcs.GetHeaderByNonceAndShardIdCalled != nil {
return hcs.GetHeaderByNonceAndShardIdCalled(hdrNonce, shardId)
}
return nil, nil, errors.New("err")
}
// GetHeaderByHash -
func (hcs *HeadersCacherStub) GetHeaderByHash(hash []byte) (data.HeaderHandler, error) {
if hcs.GetHeaderByHashCalled != nil {
return hcs.GetHeaderByHashCalled(hash)
}
return nil, nil
}
// Clear -
func (hcs *HeadersCacherStub) Clear() {
if hcs.ClearCalled != nil {
hcs.ClearCalled()
}
}
// RegisterHandler -
func (hcs *HeadersCacherStub) RegisterHandler(handler func(header data.HeaderHandler, shardHeaderHash []byte)) {
if hcs.RegisterHandlerCalled != nil {
hcs.RegisterHandlerCalled(handler)
}
}
// Nonces -
func (hcs *HeadersCacherStub) Nonces(shardId uint32) []uint64 {
if hcs.NoncesCalled != nil {
return hcs.NoncesCalled(shardId)
}
return nil
}
// Len -
func (hcs *HeadersCacherStub) Len() int {
return 0
}
// MaxSize -
func (hcs *HeadersCacherStub) MaxSize() int {
return 100
}
// IsInterfaceNil -
func (hcs *HeadersCacherStub) IsInterfaceNil() bool {
return hcs == nil
}
// GetNumHeaders -
func (hcs *HeadersCacherStub) GetNumHeaders(shardId uint32) int {
if hcs.GetNumHeadersCalled != nil {
return hcs.GetNumHeadersCalled(shardId)
}
return 0
}