49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
package ext
|
|
|
|
import (
|
|
"github.com/status-im/status-go/eth-node/types"
|
|
)
|
|
|
|
type failureMessage struct {
|
|
IDs [][]byte
|
|
Error error
|
|
}
|
|
|
|
func NewHandlerMock(buf int) HandlerMock {
|
|
return HandlerMock{
|
|
confirmations: make(chan [][]byte, buf),
|
|
expirations: make(chan failureMessage, buf),
|
|
requestsCompleted: make(chan types.Hash, buf),
|
|
requestsExpired: make(chan types.Hash, buf),
|
|
requestsFailed: make(chan types.Hash, buf),
|
|
}
|
|
}
|
|
|
|
type HandlerMock struct {
|
|
confirmations chan [][]byte
|
|
expirations chan failureMessage
|
|
requestsCompleted chan types.Hash
|
|
requestsExpired chan types.Hash
|
|
requestsFailed chan types.Hash
|
|
}
|
|
|
|
func (t HandlerMock) EnvelopeSent(ids [][]byte) {
|
|
t.confirmations <- ids
|
|
}
|
|
|
|
func (t HandlerMock) EnvelopeExpired(ids [][]byte, err error) {
|
|
t.expirations <- failureMessage{IDs: ids, Error: err}
|
|
}
|
|
|
|
func (t HandlerMock) MailServerRequestCompleted(requestID types.Hash, lastEnvelopeHash types.Hash, cursor []byte, err error) {
|
|
if err == nil {
|
|
t.requestsCompleted <- requestID
|
|
} else {
|
|
t.requestsFailed <- requestID
|
|
}
|
|
}
|
|
|
|
func (t HandlerMock) MailServerRequestExpired(hash types.Hash) {
|
|
t.requestsExpired <- hash
|
|
}
|