-
Notifications
You must be signed in to change notification settings - Fork 199
/
sposWorkerMock.go
121 lines (102 loc) · 4.08 KB
/
sposWorkerMock.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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
package mock
import (
"github.com/ElrondNetwork/elrond-go/consensus"
"github.com/ElrondNetwork/elrond-go/core"
"github.com/ElrondNetwork/elrond-go/data"
"github.com/ElrondNetwork/elrond-go/p2p"
)
// SposWorkerMock -
type SposWorkerMock struct {
AddReceivedMessageCallCalled func(
messageType consensus.MessageType,
receivedMessageCall func(cnsDta *consensus.Message) bool,
)
AddReceivedHeaderHandlerCalled func(handler func(data.HeaderHandler))
RemoveAllReceivedMessagesCallsCalled func()
ProcessReceivedMessageCalled func(message p2p.MessageP2P) error
SendConsensusMessageCalled func(cnsDta *consensus.Message) bool
ExtendCalled func(subroundId int)
GetConsensusStateChangedChannelsCalled func() chan bool
GetBroadcastBlockCalled func(data.BodyHandler, data.HeaderHandler) error
GetBroadcastHeaderCalled func(data.HeaderHandler) error
ExecuteStoredMessagesCalled func()
DisplayStatisticsCalled func()
ReceivedHeaderCalled func(headerHandler data.HeaderHandler, headerHash []byte)
SetAppStatusHandlerCalled func(ash core.AppStatusHandler) error
ResetConsensusMessagesCalled func()
}
// AddReceivedMessageCall -
func (sposWorkerMock *SposWorkerMock) AddReceivedMessageCall(messageType consensus.MessageType,
receivedMessageCall func(cnsDta *consensus.Message) bool) {
sposWorkerMock.AddReceivedMessageCallCalled(messageType, receivedMessageCall)
}
// AddReceivedHeaderHandler -
func (sposWorkerMock *SposWorkerMock) AddReceivedHeaderHandler(handler func(data.HeaderHandler)) {
if sposWorkerMock.AddReceivedHeaderHandlerCalled != nil {
sposWorkerMock.AddReceivedHeaderHandlerCalled(handler)
}
}
// RemoveAllReceivedMessagesCalls -
func (sposWorkerMock *SposWorkerMock) RemoveAllReceivedMessagesCalls() {
sposWorkerMock.RemoveAllReceivedMessagesCallsCalled()
}
// ProcessReceivedMessage -
func (sposWorkerMock *SposWorkerMock) ProcessReceivedMessage(message p2p.MessageP2P, _ core.PeerID) error {
return sposWorkerMock.ProcessReceivedMessageCalled(message)
}
// SendConsensusMessage -
func (sposWorkerMock *SposWorkerMock) SendConsensusMessage(cnsDta *consensus.Message) bool {
return sposWorkerMock.SendConsensusMessageCalled(cnsDta)
}
// Extend -
func (sposWorkerMock *SposWorkerMock) Extend(subroundId int) {
sposWorkerMock.ExtendCalled(subroundId)
}
// GetConsensusStateChangedChannel -
func (sposWorkerMock *SposWorkerMock) GetConsensusStateChangedChannel() chan bool {
return sposWorkerMock.GetConsensusStateChangedChannelsCalled()
}
// BroadcastBlock -
func (sposWorkerMock *SposWorkerMock) BroadcastBlock(body data.BodyHandler, header data.HeaderHandler) error {
return sposWorkerMock.GetBroadcastBlockCalled(body, header)
}
// ExecuteStoredMessages -
func (sposWorkerMock *SposWorkerMock) ExecuteStoredMessages() {
sposWorkerMock.ExecuteStoredMessagesCalled()
}
// DisplayStatistics -
func (sposWorkerMock *SposWorkerMock) DisplayStatistics() {
if sposWorkerMock.DisplayStatisticsCalled != nil {
sposWorkerMock.DisplayStatisticsCalled()
}
}
// ReceivedHeader -
func (sposWorkerMock *SposWorkerMock) ReceivedHeader(headerHandler data.HeaderHandler, headerHash []byte) {
if sposWorkerMock.ReceivedHeaderCalled != nil {
sposWorkerMock.ReceivedHeaderCalled(headerHandler, headerHash)
}
}
// SetAppStatusHandler -
func (sposWorkerMock *SposWorkerMock) SetAppStatusHandler(ash core.AppStatusHandler) error {
if sposWorkerMock.SetAppStatusHandlerCalled != nil {
return sposWorkerMock.SetAppStatusHandlerCalled(ash)
}
return nil
}
// Close -
func (sposWorkerMock *SposWorkerMock) Close() error {
return nil
}
// StartWorking -
func (sposWorkerMock *SposWorkerMock) StartWorking() {
}
// ResetConsensusMessages -
func (sposWorkerMock *SposWorkerMock) ResetConsensusMessages() {
if sposWorkerMock.ResetConsensusMessagesCalled != nil {
sposWorkerMock.ResetConsensusMessagesCalled()
}
}
// IsInterfaceNil returns true if there is no value under the interface
func (sposWorkerMock *SposWorkerMock) IsInterfaceNil() bool {
return sposWorkerMock == nil
}