github.com/nspcc-dev/neo-go@v0.105.2-0.20240517133400-6be757af3eba/pkg/network/payload/headers_test.go (about)

     1  package payload
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/nspcc-dev/neo-go/internal/testserdes"
     7  	"github.com/nspcc-dev/neo-go/pkg/core/block"
     8  	"github.com/nspcc-dev/neo-go/pkg/core/transaction"
     9  	"github.com/stretchr/testify/assert"
    10  )
    11  
    12  func TestHeadersEncodeDecode(t *testing.T) {
    13  	t.Run("normal case", func(t *testing.T) {
    14  		headers := newTestHeaders(3)
    15  
    16  		testHeadersEncodeDecode(t, headers, 3, nil)
    17  	})
    18  
    19  	t.Run("more than max", func(t *testing.T) {
    20  		const sent = MaxHeadersAllowed + 1
    21  		headers := newTestHeaders(sent)
    22  
    23  		testHeadersEncodeDecode(t, headers, MaxHeadersAllowed, ErrTooManyHeaders)
    24  	})
    25  
    26  	t.Run("zero", func(t *testing.T) {
    27  		headers := newTestHeaders(0)
    28  
    29  		testHeadersEncodeDecode(t, headers, 0, ErrNoHeaders)
    30  	})
    31  }
    32  
    33  func newTestHeaders(n int) *Headers {
    34  	headers := &Headers{Hdrs: make([]*block.Header, n)}
    35  
    36  	for i := range headers.Hdrs {
    37  		headers.Hdrs[i] = &block.Header{
    38  			Index: uint32(i + 1),
    39  			Script: transaction.Witness{
    40  				InvocationScript:   []byte{0x0},
    41  				VerificationScript: []byte{0x1},
    42  			},
    43  		}
    44  	}
    45  
    46  	return headers
    47  }
    48  
    49  func testHeadersEncodeDecode(t *testing.T, headers *Headers, expected int, retErr error) {
    50  	data, err := testserdes.EncodeBinary(headers)
    51  	assert.Nil(t, err)
    52  
    53  	headersDecode := &Headers{}
    54  	rErr := testserdes.DecodeBinary(data, headersDecode)
    55  
    56  	assert.Equal(t, retErr, rErr)
    57  	assert.Equal(t, expected, len(headersDecode.Hdrs))
    58  
    59  	for i := 0; i < len(headersDecode.Hdrs); i++ {
    60  		assert.Equal(t, headers.Hdrs[i].Version, headersDecode.Hdrs[i].Version)
    61  		assert.Equal(t, headers.Hdrs[i].Index, headersDecode.Hdrs[i].Index)
    62  		assert.Equal(t, headers.Hdrs[i].Script, headersDecode.Hdrs[i].Script)
    63  	}
    64  }
    65  
    66  //TODO NEO3.0: Update binary
    67  /*
    68  func TestBinEncodeDecode(t *testing.T) {
    69  	rawBlockHeaders := "010000000026b3c3df4dc1602a3b0e6989248b23275b5e4014a159af5dce69e16d4ab75f00f439321a51f425a530820cfe4d715bfd835b49687e87772f2c4737b8bc586dca7fda03580a000000bf14ff160228f0c059e75d652b5d3827bf04c165bbe9ef95cca4bf5501fd45014036fdd23248880c1c311bcd97df04fe6d740dc1bf340c26915f0466e31e81c039012eca7a760270389e04b58b99820fe49cf8c24c9afc65d696b4d3f406a1e6b5405172a9b461e68dd399c8716de11d31f7dd2ec3be327c636b024562db6ac5df1cffdbee74c994736fd49803234d2baffbc0054f28ba5ec76494a467b4106955bb4084af7746d269241628c667003e9d39288b190ad5cef218ada625cbba8be411bb153828d8d3634e8f586638e2448425bc5b671be69800392ccbdebc945a5099c7406f6a11824105ecad345e525957053e77fbc0119d6b3fa7f854527e816cfce0d95dac66888e07e8990c95103d8e46124aac16f152e088520d7ec8325e3a2456f840e5b77ef0e3c410b347ccaf8a87516d10b88d436563c80712153273993afc320ec49b638225f58de464a1345e62a564b398939f96f6f4b7cf21b583609f85495af1552102486fd15702c4490a26703112a5cc1d0923fd697a33406bd5a1c00e0013b09a7021024c7b7fb6c310fccf1ba33b082519d82964ea93868d676662d4a59ad548df0e7d2102aaec38470f6aad0042c6e877cfd8087d2676b0f516fddd362801b9bd3936399e2103b209fd4f53a7170ea4444e0cb0a6bb6a53c2bd016926989cf85f9b0fba17a70c2103b8d9d5771d8f513aa0869b9cc8d50986403b78c6da36890638c3d46a5adce04a2102ca0e27697b9c248f6f16e085fd0061e26f44da85b58ee835c110caa5ec3ba5542102df48f60e8f3e01c48ff40b9b7f1310d7a8b2a193188befe1c2e3df740e89509357ae00"
    70  
    71  	var headerMsg Headers
    72  
    73  	rawBlockBytes, _ := hex.DecodeString(rawBlockHeaders)
    74  
    75  	assert.NoError(t, testserdes.DecodeBinary(rawBlockBytes, &headerMsg))
    76  	assert.Equal(t, 1, len(headerMsg.Hdrs))
    77  
    78  	header := headerMsg.Hdrs[0]
    79  	hash := header.Hash()
    80  
    81  	assert.Equal(t, "f3c4ec44c07eccbda974f1ee34bc6654ab6d3f22cd89c2e5c593a16d6cc7e6e8", hash.StringLE())
    82  
    83  	data, err := testserdes.EncodeBinary(&headerMsg)
    84  	assert.NoError(t, err)
    85  	assert.Equal(t, hex.EncodeToString(rawBlockBytes), hex.EncodeToString(data))
    86  }
    87  */