prysm-pulse/beacon-chain/db/kv/deposit_contract_test.go
Victor Farazdagi fecbec2342
Applies assertion funcs to beacon-chain/db tests (#6938)
* part1
* go tidy
* Merge branch 'go-tidy' into apply-testutils-assertions-to-beaconchain-db
* updated beacon-chain/db/kv tests
* Merge branch 'master' into apply-testutils-assertions-to-beaconchain-db
* Merge refs/heads/master into apply-testutils-assertions-to-beaconchain-db
* Merge refs/heads/master into apply-testutils-assertions-to-beaconchain-db
2020-08-08 18:39:01 +00:00

28 lines
992 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")
}