github.com/aaa256/atlantis@v0.0.0-20210707112435-42ee889287a2/core/vm/intpool_test.go (about)

     1  // Copyright 2018 The go-athereum Authors
     2  // This file is part of the go-athereum library.
     3  //
     4  // The go-athereum 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-athereum 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-athereum library. If not, see <http://www.gnu.org/licenses/>.
    16  
    17  package vm
    18  
    19  import (
    20  	"testing"
    21  )
    22  
    23  func TestIntPoolPoolGet(t *testing.T) {
    24  	poolOfIntPools.pools = make([]*intPool, 0, poolDefaultCap)
    25  
    26  	nip := poolOfIntPools.get()
    27  	if nip == nil {
    28  		t.Fatalf("Invalid pool allocation")
    29  	}
    30  }
    31  
    32  func TestIntPoolPoolPut(t *testing.T) {
    33  	poolOfIntPools.pools = make([]*intPool, 0, poolDefaultCap)
    34  
    35  	nip := poolOfIntPools.get()
    36  	if len(poolOfIntPools.pools) != 0 {
    37  		t.Fatalf("Pool got added to list when none should have been")
    38  	}
    39  
    40  	poolOfIntPools.put(nip)
    41  	if len(poolOfIntPools.pools) == 0 {
    42  		t.Fatalf("Pool did not get added to list when one should have been")
    43  	}
    44  }
    45  
    46  func TestIntPoolPoolReUse(t *testing.T) {
    47  	poolOfIntPools.pools = make([]*intPool, 0, poolDefaultCap)
    48  	nip := poolOfIntPools.get()
    49  	poolOfIntPools.put(nip)
    50  	poolOfIntPools.get()
    51  
    52  	if len(poolOfIntPools.pools) != 0 {
    53  		t.Fatalf("Invalid number of pools. Got %d, expected %d", len(poolOfIntPools.pools), 0)
    54  	}
    55  }