96 lines
2.2 KiB
Go
96 lines
2.2 KiB
Go
package functions
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"modernc.org/quickjs"
|
|
)
|
|
|
|
type TestArgs struct {
|
|
Field1 string `json:"field1"`
|
|
}
|
|
|
|
func (t TestArgs) Validate() error {
|
|
return nil
|
|
}
|
|
|
|
func TestAsyncFunction(t *testing.T) {
|
|
RegisterAsyncFunction("testAsync", func(_ context.Context, args TestArgs) (string, error) {
|
|
return "result: " + args.Field1, nil
|
|
})
|
|
|
|
registryMutex.RLock()
|
|
fn, ok := functionRegistry["testAsync"]
|
|
registryMutex.RUnlock()
|
|
|
|
require.True(t, ok, "testAsync should be registered")
|
|
assert.Contains(t, fn.Definition(), "Promise<string>", "definition should include Promise<string>")
|
|
}
|
|
|
|
func TestAsyncFunctionResolution(t *testing.T) {
|
|
RegisterAsyncFunction("resolveTest", func(_ context.Context, args TestArgs) (string, error) {
|
|
return "test-result", nil
|
|
})
|
|
|
|
vm, err := quickjs.NewVM()
|
|
require.NoError(t, err)
|
|
defer func() {
|
|
_ = vm.Close()
|
|
}()
|
|
vm.SetCanBlock(true)
|
|
|
|
RegisterFunctions(context.Background(), vm)
|
|
|
|
result, err := vm.Eval(`resolveTest("hello")`, quickjs.EvalGlobal)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, result)
|
|
}
|
|
|
|
func TestAsyncFunctionRejection(t *testing.T) {
|
|
RegisterAsyncFunction("rejectTest", func(_ context.Context, args TestArgs) (string, error) {
|
|
return "", assert.AnError
|
|
})
|
|
|
|
vm, err := quickjs.NewVM()
|
|
require.NoError(t, err)
|
|
defer func() {
|
|
_ = vm.Close()
|
|
}()
|
|
vm.SetCanBlock(true)
|
|
|
|
RegisterFunctions(context.Background(), vm)
|
|
|
|
result, err := vm.Eval(`rejectTest({field1: "hello"})`, quickjs.EvalGlobal)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, result)
|
|
}
|
|
|
|
func TestNonPromise(t *testing.T) {
|
|
RegisterFunction("nonPromiseTest", func(_ context.Context, args TestArgs) (string, error) {
|
|
return "sync-result", nil
|
|
})
|
|
|
|
vm, err := quickjs.NewVM()
|
|
require.NoError(t, err)
|
|
defer func() {
|
|
_ = vm.Close()
|
|
}()
|
|
vm.SetCanBlock(true)
|
|
|
|
RegisterFunctions(context.Background(), vm)
|
|
|
|
result, err := vm.Eval(`nonPromiseTest({field1: "hello"})`, quickjs.EvalGlobal)
|
|
require.NoError(t, err)
|
|
|
|
if obj, ok := result.(*quickjs.Object); ok {
|
|
var arr []any
|
|
if err := obj.Into(&arr); err == nil && len(arr) > 0 {
|
|
assert.Equal(t, "sync-result", arr[0])
|
|
}
|
|
}
|
|
}
|