package mpt import ( "encoding/json" "testing" "github.com/nspcc-dev/neo-go/internal/random" "github.com/nspcc-dev/neo-go/internal/testserdes" "github.com/nspcc-dev/neo-go/pkg/io" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" ) func getTestFuncEncode(ok bool, expected, actual Node) func(t *testing.T) { return func(t *testing.T) { t.Run("IO", func(t *testing.T) { bs, err := testserdes.EncodeBinary(expected) require.NoError(t, err) if hn, ok := actual.(*HashNode); ok { hn.hashValid = true // this field is set during NodeObject decoding } err = testserdes.DecodeBinary(bs, actual) if !ok { require.Error(t, err) return } require.NoError(t, err) require.Equal(t, expected.Type(), actual.Type()) require.Equal(t, expected.Hash(), actual.Hash()) require.Equal(t, 1+expected.Size(), len(expected.Bytes())) }) t.Run("JSON", func(t *testing.T) { bs, err := json.Marshal(expected) require.NoError(t, err) err = json.Unmarshal(bs, actual) if !ok { require.Error(t, err) return } require.NoError(t, err) require.Equal(t, expected.Type(), actual.Type()) require.Equal(t, expected.Hash(), actual.Hash()) }) } } func TestNode_Serializable(t *testing.T) { t.Run("Leaf", func(t *testing.T) { t.Run("Good", func(t *testing.T) { l := NewLeafNode(random.Bytes(123)) t.Run("Raw", getTestFuncEncode(true, l, new(LeafNode))) t.Run("WithType", getTestFuncEncode(true, &NodeObject{l}, new(NodeObject))) }) t.Run("BigValue", getTestFuncEncode(false, NewLeafNode(random.Bytes(MaxValueLength+1)), new(LeafNode))) }) t.Run("Extension", func(t *testing.T) { t.Run("Good", func(t *testing.T) { e := NewExtensionNode(random.Bytes(42), NewLeafNode(random.Bytes(10))) t.Run("Raw", getTestFuncEncode(true, e, new(ExtensionNode))) t.Run("WithType", getTestFuncEncode(true, &NodeObject{e}, new(NodeObject))) }) t.Run("BigKey", getTestFuncEncode(false, NewExtensionNode(random.Bytes(maxPathLength+1), NewLeafNode(random.Bytes(10))), new(ExtensionNode))) }) t.Run("Branch", func(t *testing.T) { b := NewBranchNode() b.Children[0] = NewLeafNode(random.Bytes(10)) b.Children[lastChild] = NewHashNode(random.Uint256()) t.Run("Raw", getTestFuncEncode(true, b, new(BranchNode))) t.Run("WithType", getTestFuncEncode(true, &NodeObject{b}, new(NodeObject))) }) t.Run("Hash", func(t *testing.T) { t.Run("Good", func(t *testing.T) { h := NewHashNode(random.Uint256()) t.Run("Raw", getTestFuncEncode(true, h, new(HashNode))) t.Run("WithType", getTestFuncEncode(true, &NodeObject{h}, new(NodeObject))) }) t.Run("InvalidSize", func(t *testing.T) { buf := io.NewBufBinWriter() buf.BinWriter.WriteBytes(make([]byte, 13)) require.Error(t, testserdes.DecodeBinary(buf.Bytes(), &HashNode{BaseNode: BaseNode{hashValid: true}})) }) }) t.Run("Invalid", func(t *testing.T) { require.Error(t, testserdes.DecodeBinary([]byte{0xFF}, new(NodeObject))) }) } // https://github.com/neo-project/neo/blob/neox-2.x/neo.UnitTests/UT_MPTTrie.cs#L198 func TestJSONSharp(t *testing.T) { tr := NewTrie(nil, false, newTestStore()) require.NoError(t, tr.Put([]byte{0xac, 0x11}, []byte{0xac, 0x11})) require.NoError(t, tr.Put([]byte{0xac, 0x22}, []byte{0xac, 0x22})) require.NoError(t, tr.Put([]byte{0xac}, []byte{0xac})) require.NoError(t, tr.Delete([]byte{0xac, 0x11})) require.NoError(t, tr.Delete([]byte{0xac, 0x22})) js, err := tr.root.MarshalJSON() require.NoError(t, err) require.JSONEq(t, `{"key":"0a0c", "next":{"value":"ac"}}`, string(js)) } func TestInvalidJSON(t *testing.T) { t.Run("InvalidChildrenCount", func(t *testing.T) { var cs [childrenCount + 1]Node for i := range cs { cs[i] = EmptyNode{} } data, err := json.Marshal(cs) require.NoError(t, err) var n NodeObject require.Error(t, json.Unmarshal(data, &n)) }) testCases := []struct { name string data []byte }{ {"WrongFieldCount", []byte(`{"key":"0102", "next": {}, "field": {}}`)}, {"InvalidField1", []byte(`{"next":{}}`)}, {"InvalidField2", []byte(`{"key":"0102", "hash":{}}`)}, {"InvalidKey", []byte(`{"key":"xy", "next":{}}`)}, {"InvalidNext", []byte(`{"key":"01", "next":[]}`)}, {"InvalidHash", []byte(`{"hash":"01"}`)}, {"InvalidValue", []byte(`{"value":1}`)}, {"InvalidBranch", []byte(`[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16]`)}, } for _, tc := range testCases { var n NodeObject assert.Errorf(t, json.Unmarshal(tc.data, &n), "no error in "+tc.name) } } // C# interoperability test // https://github.com/neo-project/neo/blob/neox-2.x/neo.UnitTests/UT_MPTTrie.cs#L135 func TestRootHash(t *testing.T) { b := NewBranchNode() r := NewExtensionNode([]byte{0x0A, 0x0C}, b) v1 := NewLeafNode([]byte{0xAB, 0xCD}) l1 := NewExtensionNode([]byte{0x01}, v1) b.Children[0] = l1 v2 := NewLeafNode([]byte{0x22, 0x22}) l2 := NewExtensionNode([]byte{0x09}, v2) b.Children[9] = l2 r1 := NewExtensionNode([]byte{0x0A, 0x0C, 0x00, 0x01}, v1) require.Equal(t, "cedd9897dd1559fbd5dfe5cfb223464da6de438271028afb8d647e950cbd18e0", r1.Hash().StringLE()) require.Equal(t, "1037e779c8a0313bd0d99c4151fa70a277c43c53a549b6444079f2e67e8ffb7b", r.Hash().StringLE()) }