From 7069bd3181343d06760ed7b651bd58b9e7136092 Mon Sep 17 00:00:00 2001 From: Quentin McGaw Date: Mon, 29 Nov 2021 17:19:02 +0000 Subject: [PATCH] Fix encode decode tests --- lib/trie/encodedecode_test/branch_test.go | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/lib/trie/encodedecode_test/branch_test.go b/lib/trie/encodedecode_test/branch_test.go index 4cecf6893c3..97e4e7f8c1a 100644 --- a/lib/trie/encodedecode_test/branch_test.go +++ b/lib/trie/encodedecode_test/branch_test.go @@ -8,6 +8,7 @@ import ( "testing" "github.com/ChainSafe/gossamer/lib/trie/branch" + "github.com/ChainSafe/gossamer/lib/trie/decode" "github.com/ChainSafe/gossamer/lib/trie/leaf" "github.com/ChainSafe/gossamer/lib/trie/node" "github.com/stretchr/testify/assert" @@ -78,7 +79,9 @@ func Test_Branch_Encode_Decode(t *testing.T) { err := testCase.branchToEncode.Encode(buffer) require.NoError(t, err) - const header = 0 + header, err := decode.ReadNextByte(buffer) + require.NoError(t, err) + resultBranch, err := branch.Decode(buffer, header) require.NoError(t, err)