102 lines
3.0 KiB
Go
102 lines
3.0 KiB
Go
|
package whisper
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/ethereum/go-ethereum/common"
|
||
|
|
"github.com/ethereum/go-ethereum/crypto"
|
||
|
|
"github.com/ethereum/go-ethereum/p2p/enode"
|
||
|
|
whisper "github.com/status-im/whisper/whisperv6"
|
||
|
|
"github.com/stretchr/testify/suite"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
testHash = common.Hash{0x01}
|
||
|
|
testIDs = [][]byte{[]byte("id")}
|
||
|
|
)
|
||
|
|
|
||
|
|
type EnvelopesMonitorSuite struct {
|
||
|
|
suite.Suite
|
||
|
|
|
||
|
|
monitor *EnvelopesMonitor
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *EnvelopesMonitorSuite) SetupTest() {
|
||
|
|
s.monitor = &EnvelopesMonitor{
|
||
|
|
envelopes: map[common.Hash]EnvelopeState{},
|
||
|
|
batches: map[common.Hash]map[common.Hash]struct{}{},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *EnvelopesMonitorSuite) TestConfirmed() {
|
||
|
|
s.monitor.Add(testIDs, testHash, whisper.NewMessage{})
|
||
|
|
s.Contains(s.monitor.envelopes, testHash)
|
||
|
|
s.Equal(EnvelopePosted, s.monitor.envelopes[testHash])
|
||
|
|
s.monitor.handleEvent(whisper.EnvelopeEvent{
|
||
|
|
Event: whisper.EventEnvelopeSent,
|
||
|
|
Hash: testHash,
|
||
|
|
})
|
||
|
|
s.Contains(s.monitor.envelopes, testHash)
|
||
|
|
s.Equal(EnvelopeSent, s.monitor.envelopes[testHash])
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *EnvelopesMonitorSuite) TestConfirmedWithAcknowledge() {
|
||
|
|
testBatch := common.Hash{1}
|
||
|
|
pkey, err := crypto.GenerateKey()
|
||
|
|
s.Require().NoError(err)
|
||
|
|
node := enode.NewV4(&pkey.PublicKey, nil, 0, 0)
|
||
|
|
s.monitor.Add(testIDs, testHash, whisper.NewMessage{})
|
||
|
|
s.Contains(s.monitor.envelopes, testHash)
|
||
|
|
s.Equal(EnvelopePosted, s.monitor.envelopes[testHash])
|
||
|
|
s.monitor.handleEvent(whisper.EnvelopeEvent{
|
||
|
|
Event: whisper.EventEnvelopeSent,
|
||
|
|
Hash: testHash,
|
||
|
|
Batch: testBatch,
|
||
|
|
})
|
||
|
|
s.Equal(EnvelopePosted, s.monitor.envelopes[testHash])
|
||
|
|
s.monitor.handleEvent(whisper.EnvelopeEvent{
|
||
|
|
Event: whisper.EventBatchAcknowledged,
|
||
|
|
Batch: testBatch,
|
||
|
|
Peer: node.ID(),
|
||
|
|
})
|
||
|
|
s.Contains(s.monitor.envelopes, testHash)
|
||
|
|
s.Equal(EnvelopeSent, s.monitor.envelopes[testHash])
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *EnvelopesMonitorSuite) TestIgnored() {
|
||
|
|
s.monitor.handleEvent(whisper.EnvelopeEvent{
|
||
|
|
Event: whisper.EventEnvelopeSent,
|
||
|
|
Hash: testHash,
|
||
|
|
})
|
||
|
|
s.NotContains(s.monitor.envelopes, testHash)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *EnvelopesMonitorSuite) TestRemoved() {
|
||
|
|
s.monitor.Add(testIDs, testHash, whisper.NewMessage{})
|
||
|
|
s.Contains(s.monitor.envelopes, testHash)
|
||
|
|
s.monitor.handleEvent(whisper.EnvelopeEvent{
|
||
|
|
Event: whisper.EventEnvelopeExpired,
|
||
|
|
Hash: testHash,
|
||
|
|
})
|
||
|
|
s.NotContains(s.monitor.envelopes, testHash)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *EnvelopesMonitorSuite) TestIgnoreNotFromMailserver() {
|
||
|
|
// enables filter in the tracker to drop confirmations from non-mailserver peers
|
||
|
|
s.monitor.mailServerConfirmation = true
|
||
|
|
s.monitor.Add(testIDs, testHash, whisper.NewMessage{})
|
||
|
|
s.monitor.handleEvent(whisper.EnvelopeEvent{
|
||
|
|
Event: whisper.EventEnvelopeSent,
|
||
|
|
Hash: testHash,
|
||
|
|
Peer: enode.ID{1}, // could be empty, doesn't impact test behaviour
|
||
|
|
})
|
||
|
|
s.Require().Equal(EnvelopePosted, s.monitor.GetState(testHash))
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *EnvelopesMonitorSuite) TestReceived() {
|
||
|
|
s.monitor.Add(testIDs, testHash, whisper.NewMessage{})
|
||
|
|
s.Contains(s.monitor.envelopes, testHash)
|
||
|
|
s.monitor.handleEvent(whisper.EnvelopeEvent{
|
||
|
|
Event: whisper.EventEnvelopeReceived,
|
||
|
|
Hash: testHash})
|
||
|
|
s.NotContains(s.monitor.envelopes, testHash)
|
||
|
|
}
|