mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
81c53c26fb
* Update seed domains (#3872) * Remove Transfers (#3870) * Remove active index roots and compact committee roots (#3869) * Update inclusion reward (#3886) * Alter proposer selection logic (#3884) * Fix early committee bias (#3888) * Remove shards and committees (#3896) * Epoch spec tests v0.9 (#3907) * Block spec test v0.9 (#3905) * rm'ed in protobuf * build proto * build proto * build proto * fix core package * Gazelle * Fixed all the tests * Fixed static test * Comment out spec test for now * One more skip * fix-roundRobinSync (#3862) * Starting but need new seed function * Revert initial sync * Updated Proposer Slashing * Fixed all tests * Lint * Update inclusion reward * Fill randao mixes with eth1 data hash * Test * Fixing test part1 * All tests passing * One last test * Updated config * Build proto * Proper skip message * Conflict and fmt * Removed crosslinks and shards. Built * Format and gazelle * Fixed all the block package tests * Fixed all the helper tests * All epoch package tests pass * All core package tests pass * Fixed operation tests * Started fixing rpc test * RPC tests passed! * Fixed all init sync tests * All tests pass * Fixed blockchain tests * Lint * Lint * Preston's feedback * Starting * Remove container * Fixed block spec tests * All passing except for block_processing test * Failing block processing test * Starting * Add AggregateAndProof * All mainnet test passes * Update deposit contract (#3906) * Proto spec tests v0.9 (#3908) * Starting * Add AggregateAndProof * Unskip block util tests (#3910) * rm'ed in protobuf * build proto * build proto * build proto * fix core package * Gazelle * Fixed all the tests * Fixed static test * Comment out spec test for now * One more skip * fix-roundRobinSync (#3862) * Starting but need new seed function * Revert initial sync * Updated Proposer Slashing * Fixed all tests * Lint * Update inclusion reward * Fill randao mixes with eth1 data hash * Test * Fixing test part1 * All tests passing * One last test * Updated config * Build proto * Proper skip message * Conflict and fmt * Removed crosslinks and shards. Built * Format and gazelle * Fixed all the block package tests * Fixed all the helper tests * All epoch package tests pass * All core package tests pass * Fixed operation tests * Started fixing rpc test * RPC tests passed! * Fixed all init sync tests * All tests pass * Fixed blockchain tests * Lint * Lint * Preston's feedback * Starting * Remove container * Fixed block spec tests * All passing except for block_processing test * Failing block processing test * Starting * Add AggregateAndProof * All mainnet test passes * Unskip block util tests * Slot processing spec test V0.9 (#3912) * Starting * Add AggregateAndProof * Unskip slot processing mainnet test * Unskip minimal spec test for finalization (#3920) * Remove outdated interop tests (#3922) * Rm outdated interop tests * Rm test runner * Gazelle * Update validator to use proposer slot (#3919) * Fix committee assignment (#3931) * Replace shard with committee index (#3930) * Conflict * Clean up (#3933) * Remove shard filter in db (#3936) * Remove lightouse compatibility test (#3939) * Update Committee Cache for v0.9 (#3948) * Updated committee cache * Removed shuffled indices cache * Started testing run time * Lint * Fixed test * Safeguard against nil head state * address edge case * add test * Fixed TestRoundRobinSync by doubling the epochs * Unskip TestProtoCompatability (#3958) * Unskip TestProtoCompatability * Update WORKSPACE * Fix minimal config (#3959) * fix minimal configs * fix hardcoded value in test * Simplify verify att time (#3961) * update readme for deposit contract, regen bindings for vyper 0.1.0b12 (#3963) * update readme for deposit contract, regen bindings * medium * Check nil base state (#3964) * Copy Block When Receiving it From Sync (#3966) * copy block * clone for other service methods too * Change logging of Bitfield (#3956) * change logging of bits * preston's review * Unskip Beacon Server Test (#3962) * run test till the end * fix up proto message types * fmt * resolve broken tests * better error handling * fixing new logic to use archived proposer info * fix up logic * clip using the max effective balance * broken build fix with num arg mismatch * amend archive * archival logic changed * rename test * archive both proposer and attester seeds * page size 100 * further experiments * further experimentation, archivedProposerIndex seems wrong * test passes * rem log * fix broken test * fix test * gaz * fix imports * ethapis
170 lines
5.1 KiB
Go
170 lines
5.1 KiB
Go
package depositcontract_test
|
|
|
|
import (
|
|
"context"
|
|
"encoding/binary"
|
|
"log"
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/ethereum/go-ethereum"
|
|
"github.com/ethereum/go-ethereum/common"
|
|
depositcontract "github.com/prysmaticlabs/prysm/contracts/deposit-contract"
|
|
"github.com/prysmaticlabs/prysm/shared/interop"
|
|
)
|
|
|
|
func TestSetupRegistrationContract_OK(t *testing.T) {
|
|
_, err := depositcontract.Setup()
|
|
if err != nil {
|
|
log.Fatalf("Can not deploy validator registration contract: %v", err)
|
|
}
|
|
}
|
|
|
|
// negative test case, deposit with less than 1 ETH which is less than the top off amount.
|
|
func TestRegister_Below1ETH(t *testing.T) {
|
|
testAccount, err := depositcontract.Setup()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Generate deposit data
|
|
privKeys, pubKeys, err := interop.DeterministicallyGenerateKeys(0 /*startIndex*/, 1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
depositDataItems, depositDataRoots, err := interop.DepositDataFromKeys(privKeys, pubKeys)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
var depositDataRoot [32]byte
|
|
copy(depositDataRoot[:], depositDataRoots[0])
|
|
testAccount.TxOpts.Value = depositcontract.LessThan1Eth()
|
|
_, err = testAccount.Contract.Deposit(testAccount.TxOpts, pubKeys[0].Marshal(), depositDataItems[0].WithdrawalCredentials, depositDataItems[0].Signature, depositDataRoot)
|
|
if err == nil {
|
|
t.Error("Validator registration should have failed with insufficient deposit")
|
|
}
|
|
}
|
|
|
|
// normal test case, test depositing 32 ETH and verify HashChainValue event is correctly emitted.
|
|
func TestValidatorRegister_OK(t *testing.T) {
|
|
testAccount, err := depositcontract.Setup()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
testAccount.TxOpts.Value = depositcontract.Amount32Eth()
|
|
|
|
// Generate deposit data
|
|
privKeys, pubKeys, err := interop.DeterministicallyGenerateKeys(0 /*startIndex*/, 1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
depositDataItems, depositDataRoots, err := interop.DepositDataFromKeys(privKeys, pubKeys)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
var depositDataRoot [32]byte
|
|
copy(depositDataRoot[:], depositDataRoots[0])
|
|
_, err = testAccount.Contract.Deposit(testAccount.TxOpts, pubKeys[0].Marshal(), depositDataItems[0].WithdrawalCredentials, depositDataItems[0].Signature, depositDataRoot)
|
|
testAccount.Backend.Commit()
|
|
if err != nil {
|
|
t.Errorf("Validator registration failed: %v", err)
|
|
}
|
|
_, err = testAccount.Contract.Deposit(testAccount.TxOpts, pubKeys[0].Marshal(), depositDataItems[0].WithdrawalCredentials, depositDataItems[0].Signature, depositDataRoot)
|
|
testAccount.Backend.Commit()
|
|
if err != nil {
|
|
t.Errorf("Validator registration failed: %v", err)
|
|
}
|
|
_, err = testAccount.Contract.Deposit(testAccount.TxOpts, pubKeys[0].Marshal(), depositDataItems[0].WithdrawalCredentials, depositDataItems[0].Signature, depositDataRoot)
|
|
testAccount.Backend.Commit()
|
|
if err != nil {
|
|
t.Errorf("Validator registration failed: %v", err)
|
|
}
|
|
|
|
query := ethereum.FilterQuery{
|
|
Addresses: []common.Address{
|
|
testAccount.ContractAddr,
|
|
},
|
|
}
|
|
|
|
logs, err := testAccount.Backend.FilterLogs(context.Background(), query)
|
|
if err != nil {
|
|
t.Fatalf("Unable to get logs of deposit contract: %v", err)
|
|
}
|
|
|
|
merkleTreeIndex := make([]uint64, 5)
|
|
|
|
for i, log := range logs {
|
|
_, _, _, _, idx, err := depositcontract.UnpackDepositLogData(log.Data)
|
|
if err != nil {
|
|
t.Fatalf("Unable to unpack log data: %v", err)
|
|
}
|
|
merkleTreeIndex[i] = binary.LittleEndian.Uint64(idx[:])
|
|
}
|
|
|
|
if merkleTreeIndex[0] != 0 {
|
|
t.Errorf("Deposit event total desposit count miss matched. Want: %d, Got: %d", 1, merkleTreeIndex[0])
|
|
}
|
|
|
|
if merkleTreeIndex[1] != 1 {
|
|
t.Errorf("Deposit event total desposit count miss matched. Want: %d, Got: %d", 2, merkleTreeIndex[1])
|
|
}
|
|
|
|
if merkleTreeIndex[2] != 2 {
|
|
t.Errorf("Deposit event total desposit count miss matched. Want: %v, Got: %v", 3, merkleTreeIndex[2])
|
|
}
|
|
}
|
|
|
|
func TestDrain(t *testing.T) {
|
|
testAccount, err := depositcontract.Setup()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
testAccount.TxOpts.Value = depositcontract.Amount32Eth()
|
|
|
|
// Generate deposit data
|
|
privKeys, pubKeys, err := interop.DeterministicallyGenerateKeys(0 /*startIndex*/, 1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
depositDataItems, depositDataRoots, err := interop.DepositDataFromKeys(privKeys, pubKeys)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
var depositDataRoot [32]byte
|
|
copy(depositDataRoot[:], depositDataRoots[0])
|
|
_, err = testAccount.Contract.Deposit(testAccount.TxOpts, pubKeys[0].Marshal(), depositDataItems[0].WithdrawalCredentials, depositDataItems[0].Signature, depositDataRoot)
|
|
testAccount.Backend.Commit()
|
|
if err != nil {
|
|
t.Errorf("Validator registration failed: %v", err)
|
|
}
|
|
|
|
testAccount.Backend.Commit()
|
|
|
|
ctx := context.Background()
|
|
bal, err := testAccount.Backend.BalanceAt(ctx, testAccount.ContractAddr, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if bal.Cmp(depositcontract.Amount32Eth()) != 0 {
|
|
t.Fatal("deposit didnt work")
|
|
}
|
|
|
|
testAccount.TxOpts.Value = big.NewInt(0)
|
|
if _, err := testAccount.Contract.Drain(testAccount.TxOpts); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
testAccount.Backend.Commit()
|
|
|
|
bal, err = testAccount.Backend.BalanceAt(ctx, testAccount.ContractAddr, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if big.NewInt(0).Cmp(bal) != 0 {
|
|
t.Errorf("Drain did not drain balance: %v", bal)
|
|
}
|
|
}
|