Function TestAsyncBinding

common/decls/decls_test.go:1375–1420

(t *testing.T)

}

func TestAsyncBinding(t *testing.T) {
	fn, err := NewFunction("async_fn",
		Overload("async_fn_int", []*types.Type{types.IntType}, types.IntType,
			AsyncBinding(func(ctx context.Context, args ...ref.Val) ref.Val {
				return args[0]
			}),
		),
	)
	if err != nil {
		t.Fatalf("NewFunction() failed: %v", err)
	}
	for _, od := range fn.OverloadDecls() {
		if !od.HasBinding() {
			t.Errorf("Overload %s does not have binding, wanted async binding", od.ID())
		}
		if !od.HasLateBinding() {
			t.Errorf("Overload %s does not have late binding, wanted async binding to be late bound", od.ID())
		}
	}
	if !fn.HasLateBinding() {
		t.Errorf("Function %s does not have late binding, wanted async function to be late bound", fn.Name())
	}
	bindings, err := fn.Bindings()
	if err != nil {
		t.Fatalf("fn.Bindings() produced an err: %v", err)
	}
	if len(bindings) != 2 {
		t.Errorf("fn.Bindings() produced %d bindings, wanted 2", len(bindings))
	}
	for _, binding := range bindings {
		if binding.Async == nil {
			t.Fatal("binding missing Async implementation")
		}

		ctx := context.Background()
		ch := binding.Async(ctx, types.Int(42))
		select {
		case res := <-ch:
			if res.Equal(types.Int(42)) != types.True {
				t.Errorf("async binding returned %v, wanted 42", res)
			}
		case <-time.After(1 * time.Second):
			t.Fatal("async binding timed out")
		}
	}
}

func TestAsyncBindingTypeGuards(t *testing.T) {
	fn, err := NewFunction("async_fn",