mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-22 19:40:37 +00:00
d077483577
* v3 import renamings * tidy * fmt * rev * Update beacon-chain/core/epoch/precompute/reward_penalty_test.go * Update beacon-chain/core/helpers/validators_test.go * Update beacon-chain/db/alias.go * Update beacon-chain/db/alias.go * Update beacon-chain/db/alias.go * Update beacon-chain/db/iface/BUILD.bazel * Update beacon-chain/db/kv/kv.go * Update beacon-chain/db/kv/state.go * Update beacon-chain/rpc/prysm/v1alpha1/validator/attester_test.go * Update beacon-chain/rpc/prysm/v1alpha1/validator/attester_test.go * Update beacon-chain/sync/initial-sync/service.go * fix deps * fix bad replacements * fix bad replacements * change back * gohashtree version * fix deps Co-authored-by: Nishant Das <nishdas93@gmail.com> Co-authored-by: Potuz <potuz@prysmaticlabs.com>
103 lines
3.5 KiB
Go
103 lines
3.5 KiB
Go
package deposit_test
|
|
|
|
import (
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/ethereum/go-ethereum/accounts/abi/bind"
|
|
"github.com/prysmaticlabs/prysm/v3/config/params"
|
|
"github.com/prysmaticlabs/prysm/v3/container/trie"
|
|
depositcontract "github.com/prysmaticlabs/prysm/v3/contracts/deposit/mock"
|
|
"github.com/prysmaticlabs/prysm/v3/runtime/interop"
|
|
"github.com/prysmaticlabs/prysm/v3/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/v3/testing/require"
|
|
)
|
|
|
|
func TestDepositTrieRoot_OK(t *testing.T) {
|
|
testAcc, err := depositcontract.Setup()
|
|
require.NoError(t, err)
|
|
|
|
localTrie, err := trie.NewTrie(params.BeaconConfig().DepositContractTreeDepth)
|
|
require.NoError(t, err)
|
|
|
|
depRoot, err := testAcc.Contract.GetDepositRoot(&bind.CallOpts{})
|
|
require.NoError(t, err)
|
|
|
|
localRoot, err := localTrie.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
assert.Equal(t, depRoot, localRoot, "Local deposit trie root and contract deposit trie root are not equal")
|
|
|
|
privKeys, pubKeys, err := interop.DeterministicallyGenerateKeys(0 /*startIndex*/, 101)
|
|
require.NoError(t, err)
|
|
depositDataItems, depositDataRoots, err := interop.DepositDataFromKeys(privKeys, pubKeys)
|
|
require.NoError(t, err)
|
|
|
|
testAcc.TxOpts.Value = depositcontract.Amount32Eth()
|
|
|
|
for i := 0; i < 100; i++ {
|
|
data := depositDataItems[i]
|
|
dataRoot := [32]byte{}
|
|
copy(dataRoot[:], depositDataRoots[i])
|
|
|
|
_, err := testAcc.Contract.Deposit(testAcc.TxOpts, data.PublicKey, data.WithdrawalCredentials, data.Signature, dataRoot)
|
|
require.NoError(t, err, "Could not deposit to deposit contract")
|
|
|
|
testAcc.Backend.Commit()
|
|
item, err := data.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
|
|
assert.NoError(t, localTrie.Insert(item[:], i))
|
|
depRoot, err = testAcc.Contract.GetDepositRoot(&bind.CallOpts{})
|
|
require.NoError(t, err)
|
|
localRoot, err := localTrie.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
assert.Equal(t, depRoot, localRoot, "Local deposit trie root and contract deposit trie root are not equal for index %d", i)
|
|
}
|
|
}
|
|
|
|
func TestDepositTrieRoot_Fail(t *testing.T) {
|
|
testAcc, err := depositcontract.Setup()
|
|
require.NoError(t, err)
|
|
|
|
localTrie, err := trie.NewTrie(params.BeaconConfig().DepositContractTreeDepth)
|
|
require.NoError(t, err)
|
|
|
|
depRoot, err := testAcc.Contract.GetDepositRoot(&bind.CallOpts{})
|
|
require.NoError(t, err)
|
|
|
|
localRoot, err := localTrie.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
assert.Equal(t, depRoot, localRoot, "Local deposit trie root and contract deposit trie root are not equal")
|
|
|
|
privKeys, pubKeys, err := interop.DeterministicallyGenerateKeys(0 /*startIndex*/, 101)
|
|
require.NoError(t, err)
|
|
depositDataItems, depositDataRoots, err := interop.DepositDataFromKeys(privKeys, pubKeys)
|
|
require.NoError(t, err)
|
|
testAcc.TxOpts.Value = depositcontract.Amount32Eth()
|
|
|
|
for i := 0; i < 100; i++ {
|
|
data := depositDataItems[i]
|
|
dataRoot := [32]byte{}
|
|
copy(dataRoot[:], depositDataRoots[i])
|
|
|
|
_, err := testAcc.Contract.Deposit(testAcc.TxOpts, data.PublicKey, data.WithdrawalCredentials, data.Signature, dataRoot)
|
|
require.NoError(t, err, "Could not deposit to deposit contract")
|
|
|
|
// Change an element in the data when storing locally
|
|
copy(data.PublicKey, strconv.Itoa(i+10))
|
|
|
|
testAcc.Backend.Commit()
|
|
item, err := data.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
|
|
assert.NoError(t, localTrie.Insert(item[:], i))
|
|
|
|
depRoot, err = testAcc.Contract.GetDepositRoot(&bind.CallOpts{})
|
|
require.NoError(t, err)
|
|
|
|
localRoot, err := localTrie.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
assert.NotEqual(t, depRoot, localRoot, "Local deposit trie root and contract deposit trie root are equal for index %d", i)
|
|
}
|
|
}
|