github.com/aidoskuneen/adk-node@v0.0.0-20220315131952-2e32567cb7f4/p2p/rlpx/buffer_test.go (about) 1 // Copyright 2021 The adkgo Authors 2 // This file is part of the adkgo library (adapted for adkgo from go--ethereum v1.10.8). 3 // 4 // the adkgo 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 adkgo 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 adkgo library. If not, see <http://www.gnu.org/licenses/>. 16 17 package rlpx 18 19 import ( 20 "bytes" 21 "testing" 22 23 "github.com/aidoskuneen/adk-node/common/hexutil" 24 "github.com/stretchr/testify/assert" 25 ) 26 27 func TestReadBufferReset(t *testing.T) { 28 reader := bytes.NewReader(hexutil.MustDecode("0x010202030303040505")) 29 var b readBuffer 30 31 s1, _ := b.read(reader, 1) 32 s2, _ := b.read(reader, 2) 33 s3, _ := b.read(reader, 3) 34 35 assert.Equal(t, []byte{1}, s1) 36 assert.Equal(t, []byte{2, 2}, s2) 37 assert.Equal(t, []byte{3, 3, 3}, s3) 38 39 b.reset() 40 41 s4, _ := b.read(reader, 1) 42 s5, _ := b.read(reader, 2) 43 44 assert.Equal(t, []byte{4}, s4) 45 assert.Equal(t, []byte{5, 5}, s5) 46 47 s6, err := b.read(reader, 2) 48 49 assert.EqualError(t, err, "EOF") 50 assert.Nil(t, s6) 51 }