mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 21:07:18 +00:00
97663548a1
* Add fuzz tests for sparse merkle trie and change HTR signature to return an error * fix capitalization of error message
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/config/params"
|
|
"github.com/prysmaticlabs/prysm/container/trie"
|
|
depositcontract "github.com/prysmaticlabs/prysm/contracts/deposit/mock"
|
|
"github.com/prysmaticlabs/prysm/runtime/interop"
|
|
"github.com/prysmaticlabs/prysm/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/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)
|
|
}
|
|
}
|