status-im/status-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
}