80 lines
2.3 KiB
Go
80 lines
2.3 KiB
Go
package mailservers
|
|
|
|
import (
|
|
"fmt"
|
|
"sort"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum/p2p/enode"
|
|
"github.com/status-im/status-go/eth-node/types"
|
|
"github.com/status-im/status-go/t/utils"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestUsedConnectionPersisted(t *testing.T) {
|
|
nodes := make([]*enode.Node, 2)
|
|
fillWithRandomNodes(t, nodes)
|
|
|
|
cache := newInMemCache(t)
|
|
store := NewPeerStore(cache)
|
|
require.NoError(t, store.Update(nodes))
|
|
whisperMock := newFakeEnvelopesEvents()
|
|
monitor := NewLastUsedConnectionMonitor(store, cache, whisperMock)
|
|
monitor.Start()
|
|
|
|
// Send a confirmation that we received history from one of the peers.
|
|
select {
|
|
case whisperMock.input <- types.EnvelopeEvent{
|
|
Event: types.EventMailServerRequestCompleted, Peer: types.EnodeID(nodes[0].ID())}:
|
|
case <-time.After(time.Second):
|
|
require.FailNow(t, "can't send a 'completed' event")
|
|
}
|
|
|
|
// Wait until records will be updated in the cache.
|
|
require.NoError(t, utils.Eventually(func() error {
|
|
records, err := cache.LoadAll()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if lth := len(records); lth != 2 {
|
|
return fmt.Errorf("unexpected length of all records stored in the cache. expected %d got %d", 2, lth)
|
|
}
|
|
var used bool
|
|
for _, r := range records {
|
|
if r.Node().ID() == nodes[0].ID() {
|
|
used = !r.LastUsed.IsZero()
|
|
}
|
|
}
|
|
if !used {
|
|
return fmt.Errorf("record %s is not marked as used", types.EnodeID(nodes[0].ID()))
|
|
}
|
|
return nil
|
|
}, time.Second, 100*time.Millisecond))
|
|
|
|
// Use different peer, first will be marked as unused.
|
|
select {
|
|
case whisperMock.input <- types.EnvelopeEvent{
|
|
Event: types.EventMailServerRequestCompleted, Peer: types.EnodeID(nodes[1].ID())}:
|
|
case <-time.After(time.Second):
|
|
require.FailNow(t, "can't send a 'completed' event")
|
|
}
|
|
|
|
require.NoError(t, utils.Eventually(func() error {
|
|
records, err := cache.LoadAll()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if lth := len(records); lth != 2 {
|
|
return fmt.Errorf("unexpected length of all records stored in the cache. expected %d got %d", 2, lth)
|
|
}
|
|
sort.Slice(records, func(i, j int) bool {
|
|
return records[i].LastUsed.After(records[j].LastUsed)
|
|
})
|
|
if records[0].Node().ID() != nodes[1].ID() {
|
|
return fmt.Errorf("record wasn't updated after previous event")
|
|
}
|
|
return nil
|
|
}, time.Second, 100*time.Millisecond))
|
|
}
|