2020-07-18 18:05:04 +00:00
|
|
|
package kv
|
|
|
|
|
|
|
|
import (
|
2021-07-30 13:38:54 +00:00
|
|
|
"context"
|
2020-07-18 18:05:04 +00:00
|
|
|
"testing"
|
|
|
|
|
2023-03-17 18:52:56 +00:00
|
|
|
"github.com/prysmaticlabs/prysm/v4/encoding/bytesutil"
|
|
|
|
"github.com/prysmaticlabs/prysm/v4/testing/assert"
|
2020-07-18 18:05:04 +00:00
|
|
|
"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) {
|
2020-08-08 18:39:01 +00:00
|
|
|
err := db.Update(func(tx *bbolt.Tx) error {
|
2020-07-18 18:05:04 +00:00
|
|
|
if err := tx.Bucket(blockSlotIndicesBucket).Put([]byte("2048"), []byte("foo")); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return tx.Bucket(migrationsBucket).Put(migrationBlockSlotIndex0Key, migrationCompleted)
|
2020-08-08 18:39:01 +00:00
|
|
|
})
|
|
|
|
assert.NoError(t, err)
|
2020-07-18 18:05:04 +00:00
|
|
|
},
|
|
|
|
eval: func(t *testing.T, db *bbolt.DB) {
|
2020-08-08 18:39:01 +00:00
|
|
|
err := db.View(func(tx *bbolt.Tx) error {
|
2020-07-18 18:05:04 +00:00
|
|
|
v := tx.Bucket(blockSlotIndicesBucket).Get([]byte("2048"))
|
2021-01-22 15:15:40 +00:00
|
|
|
assert.DeepEqual(t, []byte("foo"), v, "Did not receive correct data for key 2048")
|
2020-07-18 18:05:04 +00:00
|
|
|
return nil
|
2020-08-08 18:39:01 +00:00
|
|
|
})
|
|
|
|
assert.NoError(t, err)
|
2020-07-18 18:05:04 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "migrates and deletes entries",
|
|
|
|
setup: func(t *testing.T, db *bbolt.DB) {
|
2020-08-08 18:39:01 +00:00
|
|
|
err := db.Update(func(tx *bbolt.Tx) error {
|
2020-07-18 18:05:04 +00:00
|
|
|
return tx.Bucket(blockSlotIndicesBucket).Put([]byte("2048"), []byte("foo"))
|
2020-08-08 18:39:01 +00:00
|
|
|
})
|
|
|
|
assert.NoError(t, err)
|
2020-07-18 18:05:04 +00:00
|
|
|
},
|
|
|
|
eval: func(t *testing.T, db *bbolt.DB) {
|
2020-08-08 18:39:01 +00:00
|
|
|
err := db.View(func(tx *bbolt.Tx) error {
|
2020-07-18 18:05:04 +00:00
|
|
|
k := uint64(2048)
|
2021-01-22 15:15:40 +00:00
|
|
|
v := tx.Bucket(blockSlotIndicesBucket).Get(bytesutil.Uint64ToBytesBigEndian(k))
|
|
|
|
assert.DeepEqual(t, []byte("foo"), v, "Did not receive correct data for key %d", k)
|
2020-07-18 18:05:04 +00:00
|
|
|
return nil
|
2020-08-08 18:39:01 +00:00
|
|
|
})
|
|
|
|
assert.NoError(t, err)
|
2020-07-18 18:05:04 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
db := setupDB(t).db
|
|
|
|
tt.setup(t, db)
|
2021-07-30 13:38:54 +00:00
|
|
|
assert.NoError(t, migrateBlockSlotIndex(context.Background(), db), "migrateBlockSlotIndex(tx) error")
|
2020-07-18 18:05:04 +00:00
|
|
|
tt.eval(t, db)
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|