mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-21 19:20:38 +00:00
5a66807989
* First take at updating everything to v5 * Patch gRPC gateway to use prysm v5 Fix patch * Update go ssz --------- Co-authored-by: Preston Van Loon <pvanloon@offchainlabs.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/v5/config/params"
|
|
"github.com/prysmaticlabs/prysm/v5/container/trie"
|
|
depositcontract "github.com/prysmaticlabs/prysm/v5/contracts/deposit/mock"
|
|
"github.com/prysmaticlabs/prysm/v5/runtime/interop"
|
|
"github.com/prysmaticlabs/prysm/v5/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/v5/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]
|
|
var 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]
|
|
var 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)
|
|
}
|
|
}
|