github.com/r8d8/go-ethereum@v5.5.2+incompatible/core/vm/jump_table_test.go (about)

     1  // Copyright 2016 The go-ethereum Authors
     2  // This file is part of the go-ethereum library.
     3  //
     4  // The go-ethereum library is free software: you can redistribute it and/or modify
     5  // it under the terms of the GNU Lesser General Public License as published by
     6  // the Free Software Foundation, either version 3 of the License, or
     7  // (at your option) any later version.
     8  //
     9  // The go-ethereum library is distributed in the hope that it will be useful,
    10  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    11  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    12  // GNU Lesser General Public License for more details.
    13  //
    14  // You should have received a copy of the GNU Lesser General Public License
    15  // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
    16  
    17  package vm
    18  
    19  import (
    20  	"math/big"
    21  	"testing"
    22  )
    23  
    24  type ruleSet struct {
    25  	hs *big.Int
    26  }
    27  
    28  func (r ruleSet) IsHomestead(n *big.Int) bool { return n.Cmp(r.hs) >= 0 }
    29  
    30  func (r ruleSet) GasTable(*big.Int) *GasTable {
    31  	return &GasTable{
    32  		ExtcodeSize: big.NewInt(20),
    33  		ExtcodeCopy: big.NewInt(20),
    34  		Balance:     big.NewInt(20),
    35  		SLoad:       big.NewInt(50),
    36  		Calls:       big.NewInt(40),
    37  		Suicide:     big.NewInt(0),
    38  		ExpByte:     big.NewInt(10),
    39  	}
    40  }
    41  
    42  func TestInit(t *testing.T) {
    43  	jumpTable := newJumpTable(ruleSet{big.NewInt(1)}, big.NewInt(0))
    44  	if jumpTable[DELEGATECALL].valid {
    45  		t.Error("Expected DELEGATECALL not to be present")
    46  	}
    47  
    48  	for _, n := range []int64{1, 2, 100} {
    49  		jumpTable := newJumpTable(ruleSet{big.NewInt(1)}, big.NewInt(n))
    50  		if !jumpTable[DELEGATECALL].valid {
    51  			t.Error("Expected DELEGATECALL to be present for block", n)
    52  		}
    53  	}
    54  }