mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
75 lines
2.0 KiB
Go
75 lines
2.0 KiB
Go
|
package kv
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"fmt"
|
||
|
"testing"
|
||
|
|
||
|
"github.com/prysmaticlabs/prysm/shared/bytesutil"
|
||
|
"go.etcd.io/bbolt"
|
||
|
)
|
||
|
|
||
|
func Test_migrateBlockSlotIndex(t *testing.T) {
|
||
|
tests := []struct {
|
||
|
name string
|
||
|
setup func(t *testing.T, db *bbolt.DB)
|
||
|
eval func(t *testing.T, db *bbolt.DB)
|
||
|
}{
|
||
|
{
|
||
|
name: "only runs once",
|
||
|
setup: func(t *testing.T, db *bbolt.DB) {
|
||
|
if err := db.Update(func(tx *bbolt.Tx) error {
|
||
|
if err := tx.Bucket(blockSlotIndicesBucket).Put([]byte("2048"), []byte("foo")); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
return tx.Bucket(migrationsBucket).Put(migrationBlockSlotIndex0Key, migrationCompleted)
|
||
|
}); err != nil {
|
||
|
t.Error(err)
|
||
|
}
|
||
|
},
|
||
|
eval: func(t *testing.T, db *bbolt.DB) {
|
||
|
if err := db.View(func(tx *bbolt.Tx) error {
|
||
|
v := tx.Bucket(blockSlotIndicesBucket).Get([]byte("2048"))
|
||
|
if !bytes.Equal(v, []byte("foo")) {
|
||
|
return fmt.Errorf("did not receive correct data for key 2048, wanted 'foo' got %s", v)
|
||
|
}
|
||
|
return nil
|
||
|
}); err != nil {
|
||
|
t.Error(err)
|
||
|
}
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
name: "migrates and deletes entries",
|
||
|
setup: func(t *testing.T, db *bbolt.DB) {
|
||
|
if err := db.Update(func(tx *bbolt.Tx) error {
|
||
|
return tx.Bucket(blockSlotIndicesBucket).Put([]byte("2048"), []byte("foo"))
|
||
|
}); err != nil {
|
||
|
t.Error(err)
|
||
|
}
|
||
|
},
|
||
|
eval: func(t *testing.T, db *bbolt.DB) {
|
||
|
if err := db.View(func(tx *bbolt.Tx) error {
|
||
|
k := uint64(2048)
|
||
|
if v := tx.Bucket(blockSlotIndicesBucket).Get(bytesutil.Uint64ToBytesBigEndian(k)); !bytes.Equal(v, []byte("foo")) {
|
||
|
return fmt.Errorf("did not receive correct data for key %d, wanted 'foo' got %v", k, v)
|
||
|
}
|
||
|
return nil
|
||
|
}); err != nil {
|
||
|
t.Error(err)
|
||
|
}
|
||
|
},
|
||
|
},
|
||
|
}
|
||
|
for _, tt := range tests {
|
||
|
t.Run(tt.name, func(t *testing.T) {
|
||
|
db := setupDB(t).db
|
||
|
tt.setup(t, db)
|
||
|
if err := db.Update(migrateBlockSlotIndex); err != nil {
|
||
|
t.Errorf("migrateBlockSlotIndex(tx) error = %v", err)
|
||
|
}
|
||
|
tt.eval(t, db)
|
||
|
})
|
||
|
}
|
||
|
}
|