wip rename

internal/functions/functions_test.go

@@ -0,0 +1,95 @@
+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])
+		}
+	}
+}