mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-09 19:21:19 +00:00
8ebbde7836
* Add AttestationUtil receiver * Modify usage to account for the receiver * Add missing explanatory comments Co-authored-by: Radosław Kapka <rkapka@wp.pl>
59 lines
2.3 KiB
Go
59 lines
2.3 KiB
Go
package kv
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/go-bitfield"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/testing/require"
|
|
"github.com/prysmaticlabs/prysm/testing/util"
|
|
)
|
|
|
|
func TestAttCaches_hasSeenBit(t *testing.T) {
|
|
c := NewAttCaches()
|
|
|
|
au := util.AttestationUtil{}
|
|
seenA1 := au.HydrateAttestation(ðpb.Attestation{AggregationBits: bitfield.Bitlist{0b10000011}})
|
|
seenA2 := au.HydrateAttestation(ðpb.Attestation{AggregationBits: bitfield.Bitlist{0b11100000}})
|
|
require.NoError(t, c.insertSeenBit(seenA1))
|
|
require.NoError(t, c.insertSeenBit(seenA2))
|
|
tests := []struct {
|
|
att *ethpb.Attestation
|
|
want bool
|
|
}{
|
|
{att: au.HydrateAttestation(ðpb.Attestation{AggregationBits: bitfield.Bitlist{0b10000000}}), want: true},
|
|
{att: au.HydrateAttestation(ðpb.Attestation{AggregationBits: bitfield.Bitlist{0b10000001}}), want: true},
|
|
{att: au.HydrateAttestation(ðpb.Attestation{AggregationBits: bitfield.Bitlist{0b11100000}}), want: true},
|
|
{att: au.HydrateAttestation(ðpb.Attestation{AggregationBits: bitfield.Bitlist{0b10000011}}), want: true},
|
|
{att: au.HydrateAttestation(ðpb.Attestation{AggregationBits: bitfield.Bitlist{0b10001000}}), want: false},
|
|
{att: au.HydrateAttestation(ðpb.Attestation{AggregationBits: bitfield.Bitlist{0b11110111}}), want: false},
|
|
}
|
|
for _, tt := range tests {
|
|
got, err := c.hasSeenBit(tt.att)
|
|
require.NoError(t, err)
|
|
if got != tt.want {
|
|
t.Errorf("hasSeenBit() got = %v, want %v", got, tt.want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestAttCaches_insertSeenBitDuplicates(t *testing.T) {
|
|
c := NewAttCaches()
|
|
|
|
att1 := util.NewAttestationUtil().HydrateAttestation(ðpb.Attestation{AggregationBits: bitfield.Bitlist{0b10000011}})
|
|
r, err := hashFn(att1.Data)
|
|
require.NoError(t, err)
|
|
require.NoError(t, c.insertSeenBit(att1))
|
|
require.Equal(t, 1, c.seenAtt.ItemCount())
|
|
|
|
_, expirationTime1, ok := c.seenAtt.GetWithExpiration(string(r[:]))
|
|
require.Equal(t, true, ok)
|
|
|
|
// Make sure that duplicates are not inserted, but expiration time gets updated.
|
|
require.NoError(t, c.insertSeenBit(att1))
|
|
require.Equal(t, 1, c.seenAtt.ItemCount())
|
|
_, expirationprysmTime, ok := c.seenAtt.GetWithExpiration(string(r[:]))
|
|
require.Equal(t, true, ok)
|
|
require.Equal(t, true, expirationprysmTime.After(expirationTime1), "Expiration time is not updated")
|
|
}
|