86 lines
2.2 KiB
Go
86 lines
2.2 KiB
Go
|
package shhext
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/ethereum/go-ethereum/common"
|
||
|
|
whisper "github.com/status-im/whisper/whisperv6"
|
||
|
|
"github.com/stretchr/testify/suite"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
testHash = common.Hash{0x01}
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestMailRequestMonitorSuite(t *testing.T) {
|
||
|
|
suite.Run(t, new(MailRequestMonitorSuite))
|
||
|
|
}
|
||
|
|
|
||
|
|
type MailRequestMonitorSuite struct {
|
||
|
|
suite.Suite
|
||
|
|
|
||
|
|
monitor *MailRequestMonitor
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *MailRequestMonitorSuite) SetupTest() {
|
||
|
|
s.monitor = &MailRequestMonitor{
|
||
|
|
cache: map[common.Hash]EnvelopeState{},
|
||
|
|
requestsRegistry: NewRequestsRegistry(0),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *MailRequestMonitorSuite) TestRequestCompleted() {
|
||
|
|
mock := newHandlerMock(1)
|
||
|
|
s.monitor.handler = mock
|
||
|
|
s.monitor.cache[testHash] = MailServerRequestSent
|
||
|
|
s.monitor.handleEvent(whisper.EnvelopeEvent{
|
||
|
|
Event: whisper.EventMailServerRequestCompleted,
|
||
|
|
Hash: testHash,
|
||
|
|
Data: &whisper.MailServerResponse{},
|
||
|
|
})
|
||
|
|
select {
|
||
|
|
case requestID := <-mock.requestsCompleted:
|
||
|
|
s.Equal(testHash, requestID)
|
||
|
|
s.NotContains(s.monitor.cache, testHash)
|
||
|
|
case <-time.After(10 * time.Second):
|
||
|
|
s.Fail("timed out while waiting for a request to be completed")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *MailRequestMonitorSuite) TestRequestFailed() {
|
||
|
|
mock := newHandlerMock(1)
|
||
|
|
s.monitor.handler = mock
|
||
|
|
s.monitor.cache[testHash] = MailServerRequestSent
|
||
|
|
s.monitor.handleEvent(whisper.EnvelopeEvent{
|
||
|
|
Event: whisper.EventMailServerRequestCompleted,
|
||
|
|
Hash: testHash,
|
||
|
|
Data: &whisper.MailServerResponse{Error: errors.New("test error")},
|
||
|
|
})
|
||
|
|
select {
|
||
|
|
case requestID := <-mock.requestsFailed:
|
||
|
|
s.Equal(testHash, requestID)
|
||
|
|
s.NotContains(s.monitor.cache, testHash)
|
||
|
|
case <-time.After(10 * time.Second):
|
||
|
|
s.Fail("timed out while waiting for a request to be failed")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *MailRequestMonitorSuite) TestRequestExpiration() {
|
||
|
|
mock := newHandlerMock(1)
|
||
|
|
s.monitor.handler = mock
|
||
|
|
s.monitor.cache[testHash] = MailServerRequestSent
|
||
|
|
s.monitor.handleEvent(whisper.EnvelopeEvent{
|
||
|
|
Event: whisper.EventMailServerRequestExpired,
|
||
|
|
Hash: testHash,
|
||
|
|
})
|
||
|
|
select {
|
||
|
|
case requestID := <-mock.requestsExpired:
|
||
|
|
s.Equal(testHash, requestID)
|
||
|
|
s.NotContains(s.monitor.cache, testHash)
|
||
|
|
case <-time.After(10 * time.Second):
|
||
|
|
s.Fail("timed out while waiting for request expiration")
|
||
|
|
}
|
||
|
|
}
|