129 lines
2.9 KiB
Go
129 lines
2.9 KiB
Go
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")
|
|
}
|
|
}
|