mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-05 09:14:28 +00:00
b7175b3482
* Revert "Revert "Update fastssz" (#7100)"
This reverts commit b954db9704.
* Preston's patch
* Merge branch 'master' of github.com:prysmaticlabs/prysm into revert-7100-revert-6760-update-fssz
* Update fssz, add regression test case
* more HTR with fssz
* fix some tests
* only one test left
* Make it so that HTR will work
* gofmt, imports
* gofmt, imports
* fix
* Merge branch 'master' of github.com:prysmaticlabs/prysm into revert-7100-revert-6760-update-fssz
* fix
* Merge branch 'master' into revert-7100-revert-6760-update-fssz
* Merge refs/heads/master into revert-7100-revert-6760-update-fssz
* gaz
* Merge branch 'revert-7100-revert-6760-update-fssz' of github.com:prysmaticlabs/prysm into revert-7100-revert-6760-update-fssz
* Merge refs/heads/master into revert-7100-revert-6760-update-fssz
* fix test
* Merge branch 'revert-7100-revert-6760-update-fssz' of github.com:prysmaticlabs/prysm into revert-7100-revert-6760-update-fssz
* Merge refs/heads/master into revert-7100-revert-6760-update-fssz
95 lines
3.3 KiB
Go
95 lines
3.3 KiB
Go
package depositcontract_test
|
|
|
|
import (
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/ethereum/go-ethereum/accounts/abi/bind"
|
|
depositcontract "github.com/prysmaticlabs/prysm/contracts/deposit-contract"
|
|
"github.com/prysmaticlabs/prysm/shared/interop"
|
|
"github.com/prysmaticlabs/prysm/shared/params"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/assert"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/require"
|
|
"github.com/prysmaticlabs/prysm/shared/trieutil"
|
|
)
|
|
|
|
func TestDepositTrieRoot_OK(t *testing.T) {
|
|
testAcc, err := depositcontract.Setup()
|
|
require.NoError(t, err)
|
|
|
|
localTrie, err := trieutil.NewTrie(int(params.BeaconConfig().DepositContractTreeDepth))
|
|
require.NoError(t, err)
|
|
|
|
depRoot, err := testAcc.Contract.GetDepositRoot(&bind.CallOpts{})
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, depRoot, localTrie.HashTreeRoot(), "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)
|
|
|
|
localTrie.Insert(item[:], i)
|
|
depRoot, err = testAcc.Contract.GetDepositRoot(&bind.CallOpts{})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, depRoot, localTrie.HashTreeRoot(), "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 := trieutil.NewTrie(int(params.BeaconConfig().DepositContractTreeDepth))
|
|
require.NoError(t, err)
|
|
|
|
depRoot, err := testAcc.Contract.GetDepositRoot(&bind.CallOpts{})
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, depRoot, localTrie.HashTreeRoot(), "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)
|
|
|
|
localTrie.Insert(item[:], i)
|
|
|
|
depRoot, err = testAcc.Contract.GetDepositRoot(&bind.CallOpts{})
|
|
require.NoError(t, err)
|
|
|
|
assert.NotEqual(t, depRoot, localTrie.HashTreeRoot(), "Local deposit trie root and contract deposit trie root are equal for index %d", i)
|
|
}
|
|
}
|