status-go/abi-spec/types_test.go

65 lines
1.7 KiB
Go
Raw Normal View History

2022-08-24 12:42:41 +00:00
package abispec
import (
"encoding/json"
"testing"
"github.com/stretchr/testify/require"
)
func TestToGoTypeValue(t *testing.T) {
var raw json.RawMessage
err := json.Unmarshal([]byte(`"dave"`), &raw)
require.NoError(t, err)
val, err := toGoTypeValue("bytes", raw)
require.NoError(t, err)
require.Equal(t, []byte("dave"), val.Elem().Bytes())
err = json.Unmarshal([]byte(`true`), &raw)
require.NoError(t, err)
val, err = toGoTypeValue("bool", raw)
require.NoError(t, err)
require.True(t, val.Elem().Bool())
}
func TestToGoType(t *testing.T) {
var raws []json.RawMessage
err := json.Unmarshal([]byte("[8]"), &raws)
require.NoError(t, err)
value, err := toGoTypeValue("uint8", raws[0])
require.NoError(t, err)
require.Equal(t, uint8(8), *value.Interface().(*uint8))
goType, err := toGoType("uint256[][3][]")
require.NoError(t, err)
require.Equal(t, "[][3][]*big.Int", goType.String())
goType, err = toGoType("uint256[][][3]")
require.NoError(t, err)
require.Equal(t, "[3][][]*big.Int", goType.String())
goType, err = toGoType("uint256[3][][]")
require.NoError(t, err)
require.Equal(t, "[][][3]*big.Int", goType.String())
goType, err = toGoType("bytes3[2]")
require.NoError(t, err)
require.Equal(t, "[2][3]uint8", goType.String())
}
func TestArrayTypePattern(t *testing.T) {
require.True(t, arrayTypePattern.MatchString(`uint8[]`))
require.False(t, arrayTypePattern.MatchString(`uint8`))
s := "uint8[][2][1][]"
matches := arrayTypePattern.FindAllStringSubmatch(s, -1)
require.Equal(t, 3, len(matches[0]))
require.Equal(t, "", matches[0][2])
require.Equal(t, "2", matches[1][2])
index := arrayTypePattern.FindStringIndex(s)[0]
require.Equal(t, 5, index)
require.Equal(t, "uint8", s[0:index])
}