mirror of
https://github.com/ii64/sonic.git
synced 2026-06-21 00:46:43 +08:00
76 lines
No EOL
2.4 KiB
Go
76 lines
No EOL
2.4 KiB
Go
// +build go1.18
|
|
|
|
/*
|
|
* Copyright 2021 ByteDance Inc.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
package sonic_fuzz
|
|
|
|
import (
|
|
`testing`
|
|
`github.com/bytedance/sonic`
|
|
`github.com/bytedance/sonic/ast`
|
|
`github.com/stretchr/testify/require`
|
|
)
|
|
|
|
func fuzzASTGetFromObject(t *testing.T, data []byte, m map[string]interface{}) {
|
|
for k, expv := range(m) {
|
|
node, err := sonic.Get(data, k)
|
|
require.NoErrorf(t, err, "error in ast get key -> %s", k)
|
|
assertAstNode(t, node, expv)
|
|
}
|
|
}
|
|
|
|
func fuzzASTGetFromArray(t *testing.T, data []byte, a []interface{}) {
|
|
var i = 0
|
|
for ; i < len(a); i++ {
|
|
node, err := sonic.Get(data, i)
|
|
require.NoErrorf(t, err, "error in ast get index -> %s", i)
|
|
assertAstNode(t, node, a[i])
|
|
}
|
|
_, err := sonic.Get(data, i)
|
|
require.Errorf(t, err, "error in ast get index -> %s", i)
|
|
}
|
|
|
|
func assertAstNode(t *testing.T, node ast.Node, expv interface{}) {
|
|
switch node.Type() {
|
|
case ast.V_NULL: require.Nilf(t, expv, "wrong in ast null")
|
|
case ast.V_TRUE: fallthrough
|
|
case ast.V_FALSE:
|
|
gotv, err := node.Bool()
|
|
require.NoErrorf(t, err, "error in ast get bool")
|
|
require.Equalf(t, gotv, expv, "wrong in get bool")
|
|
case ast.V_STRING:
|
|
gotv, err := node.String()
|
|
require.NoErrorf(t, err, "error in ast get string")
|
|
require.Equalf(t, gotv, expv, "wrong in get string")
|
|
case ast.V_ARRAY:
|
|
gotv, err := node.Array()
|
|
require.NoErrorf(t, err, "error in ast get array")
|
|
require.Equalf(t, gotv, expv, "wrong in get array")
|
|
case ast.V_OBJECT:
|
|
gotv, err := node.Map()
|
|
require.NoErrorf(t, err, "error in ast get object")
|
|
require.Equalf(t, gotv, expv, "wrong in get object")
|
|
case ast.V_NUMBER:
|
|
gotv, err := node.Float64()
|
|
require.NoErrorf(t, err, "error in ast get number")
|
|
require.Equalf(t, gotv, expv, "wrong in get number")
|
|
case ast.V_ANY:
|
|
gotv, err := node.Interface()
|
|
require.NoErrorf(t, err, "error in ast get any")
|
|
require.Equalf(t, gotv, expv, "wrong in get any")
|
|
}
|
|
} |