Function TestLinkerFuncs

linker_test.go:126–183

(t *testing.T)

}

func TestLinkerFuncs(t *testing.T) {
	engine := NewEngine()
	linker := NewLinker(engine)
	var called int
	var callers []string
	err := linker.FuncWrap("foo", "bar", func(c *Caller) {
		called += 1

		data := c.Data()
		if data != nil {
			callers = append(callers, data.(string))
		}
	})
	require.NoError(t, err)

	wasm, err := Wat2Wasm(`
	    (module
		(import "foo" "bar" (func))
		(start 0)
	    )
	`)
	require.NoError(t, err)

	module, err := NewModule(engine, wasm)
	require.NoError(t, err)

	_, err = linker.Instantiate(NewStoreWithData(engine, "one"), module)
	require.NoError(t, err)
	require.Equal(t, 1, called, "expected a call")
	require.Equal(t, []string{"one"}, callers)

	_, err = linker.Instantiate(NewStore(engine), module)
	require.NoError(t, err)
	require.Equal(t, 2, called, "expected a call")

	cb := func(caller *Caller, args []Val) ([]Val, *Trap) {
		called += 2
		data := caller.Data()
		if data != nil {
			callers = append(callers, data.(string))
		}

		return []Val{}, nil
	}
	ty := NewFuncType([]*ValType{}, []*ValType{})
	linker.AllowShadowing(true)
	err = linker.FuncNew("foo", "bar", ty, cb)
	require.NoError(t, err)

	_, err = linker.Instantiate(NewStoreWithData(engine, "two"), module)
	require.NoError(t, err)
	require.Equal(t, 4, called, "expected a call")
	require.Equal(t, []string{"one", "two"}, callers)

	_, err = linker.Instantiate(NewStore(engine), module)
	require.NoError(t, err)
	require.Equal(t, 6, called, "expected a call")
}