mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-27 21:57:16 +00:00
1abe92fd8b
Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
28 lines
990 B
Go
28 lines
990 B
Go
package kv
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/assert"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/require"
|
|
)
|
|
|
|
func TestStore_DepositContract(t *testing.T) {
|
|
db := setupDB(t)
|
|
ctx := context.Background()
|
|
contractAddress := common.Address{1, 2, 3}
|
|
retrieved, err := db.DepositContractAddress(ctx)
|
|
require.NoError(t, err)
|
|
assert.DeepEqual(t, []uint8(nil), retrieved, "Expected nil contract address")
|
|
require.NoError(t, db.SaveDepositContractAddress(ctx, contractAddress))
|
|
retrieved, err = db.DepositContractAddress(ctx)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, contractAddress, common.BytesToAddress(retrieved), "Unexpected address")
|
|
otherAddress := common.Address{4, 5, 6}
|
|
err = db.SaveDepositContractAddress(ctx, otherAddress)
|
|
want := "cannot override deposit contract address"
|
|
assert.ErrorContains(t, want, err, "Should not have been able to override old deposit contract address")
|
|
}
|