22945a3ff4
This centralizes our management of external types by choosing the word "external" also used in the spec instead of defining the same constants for imports vs exports. One advantage is more coherent reference searching as constants are no longer split. Another is future work can use a single type to manage namespace length checks. Signed-off-by: Adrian Cole <adrian@tetrate.io>
122 lines
2.9 KiB
Go
122 lines
2.9 KiB
Go
package binary
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
wasm "github.com/tetratelabs/wazero/internal/wasm"
|
|
)
|
|
|
|
func TestMemorySection(t *testing.T) {
|
|
three := uint32(3)
|
|
tests := []struct {
|
|
name string
|
|
input []byte
|
|
expected []*wasm.MemoryType
|
|
}{
|
|
{
|
|
name: "min and min with max",
|
|
input: []byte{
|
|
0x02, // 2 memories
|
|
0x00, 1, // (memory 1)
|
|
0x01, 2, 3, // (memory 2, 3)
|
|
},
|
|
expected: []*wasm.MemoryType{{Min: 1}, {Min: 2, Max: &three}},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
tc := tt
|
|
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
memories, err := decodeMemorySection(bytes.NewReader(tc.input))
|
|
require.NoError(t, err)
|
|
require.Equal(t, tc.expected, memories)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestDecodeExportSection(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
input []byte
|
|
expected map[string]*wasm.Export
|
|
}{
|
|
{
|
|
name: "empty and non-empty name",
|
|
input: []byte{
|
|
0x02, // 2 exports
|
|
0x00, // Size of empty name
|
|
wasm.ExternTypeFunc, 0x02, // func[2]
|
|
0x01, 'a', // Size of name, name
|
|
wasm.ExternTypeFunc, 0x01, // func[1]
|
|
},
|
|
expected: map[string]*wasm.Export{
|
|
"": {Name: "", Type: wasm.ExternTypeFunc, Index: wasm.Index(2)},
|
|
"a": {Name: "a", Type: wasm.ExternTypeFunc, Index: wasm.Index(1)},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
tc := tt
|
|
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
exports, err := decodeExportSection(bytes.NewReader(tc.input))
|
|
require.NoError(t, err)
|
|
require.Equal(t, tc.expected, exports)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestDecodeExportSection_Errors(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
input []byte
|
|
expectedErr string
|
|
}{
|
|
{
|
|
name: "duplicates empty name",
|
|
input: []byte{
|
|
0x02, // 2 exports
|
|
0x00, // Size of empty name
|
|
wasm.ExternTypeFunc, 0x00, // func[0]
|
|
0x00, // Size of empty name
|
|
wasm.ExternTypeFunc, 0x00, // func[0]
|
|
},
|
|
expectedErr: "export[1] duplicates name \"\"",
|
|
},
|
|
{
|
|
name: "duplicates name",
|
|
input: []byte{
|
|
0x02, // 2 exports
|
|
0x01, 'a', // Size of name, name
|
|
wasm.ExternTypeFunc, 0x00, // func[0]
|
|
0x01, 'a', // Size of name, name
|
|
wasm.ExternTypeFunc, 0x00, // func[0]
|
|
},
|
|
expectedErr: "export[1] duplicates name \"a\"",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
tc := tt
|
|
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
_, err := decodeExportSection(bytes.NewReader(tc.input))
|
|
require.EqualError(t, err, tc.expectedErr)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestEncodeFunctionSection(t *testing.T) {
|
|
require.Equal(t, []byte{wasm.SectionIDFunction, 0x2, 0x01, 0x05}, encodeFunctionSection([]wasm.Index{5}))
|
|
}
|
|
|
|
// TestEncodeStartSection uses the same index as TestEncodeFunctionSection to highlight the encoding is different.
|
|
func TestEncodeStartSection(t *testing.T) {
|
|
require.Equal(t, []byte{wasm.SectionIDStart, 0x01, 0x05}, encodeStartSection(5))
|
|
}
|