mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-12 04:30:04 +00:00
d077483577
* v3 import renamings * tidy * fmt * rev * Update beacon-chain/core/epoch/precompute/reward_penalty_test.go * Update beacon-chain/core/helpers/validators_test.go * Update beacon-chain/db/alias.go * Update beacon-chain/db/alias.go * Update beacon-chain/db/alias.go * Update beacon-chain/db/iface/BUILD.bazel * Update beacon-chain/db/kv/kv.go * Update beacon-chain/db/kv/state.go * Update beacon-chain/rpc/prysm/v1alpha1/validator/attester_test.go * Update beacon-chain/rpc/prysm/v1alpha1/validator/attester_test.go * Update beacon-chain/sync/initial-sync/service.go * fix deps * fix bad replacements * fix bad replacements * change back * gohashtree version * fix deps Co-authored-by: Nishant Das <nishdas93@gmail.com> Co-authored-by: Potuz <potuz@prysmaticlabs.com>
83 lines
2.3 KiB
Go
83 lines
2.3 KiB
Go
package peerdata_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/libp2p/go-libp2p-core/peer"
|
|
"github.com/prysmaticlabs/prysm/v3/beacon-chain/p2p/peers/peerdata"
|
|
"github.com/prysmaticlabs/prysm/v3/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/v3/testing/require"
|
|
)
|
|
|
|
func TestStore_GetSetDelete(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
store := peerdata.NewStore(ctx, &peerdata.StoreConfig{
|
|
MaxPeers: 12,
|
|
})
|
|
assert.NotNil(t, store)
|
|
assert.Equal(t, 12, store.Config().MaxPeers)
|
|
|
|
// Check non-existent data.
|
|
pid := peer.ID("00001")
|
|
peerData, ok := store.PeerData(pid)
|
|
assert.Equal(t, false, ok)
|
|
assert.Equal(t, (*peerdata.PeerData)(nil), peerData)
|
|
assert.Equal(t, 0, len(store.Peers()))
|
|
|
|
// Associate data.
|
|
store.SetPeerData(pid, &peerdata.PeerData{
|
|
BadResponses: 3,
|
|
ProcessedBlocks: 42,
|
|
})
|
|
peerData, ok = store.PeerData(pid)
|
|
assert.Equal(t, true, ok)
|
|
assert.Equal(t, 3, peerData.BadResponses)
|
|
assert.Equal(t, uint64(42), peerData.ProcessedBlocks)
|
|
require.Equal(t, 1, len(store.Peers()))
|
|
peers := store.Peers()
|
|
_, ok = peers[pid]
|
|
require.Equal(t, true, ok)
|
|
assert.Equal(t, 3, peers[pid].BadResponses)
|
|
assert.Equal(t, uint64(42), peers[pid].ProcessedBlocks)
|
|
|
|
// Remove data from peer.
|
|
store.DeletePeerData(pid)
|
|
peerData, ok = store.PeerData(pid)
|
|
assert.Equal(t, false, ok)
|
|
assert.Equal(t, (*peerdata.PeerData)(nil), peerData)
|
|
assert.Equal(t, 0, len(store.Peers()))
|
|
}
|
|
|
|
func TestStore_PeerDataGetOrCreate(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
store := peerdata.NewStore(ctx, &peerdata.StoreConfig{
|
|
MaxPeers: 12,
|
|
})
|
|
assert.NotNil(t, store)
|
|
assert.Equal(t, 12, store.Config().MaxPeers)
|
|
|
|
pid := peer.ID("00001")
|
|
peerData, ok := store.PeerData(pid)
|
|
assert.Equal(t, false, ok)
|
|
assert.Equal(t, (*peerdata.PeerData)(nil), peerData)
|
|
assert.Equal(t, 0, len(store.Peers()))
|
|
|
|
peerData = store.PeerDataGetOrCreate(pid)
|
|
assert.NotNil(t, peerData)
|
|
assert.Equal(t, 0, peerData.BadResponses)
|
|
assert.Equal(t, uint64(0), peerData.ProcessedBlocks)
|
|
require.Equal(t, 1, len(store.Peers()))
|
|
|
|
// Method must be idempotent, check.
|
|
peerData = store.PeerDataGetOrCreate(pid)
|
|
assert.NotNil(t, peerData)
|
|
assert.Equal(t, 0, peerData.BadResponses)
|
|
assert.Equal(t, uint64(0), peerData.ProcessedBlocks)
|
|
require.Equal(t, 1, len(store.Peers()))
|
|
}
|