You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
3.3 KiB
100 lines
3.3 KiB
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"math/big"
|
|
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
"github.com/ethereum/go-ethereum/params"
|
|
"github.com/harmony-one/harmony/consensus"
|
|
"github.com/harmony-one/harmony/core"
|
|
"github.com/harmony-one/harmony/core/types"
|
|
"github.com/harmony-one/harmony/core/vm"
|
|
"github.com/harmony-one/harmony/db"
|
|
"github.com/harmony-one/harmony/node/worker"
|
|
)
|
|
|
|
var (
|
|
// Test accounts
|
|
testBankKey, _ = crypto.GenerateKey()
|
|
testBankAddress = crypto.PubkeyToAddress(testBankKey.PublicKey)
|
|
testBankFunds = big.NewInt(1000000000000000000)
|
|
|
|
testUserKey, _ = crypto.GenerateKey()
|
|
testUserAddress = crypto.PubkeyToAddress(testUserKey.PublicKey)
|
|
|
|
chainConfig = params.TestChainConfig
|
|
|
|
// Test transactions
|
|
pendingTxs []*types.Transaction
|
|
newTxs []*types.Transaction
|
|
)
|
|
|
|
func init() {
|
|
tx1, _ := types.SignTx(types.NewTransaction(0, testUserAddress, big.NewInt(1000), params.TxGas, nil, nil), types.HomesteadSigner{}, testBankKey)
|
|
tx2, _ := types.SignTx(types.NewTransaction(1, testUserAddress, big.NewInt(1000), params.TxGas, nil, nil), types.HomesteadSigner{}, testBankKey)
|
|
tx3, _ := types.SignTx(types.NewTransaction(2, testUserAddress, big.NewInt(1000), params.TxGas, nil, nil), types.HomesteadSigner{}, testBankKey)
|
|
pendingTxs = append(pendingTxs, tx1)
|
|
pendingTxs = append(pendingTxs, tx2)
|
|
pendingTxs = append(pendingTxs, tx3)
|
|
tx4, _ := types.SignTx(types.NewTransaction(1, testUserAddress, big.NewInt(1000), params.TxGas, nil, nil), types.HomesteadSigner{}, testBankKey)
|
|
newTxs = append(newTxs, tx4)
|
|
}
|
|
|
|
type testWorkerBackend struct {
|
|
db db.Database
|
|
txPool *core.TxPool
|
|
chain *core.BlockChain
|
|
}
|
|
|
|
func main() {
|
|
var (
|
|
database = db.NewMemDatabase()
|
|
gspec = core.Genesis{
|
|
Config: chainConfig,
|
|
Alloc: core.GenesisAlloc{testBankAddress: {Balance: testBankFunds}},
|
|
}
|
|
)
|
|
|
|
genesis := gspec.MustCommit(database)
|
|
_ = genesis
|
|
chain, _ := core.NewBlockChain(database, nil, gspec.Config, consensus.NewFaker(), vm.Config{}, nil)
|
|
|
|
txpool := core.NewTxPool(core.DefaultTxPoolConfig, chainConfig, chain)
|
|
|
|
backend := &testWorkerBackend{
|
|
db: database,
|
|
chain: chain,
|
|
txPool: txpool,
|
|
}
|
|
backend.txPool.AddLocals(pendingTxs)
|
|
|
|
//// Generate a small n-block chain and an uncle block for it
|
|
n := 3
|
|
if n > 0 {
|
|
blocks, _ := core.GenerateChain(chainConfig, genesis, consensus.NewFaker(), database, n, func(i int, gen *core.BlockGen) {
|
|
gen.SetCoinbase(testBankAddress)
|
|
gen.AddTx(pendingTxs[i])
|
|
})
|
|
if _, err := chain.InsertChain(blocks); err != nil {
|
|
fmt.Errorf("failed to insert origin chain: %v", err)
|
|
}
|
|
}
|
|
|
|
txs := make([]*types.Transaction, 100)
|
|
worker := worker.New(params.TestChainConfig, chain, consensus.NewFaker())
|
|
fmt.Println(worker.GetCurrentState().GetBalance(testBankAddress))
|
|
fmt.Println(worker.Commit().Root())
|
|
|
|
for i, _ := range txs {
|
|
randomUserKey, _ := crypto.GenerateKey()
|
|
randomUserAddress := crypto.PubkeyToAddress(randomUserKey.PublicKey)
|
|
tx, _ := types.SignTx(types.NewTransaction(worker.GetCurrentState().GetNonce(crypto.PubkeyToAddress(testBankKey.PublicKey)), randomUserAddress, big.NewInt(1000), params.TxGas, nil, nil), types.HomesteadSigner{}, testBankKey)
|
|
txs[i] = tx
|
|
}
|
|
|
|
worker.CommitTransactions(txs, crypto.PubkeyToAddress(testBankKey.PublicKey))
|
|
|
|
fmt.Println(worker.GetCurrentState().GetBalance(testBankAddress))
|
|
fmt.Println(worker.Commit().Root())
|
|
}
|
|
|