Initial support to variables and to producer/consumer logic

test/variables_test.go

@@ -0,0 +1,72 @@
+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")
+	}
+}