move setup staking contract

pull/573/head
Minh Doan 6 years ago committed by Minh Doan
parent 6099b8e575
commit 4845384266
  1. 201
      test/chain/main.go

@ -191,7 +191,7 @@ func playFaucetContract(chain *core.BlockChain) {
callFaucetContractToFundAnAddress(chain)
}
func playStakingContract(chain *core.BlockChain) {
func playSetupStakingContract(chain *core.BlockChain) {
fmt.Println()
fmt.Println("--------- ************************** ---------")
fmt.Println()
@ -220,105 +220,108 @@ func playStakingContract(chain *core.BlockChain) {
fmt.Println(state.GetBalance(StakeContractAddress))
fmt.Println("stake address balance :")
fmt.Println(state.GetBalance(StakingAddress))
}
depositFnSignature := []byte("deposit()")
//hash = sha3.NewKeccak256()
hash := sha3.NewLegacyKeccak256()
hash.Write(depositFnSignature)
methodID := hash.Sum(nil)[:4]
var callEncl []byte
callEncl = append(callEncl, methodID...)
stake := 100000
fmt.Println()
fmt.Println("--------- Staking Contract added to txns ---------")
fmt.Println()
fmt.Printf("-- Now Staking with stake: %d --\n", stake)
fmt.Println()
for i := 0; i <= 2; i++ {
//Deposit Does not take a argument, stake is transferred via amount.
tx, _ := types.SignTx(types.NewTransaction(0, StakeContractAddress, 0, big.NewInt(int64(stake)), params.TxGas*5, nil, callEncl), types.HomesteadSigner{}, AllRandomUserKey[i])
stakingtxns = append(stakingtxns, tx)
}
err = contractworker.CommitTransactions(stakingtxns)
if err != nil {
fmt.Println(err)
}
block, _ = contractworker.Commit()
_, err = chain.InsertChain(types.Blocks{block})
if err != nil {
fmt.Println(err)
}
// receipts := contractworker.GetCurrentReceipts()
// fmt.Println(receipts[len(receipts)-4].ContractAddress)
state = contractworker.GetCurrentState()
fmt.Printf("After Staking Balances (should be less by %d)\n", stake)
fmt.Println("user address balance")
fmt.Println(state.GetBalance(AllRandomUserAddress[0]))
fmt.Println("The balances for 2 more users:")
fmt.Println(state.GetBalance(AllRandomUserAddress[1]))
fmt.Println(state.GetBalance(AllRandomUserAddress[2]))
fmt.Println("faucet contract balance (unchanged):")
fmt.Println(state.GetBalance(FaucetContractAddress))
fmt.Println("stake contract balance :")
fmt.Println(state.GetBalance(StakeContractAddress))
fmt.Println("stake address balance :")
fmt.Println(state.GetBalance(StakingAddress))
fmt.Println()
fmt.Println("--------- Now Setting up Withdrawing Stakes ---------")
withdrawFnSignature := []byte("withdraw(uint256)")
//hash = sha3.NewKeccak256()
hash = sha3.NewLegacyKeccak256()
hash.Write(withdrawFnSignature)
methodID = hash.Sum(nil)[:4]
withdraw := "5000"
withdrawstake := new(big.Int)
withdrawstake.SetString(withdraw, 10)
paddedAmount := common.LeftPadBytes(withdrawstake.Bytes(), 32)
var dataEncl []byte
dataEncl = append(dataEncl, methodID...)
dataEncl = append(dataEncl, paddedAmount...)
var withdrawstakingtxns []*types.Transaction
fmt.Println()
fmt.Printf("-- Withdrawing Stake by amount: %s --\n", withdraw)
fmt.Println()
for i := 0; i <= 2; i++ {
cnonce := contractworker.GetCurrentState().GetNonce(AllRandomUserAddress[i])
tx, _ := types.SignTx(types.NewTransaction(cnonce, StakeContractAddress, 0, big.NewInt(0), params.TxGas*5, nil, dataEncl), types.HomesteadSigner{}, AllRandomUserKey[i])
withdrawstakingtxns = append(withdrawstakingtxns, tx)
}
err = contractworker.CommitTransactions(withdrawstakingtxns)
if err != nil {
fmt.Println("error:")
fmt.Println(err)
}
block, _ = contractworker.Commit()
_, err = chain.InsertChain(types.Blocks{block})
if err != nil {
fmt.Println(err)
}
state = contractworker.GetCurrentState()
fmt.Printf("Withdraw Staking Balances (should be up by %s)\n", withdraw)
fmt.Println(state.GetBalance(AllRandomUserAddress[0]))
fmt.Println(state.GetBalance(AllRandomUserAddress[1]))
fmt.Println(state.GetBalance(AllRandomUserAddress[2]))
fmt.Println("faucet contract balance (unchanged):")
fmt.Println(state.GetBalance(FaucetContractAddress))
fmt.Printf("stake contract balance (should downup by %s)\n", withdraw)
fmt.Println(state.GetBalance(StakeContractAddress))
fmt.Println("stake address balance :")
fmt.Println(state.GetBalance(StakingAddress))
func playStakingContract(chain *core.BlockChain) {
playSetupStakingContract(chain)
// fmt.Println()
// fmt.Println("--------- Staking Contract added to txns ---------")
// fmt.Println()
// fmt.Printf("-- Now Staking with stake: %d --\n", stake)
// fmt.Println()
// depositFnSignature := []byte("deposit()")
// //hash = sha3.NewKeccak256()
// hash := sha3.NewLegacyKeccak256()
// hash.Write(depositFnSignature)
// methodID := hash.Sum(nil)[:4]
// var callEncl []byte
// callEncl = append(callEncl, methodID...)
// stake := 100000
// for i := 0; i <= 2; i++ {
// //Deposit Does not take a argument, stake is transferred via amount.
// tx, _ := types.SignTx(types.NewTransaction(0, StakeContractAddress, 0, big.NewInt(int64(stake)), params.TxGas*5, nil, callEncl), types.HomesteadSigner{}, AllRandomUserKey[i])
// stakingtxns = append(stakingtxns, tx)
// }
// err = contractworker.CommitTransactions(stakingtxns)
// if err != nil {
// fmt.Println(err)
// }
// block, _ = contractworker.Commit()
// _, err = chain.InsertChain(types.Blocks{block})
// if err != nil {
// fmt.Println(err)
// }
// // receipts := contractworker.GetCurrentReceipts()
// // fmt.Println(receipts[len(receipts)-4].ContractAddress)
// state = contractworker.GetCurrentState()
// fmt.Printf("After Staking Balances (should be less by %d)\n", stake)
// fmt.Println("user address balance")
// fmt.Println(state.GetBalance(AllRandomUserAddress[0]))
// fmt.Println("The balances for 2 more users:")
// fmt.Println(state.GetBalance(AllRandomUserAddress[1]))
// fmt.Println(state.GetBalance(AllRandomUserAddress[2]))
// fmt.Println("faucet contract balance (unchanged):")
// fmt.Println(state.GetBalance(FaucetContractAddress))
// fmt.Println("stake contract balance :")
// fmt.Println(state.GetBalance(StakeContractAddress))
// fmt.Println("stake address balance :")
// fmt.Println(state.GetBalance(StakingAddress))
// fmt.Println()
// fmt.Println("--------- Now Setting up Withdrawing Stakes ---------")
// withdrawFnSignature := []byte("withdraw(uint256)")
// //hash = sha3.NewKeccak256()
// hash = sha3.NewLegacyKeccak256()
// hash.Write(withdrawFnSignature)
// methodID = hash.Sum(nil)[:4]
// withdraw := "5000"
// withdrawstake := new(big.Int)
// withdrawstake.SetString(withdraw, 10)
// paddedAmount := common.LeftPadBytes(withdrawstake.Bytes(), 32)
// var dataEncl []byte
// dataEncl = append(dataEncl, methodID...)
// dataEncl = append(dataEncl, paddedAmount...)
// var withdrawstakingtxns []*types.Transaction
// fmt.Println()
// fmt.Printf("-- Withdrawing Stake by amount: %s --\n", withdraw)
// fmt.Println()
// for i := 0; i <= 2; i++ {
// cnonce := contractworker.GetCurrentState().GetNonce(AllRandomUserAddress[i])
// tx, _ := types.SignTx(types.NewTransaction(cnonce, StakeContractAddress, 0, big.NewInt(0), params.TxGas*5, nil, dataEncl), types.HomesteadSigner{}, AllRandomUserKey[i])
// withdrawstakingtxns = append(withdrawstakingtxns, tx)
// }
// err = contractworker.CommitTransactions(withdrawstakingtxns)
// if err != nil {
// fmt.Println("error:")
// fmt.Println(err)
// }
// block, _ = contractworker.Commit()
// _, err = chain.InsertChain(types.Blocks{block})
// if err != nil {
// fmt.Println(err)
// }
// state = contractworker.GetCurrentState()
// fmt.Printf("Withdraw Staking Balances (should be up by %s)\n", withdraw)
// fmt.Println(state.GetBalance(AllRandomUserAddress[0]))
// fmt.Println(state.GetBalance(AllRandomUserAddress[1]))
// fmt.Println(state.GetBalance(AllRandomUserAddress[2]))
// fmt.Println("faucet contract balance (unchanged):")
// fmt.Println(state.GetBalance(FaucetContractAddress))
// fmt.Printf("stake contract balance (should downup by %s)\n", withdraw)
// fmt.Println(state.GetBalance(StakeContractAddress))
// fmt.Println("stake address balance :")
// fmt.Println(state.GetBalance(StakingAddress))
}

Loading…
Cancel
Save