73 lines
1.5 KiB
Go
73 lines
1.5 KiB
Go
package integration
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/marte-community/marte-dev-tools/internal/builder"
|
|
"github.com/marte-community/marte-dev-tools/internal/parser"
|
|
)
|
|
|
|
func TestVariables(t *testing.T) {
|
|
content := `
|
|
#var MyInt: int = 10
|
|
#var MyStr: string = "default"
|
|
|
|
+Obj = {
|
|
Class = Test
|
|
Field1 = @MyInt
|
|
Field2 = @MyStr
|
|
}
|
|
`
|
|
// Test Parsing
|
|
p := parser.NewParser(content)
|
|
cfg, err := p.Parse()
|
|
if err != nil {
|
|
t.Fatalf("Parse failed: %v", err)
|
|
}
|
|
|
|
// Check definitions: #var, #var, +Obj
|
|
if len(cfg.Definitions) != 3 {
|
|
t.Errorf("Expected 3 definitions, got %d", len(cfg.Definitions))
|
|
}
|
|
|
|
// Test Builder resolution
|
|
f, _ := os.CreateTemp("", "vars.marte")
|
|
f.WriteString(content)
|
|
f.Close()
|
|
defer os.Remove(f.Name())
|
|
|
|
// Build with override
|
|
overrides := map[string]string{
|
|
"MyInt": "999",
|
|
}
|
|
|
|
b := builder.NewBuilder([]string{f.Name()}, overrides)
|
|
|
|
outF, _ := os.CreateTemp("", "out.marte")
|
|
outName := outF.Name()
|
|
defer os.Remove(outName)
|
|
|
|
err = b.Build(outF)
|
|
outF.Close()
|
|
|
|
if err != nil {
|
|
t.Fatalf("Build failed: %v", err)
|
|
}
|
|
|
|
outContent, _ := os.ReadFile(outName)
|
|
outStr := string(outContent)
|
|
|
|
if !strings.Contains(outStr, "Field1 = 999") {
|
|
t.Errorf("Variable override failed for MyInt. Got:\n%s", outStr)
|
|
}
|
|
if !strings.Contains(outStr, "Field2 = \"default\"") {
|
|
t.Errorf("Default value failed for MyStr. Got:\n%s", outStr)
|
|
}
|
|
// Check #var is removed
|
|
if strings.Contains(outStr, "#var") {
|
|
t.Error("#var definition present in output")
|
|
}
|
|
}
|