Fix encode decode tests

ChainSafe · Nov 29, 2021 · 7069bd3 · 7069bd3
1 parent 8ffb440
commit 7069bd3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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)