github.com/sberex/go-sberex@v1.8.2-0.20181113200658-ed96ac38f7d7/core/vm/runtime/runtime_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 runtime
    13  
    14  import (
    15  	"math/big"
    16  	"strings"
    17  	"testing"
    18  
    19  	"github.com/Sberex/go-sberex/accounts/abi"
    20  	"github.com/Sberex/go-sberex/common"
    21  	"github.com/Sberex/go-sberex/core/state"
    22  	"github.com/Sberex/go-sberex/core/vm"
    23  	"github.com/Sberex/go-sberex/ethdb"
    24  )
    25  
    26  func TestDefaults(t *testing.T) {
    27  	cfg := new(Config)
    28  	setDefaults(cfg)
    29  
    30  	if cfg.Difficulty == nil {
    31  		t.Error("expected difficulty to be non nil")
    32  	}
    33  
    34  	if cfg.Time == nil {
    35  		t.Error("expected time to be non nil")
    36  	}
    37  	if cfg.GasLimit == 0 {
    38  		t.Error("didn't expect gaslimit to be zero")
    39  	}
    40  	if cfg.GasPrice == nil {
    41  		t.Error("expected time to be non nil")
    42  	}
    43  	if cfg.Value == nil {
    44  		t.Error("expected time to be non nil")
    45  	}
    46  	if cfg.GetHashFn == nil {
    47  		t.Error("expected time to be non nil")
    48  	}
    49  	if cfg.BlockNumber == nil {
    50  		t.Error("expected block number to be non nil")
    51  	}
    52  }
    53  
    54  func TestEVM(t *testing.T) {
    55  	defer func() {
    56  		if r := recover(); r != nil {
    57  			t.Fatalf("crashed with: %v", r)
    58  		}
    59  	}()
    60  
    61  	Execute([]byte{
    62  		byte(vm.DIFFICULTY),
    63  		byte(vm.TIMESTAMP),
    64  		byte(vm.GASLIMIT),
    65  		byte(vm.PUSH1),
    66  		byte(vm.ORIGIN),
    67  		byte(vm.BLOCKHASH),
    68  		byte(vm.COINBASE),
    69  	}, nil, nil)
    70  }
    71  
    72  func TestExecute(t *testing.T) {
    73  	ret, _, err := Execute([]byte{
    74  		byte(vm.PUSH1), 10,
    75  		byte(vm.PUSH1), 0,
    76  		byte(vm.MSTORE),
    77  		byte(vm.PUSH1), 32,
    78  		byte(vm.PUSH1), 0,
    79  		byte(vm.RETURN),
    80  	}, nil, nil)
    81  	if err != nil {
    82  		t.Fatal("didn't expect error", err)
    83  	}
    84  
    85  	num := new(big.Int).SetBytes(ret)
    86  	if num.Cmp(big.NewInt(10)) != 0 {
    87  		t.Error("Expected 10, got", num)
    88  	}
    89  }
    90  
    91  func TestCall(t *testing.T) {
    92  	db, _ := ethdb.NewMemDatabase()
    93  	state, _ := state.New(common.Hash{}, state.NewDatabase(db))
    94  	address := common.HexToAddress("0x0a")
    95  	state.SetCode(address, []byte{
    96  		byte(vm.PUSH1), 10,
    97  		byte(vm.PUSH1), 0,
    98  		byte(vm.MSTORE),
    99  		byte(vm.PUSH1), 32,
   100  		byte(vm.PUSH1), 0,
   101  		byte(vm.RETURN),
   102  	})
   103  
   104  	ret, _, err := Call(address, nil, &Config{State: state})
   105  	if err != nil {
   106  		t.Fatal("didn't expect error", err)
   107  	}
   108  
   109  	num := new(big.Int).SetBytes(ret)
   110  	if num.Cmp(big.NewInt(10)) != 0 {
   111  		t.Error("Expected 10, got", num)
   112  	}
   113  }
   114  
   115  func BenchmarkCall(b *testing.B) {
   116  	var definition = `[{"constant":true,"inputs":[],"name":"seller","outputs":[{"name":"","type":"address"}],"type":"function"},{"constant":false,"inputs":[],"name":"abort","outputs":[],"type":"function"},{"constant":true,"inputs":[],"name":"value","outputs":[{"name":"","type":"uint256"}],"type":"function"},{"constant":false,"inputs":[],"name":"refund","outputs":[],"type":"function"},{"constant":true,"inputs":[],"name":"buyer","outputs":[{"name":"","type":"address"}],"type":"function"},{"constant":false,"inputs":[],"name":"confirmReceived","outputs":[],"type":"function"},{"constant":true,"inputs":[],"name":"state","outputs":[{"name":"","type":"uint8"}],"type":"function"},{"constant":false,"inputs":[],"name":"confirmPurchase","outputs":[],"type":"function"},{"inputs":[],"type":"constructor"},{"anonymous":false,"inputs":[],"name":"Aborted","type":"event"},{"anonymous":false,"inputs":[],"name":"PurchaseConfirmed","type":"event"},{"anonymous":false,"inputs":[],"name":"ItemReceived","type":"event"},{"anonymous":false,"inputs":[],"name":"Refunded","type":"event"}]`
   117  
   118  	var code = common.Hex2Bytes("6060604052361561006c5760e060020a600035046308551a53811461007457806335a063b4146100865780633fa4f245146100a6578063590e1ae3146100af5780637150d8ae146100cf57806373fac6f0146100e1578063c19d93fb146100fe578063d696069714610112575b610131610002565b610133600154600160a060020a031681565b610131600154600160a060020a0390811633919091161461015057610002565b61014660005481565b610131600154600160a060020a039081163391909116146102d557610002565b610133600254600160a060020a031681565b610131600254600160a060020a0333811691161461023757610002565b61014660025460ff60a060020a9091041681565b61013160025460009060ff60a060020a9091041681146101cc57610002565b005b600160a060020a03166060908152602090f35b6060908152602090f35b60025460009060a060020a900460ff16811461016b57610002565b600154600160a060020a03908116908290301631606082818181858883f150506002805460a060020a60ff02191660a160020a179055506040517f72c874aeff0b183a56e2b79c71b46e1aed4dee5e09862134b8821ba2fddbf8bf9250a150565b80546002023414806101dd57610002565b6002805460a060020a60ff021973ffffffffffffffffffffffffffffffffffffffff1990911633171660a060020a1790557fd5d55c8a68912e9a110618df8d5e2e83b8d83211c57a8ddd1203df92885dc881826060a15050565b60025460019060a060020a900460ff16811461025257610002565b60025460008054600160a060020a0390921691606082818181858883f150508354604051600160a060020a0391821694503090911631915082818181858883f150506002805460a060020a60ff02191660a160020a179055506040517fe89152acd703c9d8c7d28829d443260b411454d45394e7995815140c8cbcbcf79250a150565b60025460019060a060020a900460ff1681146102f057610002565b6002805460008054600160a060020a0390921692909102606082818181858883f150508354604051600160a060020a0391821694503090911631915082818181858883f150506002805460a060020a60ff02191660a160020a179055506040517f8616bbbbad963e4e65b1366f1d75dfb63f9e9704bbbf91fb01bec70849906cf79250a15056")
   119  
   120  	abi, err := abi.JSON(strings.NewReader(definition))
   121  	if err != nil {
   122  		b.Fatal(err)
   123  	}
   124  
   125  	cpurchase, err := abi.Pack("confirmPurchase")
   126  	if err != nil {
   127  		b.Fatal(err)
   128  	}
   129  	creceived, err := abi.Pack("confirmReceived")
   130  	if err != nil {
   131  		b.Fatal(err)
   132  	}
   133  	refund, err := abi.Pack("refund")
   134  	if err != nil {
   135  		b.Fatal(err)
   136  	}
   137  
   138  	b.ResetTimer()
   139  	for i := 0; i < b.N; i++ {
   140  		for j := 0; j < 400; j++ {
   141  			Execute(code, cpurchase, nil)
   142  			Execute(code, creceived, nil)
   143  			Execute(code, refund, nil)
   144  		}
   145  	}
   146  }