github.com/sberex/go-sberex@v1.8.2-0.20181113200658-ed96ac38f7d7/eth/tracers/tracers_test.go (about) 1 // This file is part of the go-sberex library. The go-sberex library is 2 // free software: you can redistribute it and/or modify it under the terms 3 // of the GNU Lesser General Public License as published by the Free 4 // Software Foundation, either version 3 of the License, or (at your option) 5 // any later version. 6 // 7 // The go-sberex library is distributed in the hope that it will be useful, 8 // but WITHOUT ANY WARRANTY; without even the implied warranty of 9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser 10 // General Public License <http://www.gnu.org/licenses/> for more details. 11 12 package tracers 13 14 import ( 15 "encoding/json" 16 "io/ioutil" 17 "math/big" 18 "path/filepath" 19 "reflect" 20 "strings" 21 "testing" 22 23 "github.com/Sberex/go-sberex/common" 24 "github.com/Sberex/go-sberex/common/hexutil" 25 "github.com/Sberex/go-sberex/common/math" 26 "github.com/Sberex/go-sberex/core" 27 "github.com/Sberex/go-sberex/core/types" 28 "github.com/Sberex/go-sberex/core/vm" 29 "github.com/Sberex/go-sberex/ethdb" 30 "github.com/Sberex/go-sberex/rlp" 31 "github.com/Sberex/go-sberex/tests" 32 ) 33 34 // To generate a new callTracer test, copy paste the makeTest method below into 35 // a Geth console and call it with a transaction hash you which to export. 36 37 /* 38 // makeTest generates a callTracer test by running a prestate reassembled and a 39 // call trace run, assembling all the gathered information into a test case. 40 var makeTest = function(tx, rewind) { 41 // Generate the genesis block from the block, transaction and prestate data 42 var block = eth.getBlock(eth.getTransaction(tx).blockHash); 43 var genesis = eth.getBlock(block.parentHash); 44 45 delete genesis.gasUsed; 46 delete genesis.logsBloom; 47 delete genesis.parentHash; 48 delete genesis.receiptsRoot; 49 delete genesis.sha3Uncles; 50 delete genesis.size; 51 delete genesis.transactions; 52 delete genesis.transactionsRoot; 53 delete genesis.uncles; 54 55 genesis.gasLimit = genesis.gasLimit.toString(); 56 genesis.number = genesis.number.toString(); 57 genesis.timestamp = genesis.timestamp.toString(); 58 59 genesis.alloc = debug.traceTransaction(tx, {tracer: "prestateTracer", rewind: rewind}); 60 for (var key in genesis.alloc) { 61 genesis.alloc[key].nonce = genesis.alloc[key].nonce.toString(); 62 } 63 genesis.config = admin.nodeInfo.protocols.eth.config; 64 65 // Generate the call trace and produce the test input 66 var result = debug.traceTransaction(tx, {tracer: "callTracer", rewind: rewind}); 67 delete result.time; 68 69 console.log(JSON.stringify({ 70 genesis: genesis, 71 context: { 72 number: block.number.toString(), 73 difficulty: block.difficulty, 74 timestamp: block.timestamp.toString(), 75 gasLimit: block.gasLimit.toString(), 76 miner: block.miner, 77 }, 78 input: eth.getRawTransaction(tx), 79 result: result, 80 }, null, 2)); 81 } 82 */ 83 84 // callTrace is the result of a callTracer run. 85 type callTrace struct { 86 Type string `json:"type"` 87 From common.Address `json:"from"` 88 To common.Address `json:"to"` 89 Input hexutil.Bytes `json:"input"` 90 Output hexutil.Bytes `json:"output"` 91 Gas *hexutil.Uint64 `json:"gas,omitempty"` 92 GasUsed *hexutil.Uint64 `json:"gasUsed,omitempty"` 93 Value *hexutil.Big `json:"value,omitempty"` 94 Error string `json:"error,omitempty"` 95 Calls []callTrace `json:"calls,omitempty"` 96 } 97 98 type callContext struct { 99 Number math.HexOrDecimal64 `json:"number"` 100 Difficulty *math.HexOrDecimal256 `json:"difficulty"` 101 Time math.HexOrDecimal64 `json:"timestamp"` 102 GasLimit math.HexOrDecimal64 `json:"gasLimit"` 103 Miner common.Address `json:"miner"` 104 } 105 106 // callTracerTest defines a single test to check the call tracer against. 107 type callTracerTest struct { 108 Genesis *core.Genesis `json:"genesis"` 109 Context *callContext `json:"context"` 110 Input string `json:"input"` 111 Result *callTrace `json:"result"` 112 } 113 114 // Iterates over all the input-output datasets in the tracer test harness and 115 // runs the JavaScript tracers against them. 116 func TestCallTracer(t *testing.T) { 117 files, err := ioutil.ReadDir("testdata") 118 if err != nil { 119 t.Fatalf("failed to retrieve tracer test suite: %v", err) 120 } 121 for _, file := range files { 122 if !strings.HasPrefix(file.Name(), "call_tracer_") { 123 continue 124 } 125 file := file // capture range variable 126 t.Run(camel(strings.TrimSuffix(strings.TrimPrefix(file.Name(), "call_tracer_"), ".json")), func(t *testing.T) { 127 t.Parallel() 128 129 // Call tracer test found, read if from disk 130 blob, err := ioutil.ReadFile(filepath.Join("testdata", file.Name())) 131 if err != nil { 132 t.Fatalf("failed to read testcase: %v", err) 133 } 134 test := new(callTracerTest) 135 if err := json.Unmarshal(blob, test); err != nil { 136 t.Fatalf("failed to parse testcase: %v", err) 137 } 138 // Configure a blockchain with the given prestate 139 tx := new(types.Transaction) 140 if err := rlp.DecodeBytes(common.FromHex(test.Input), tx); err != nil { 141 t.Fatalf("failed to parse testcase input: %v", err) 142 } 143 signer := types.MakeSigner(test.Genesis.Config, new(big.Int).SetUint64(uint64(test.Context.Number))) 144 origin, _ := signer.Sender(tx) 145 146 context := vm.Context{ 147 CanTransfer: core.CanTransfer, 148 Transfer: core.Transfer, 149 Origin: origin, 150 Coinbase: test.Context.Miner, 151 BlockNumber: new(big.Int).SetUint64(uint64(test.Context.Number)), 152 Time: new(big.Int).SetUint64(uint64(test.Context.Time)), 153 Difficulty: (*big.Int)(test.Context.Difficulty), 154 GasLimit: uint64(test.Context.GasLimit), 155 GasPrice: tx.GasPrice(), 156 } 157 db, _ := ethdb.NewMemDatabase() 158 statedb := tests.MakePreState(db, test.Genesis.Alloc) 159 160 // Create the tracer, the EVM environment and run it 161 tracer, err := New("callTracer") 162 if err != nil { 163 t.Fatalf("failed to create call tracer: %v", err) 164 } 165 evm := vm.NewEVM(context, statedb, test.Genesis.Config, vm.Config{Debug: true, Tracer: tracer}) 166 167 msg, err := tx.AsMessage(signer) 168 if err != nil { 169 t.Fatalf("failed to prepare transaction for tracing: %v", err) 170 } 171 st := core.NewStateTransition(evm, msg, new(core.GasPool).AddGas(tx.Gas())) 172 if _, _, _, err = st.TransitionDb(); err != nil { 173 t.Fatalf("failed to execute transaction: %v", err) 174 } 175 // Retrieve the trace result and compare against the etalon 176 res, err := tracer.GetResult() 177 if err != nil { 178 t.Fatalf("failed to retrieve trace result: %v", err) 179 } 180 ret := new(callTrace) 181 if err := json.Unmarshal(res, ret); err != nil { 182 t.Fatalf("failed to unmarshal trace result: %v", err) 183 } 184 if !reflect.DeepEqual(ret, test.Result) { 185 t.Fatalf("trace mismatch: have %+v, want %+v", ret, test.Result) 186 } 187 }) 188 } 189 }