github.com/zhiqiangxu/go-ethereum@v1.9.16-0.20210824055606-be91cfdebc48/rlp/iterator_test.go (about)

     1  // Copyright 2019 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 rlp
    18  
    19  import (
    20  	"testing"
    21  
    22  	"github.com/zhiqiangxu/go-ethereum/common/hexutil"
    23  )
    24  
    25  // TestIterator tests some basic things about the ListIterator. A more
    26  // comprehensive test can be found in core/rlp_test.go, where we can
    27  // use both types and rlp without dependency cycles
    28  func TestIterator(t *testing.T) {
    29  	bodyRlpHex := "0xf902cbf8d6f869800182c35094000000000000000000000000000000000000aaaa808a000000000000000000001ba01025c66fad28b4ce3370222624d952c35529e602af7cbe04f667371f61b0e3b3a00ab8813514d1217059748fd903288ace1b4001a4bc5fbde2790debdc8167de2ff869010182c35094000000000000000000000000000000000000aaaa808a000000000000000000001ca05ac4cf1d19be06f3742c21df6c49a7e929ceb3dbaf6a09f3cfb56ff6828bd9a7a06875970133a35e63ac06d360aa166d228cc013e9b96e0a2cae7f55b22e1ee2e8f901f0f901eda0c75448377c0e426b8017b23c5f77379ecf69abc1d5c224284ad3ba1c46c59adaa00000000000000000000000000000000000000000000000000000000000000000940000000000000000000000000000000000000000a00000000000000000000000000000000000000000000000000000000000000000a00000000000000000000000000000000000000000000000000000000000000000a00000000000000000000000000000000000000000000000000000000000000000b9010000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000808080808080a00000000000000000000000000000000000000000000000000000000000000000880000000000000000"
    30  	bodyRlp := hexutil.MustDecode(bodyRlpHex)
    31  
    32  	it, err := NewListIterator(bodyRlp)
    33  	if err != nil {
    34  		t.Fatal(err)
    35  	}
    36  	// Check that txs exist
    37  	if !it.Next() {
    38  		t.Fatal("expected two elems, got zero")
    39  	}
    40  	txs := it.Value()
    41  	// Check that uncles exist
    42  	if !it.Next() {
    43  		t.Fatal("expected two elems, got one")
    44  	}
    45  	txit, err := NewListIterator(txs)
    46  	if err != nil {
    47  		t.Fatal(err)
    48  	}
    49  	var i = 0
    50  	for txit.Next() {
    51  		if txit.err != nil {
    52  			t.Fatal(txit.err)
    53  		}
    54  		i++
    55  	}
    56  	if exp := 2; i != exp {
    57  		t.Errorf("count wrong, expected %d got %d", i, exp)
    58  	}
    59  }