goscript 0.1.0 → 0.1.1

package/compiler/semantic-model_test.go

@@ -2,6 +2,7 @@ package compiler
 
 import (
 	"context"
+	"go/token"
 	"go/types"
 	"path/filepath"
 	"strings"
@@ -281,6 +282,60 @@ func TestSemanticModelColorsAsyncFunctionsAndOverrides(t *testing.T) {
 	}
 }
 
+func TestSemanticModelIndexesFunctionsByFullName(t *testing.T) {
+	model := newSemanticModel()
+	semPkg := &semanticPackage{}
+	pkg := types.NewPackage("example.test/indexed", "indexed")
+	signature := types.NewSignatureType(nil, nil, nil, types.NewTuple(), types.NewTuple(), false)
+	fn := types.NewFunc(token.NoPos, pkg, "Call", signature)
+	semFn := NewSemanticModelOwner().addFunction(model, semPkg, fn, sourcePosition{})
+
+	duplicatePkg := types.NewPackage("example.test/indexed", "indexed")
+	duplicate := types.NewFunc(token.NoPos, duplicatePkg, "Call", signature)
+	if got := semanticFunctionFor(model, duplicate); got != semFn {
+		t.Fatalf("semanticFunctionFor() = %#v, want %#v", got, semFn)
+	}
+}
+
+func TestSemanticModelAsyncPropagationObservesContext(t *testing.T) {
+	ctx, cancel := context.WithCancel(context.Background())
+	cancel()
+
+	diagnostics := NewSemanticModelOwner().propagateFunctionAsync(ctx, newSemanticModel())
+	if !diagnosticsHaveErrors(diagnostics) {
+		t.Fatalf("expected canceled context diagnostic")
+	}
+	if diagnostics[0].Code != "goscript/context:canceled" {
+		t.Fatalf("unexpected diagnostic: %#v", diagnostics[0])
+	}
+}
+
+func TestSemanticModelMarksFunctionIdentifierCallAsync(t *testing.T) {
+	moduleDir := writePackageGraphFixture(t, map[string]string{
+		"go.mod": "module example.test/callback\n\ngo 1.25.3\n",
+		"callback.go": strings.Join([]string{
+			"package callback",
+			"func Run(fn func() error) error {",
+			"  return fn()",
+			"}",
+			"",
+		}, "\n"),
+	})
+	graph := loadPackageGraph(t, &CompileRequest{
+		Patterns:            []string{"."},
+		Dir:                 moduleDir,
+		OutputPath:          filepath.Join(t.TempDir(), "out"),
+		DependencyMode:      DependencyModeRequested,
+		RuntimeEmissionMode: RuntimeEmissionModeEmit,
+	})
+	model := buildSemanticModel(t, graph)
+	run := requireDefinedFunc(t, graph, "example.test/callback", "Run")
+	semFn := model.functions[run]
+	if semFn == nil || !semFn.async {
+		t.Fatalf("expected Run to be async after calling function parameter, got %#v", semFn)
+	}
+}
+
 func buildSemanticModel(t *testing.T, graph *PackageGraph) *SemanticModel {
 	t.Helper()
 

package/compiler/service.go

@@ -19,7 +19,7 @@ func NewCompileService() *CompileService {
 	runtimeOwner := NewRuntimeContractOwner()
 	return &CompileService{
 		requestOwner:  NewCompileRequestOwner(),
-		graphOwner:    NewPackageGraphOwner(),
+		graphOwner:    NewPackageGraphOwner(overrideOwner),
 		semanticOwner: NewSemanticModelOwner(overrideOwner),
 		loweringOwner: NewLoweringOwner(runtimeOwner, overrideOwner),
 		emitterOwner:  NewTypeScriptEmitOwner(runtimeOwner),