Implemented suggestion / autocompletion for signal in GAM

test/lsp_completion_signals_test.go

@@ -0,0 +1,128 @@
+package integration
+
+import (
+	"strings"
+	"testing"
+
+	"github.com/marte-community/marte-dev-tools/internal/index"
+	"github.com/marte-community/marte-dev-tools/internal/lsp"
+	"github.com/marte-community/marte-dev-tools/internal/parser"
+	"github.com/marte-community/marte-dev-tools/internal/schema"
+)
+
+func TestSuggestSignalsInGAM(t *testing.T) {
+	// Setup
+	lsp.Tree = index.NewProjectTree()
+	lsp.Documents = make(map[string]string)
+	lsp.ProjectRoot = "."
+	lsp.GlobalSchema = schema.NewSchema()
+
+	// Inject schema for directionality
+	custom := []byte(`
+package schema
+#Classes: {
+    FileReader: { direction: "IN" }
+    FileWriter: { direction: "OUT" }
+}
+`)
+	val := lsp.GlobalSchema.Context.CompileBytes(custom)
+	lsp.GlobalSchema.Value = lsp.GlobalSchema.Value.Unify(val)
+
+	content := `
++InDS = {
+    Class = FileReader
+    +Signals = {
+        InSig = { Type = uint32 }
+    }
+}
++OutDS = {
+    Class = FileWriter
+    +Signals = {
+        OutSig = { Type = uint32 }
+    }
+}
++GAM = {
+    Class = IOGAM
+    +InputSignals = {
+        
+    }
+    +OutputSignals = {
+        
+    }
+}
+`
+	uri := "file://signals.marte"
+	lsp.Documents[uri] = content
+	p := parser.NewParser(content)
+	cfg, err := p.Parse()
+	if err != nil {
+		t.Fatal(err)
+	}
+	lsp.Tree.AddFile("signals.marte", cfg)
+
+	// 1. Suggest in InputSignals
+	// Line 16 (empty line inside InputSignals)
+	paramsIn := lsp.CompletionParams{
+		TextDocument: lsp.TextDocumentIdentifier{URI: uri},
+		Position:     lsp.Position{Line: 16, Character: 8},
+	}
+
+	listIn := lsp.HandleCompletion(paramsIn)
+	if listIn == nil {
+		t.Fatal("Expected suggestions in InputSignals")
+	}
+
+	foundIn := false
+	foundOut := false
+	for _, item := range listIn.Items {
+		if item.Label == "InSig:InDS" {
+			foundIn = true
+			// Normalize spaces for check
+			insert := strings.ReplaceAll(item.InsertText, " ", "")
+			expected := "InSig={DataSource=InDS}"
+			if !strings.Contains(insert, expected) && !strings.Contains(item.InsertText, "InSig = {") {
+				// Snippet might differ slightly, but should contain essentials
+				t.Errorf("InsertText mismatch: %s", item.InsertText)
+			}
+		}
+		if item.Label == "OutSig:OutDS" {
+			foundOut = true
+		}
+	}
+
+	if !foundIn {
+		t.Error("Did not find InSig:InDS")
+	}
+	if foundOut {
+		t.Error("Should not find OutSig:OutDS in InputSignals")
+	}
+
+	// 2. Suggest in OutputSignals
+	// Line 19
+	paramsOut := lsp.CompletionParams{
+		TextDocument: lsp.TextDocumentIdentifier{URI: uri},
+		Position:     lsp.Position{Line: 19, Character: 8},
+	}
+	listOut := lsp.HandleCompletion(paramsOut)
+	if listOut == nil {
+		t.Fatal("Expected suggestions in OutputSignals")
+	}
+
+	foundIn = false
+	foundOut = false
+	for _, item := range listOut.Items {
+		if item.Label == "InSig:InDS" {
+			foundIn = true
+		}
+		if item.Label == "OutSig:OutDS" {
+			foundOut = true
+		}
+	}
+
+	if foundIn {
+		t.Error("Should not find InSig:InDS in OutputSignals")
+	}
+	if !foundOut {
+		t.Error("Did not find OutSig:OutDS in OutputSignals")
+	}
+}