feat: Add start node for PUML format

Signed-off-by: Sergiy Kulanov <[email protected]>
SergK · Oct 8, 2023 · fe6ed13 · fe6ed13
1 parent 74e7618
commit fe6ed13
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/pkg/taskgraph/taskgraph.go b/pkg/taskgraph/taskgraph.go
@@ -132,6 +132,9 @@ func (g *TaskGraph) ToPlantUML() string {
 		// Replace dashes with underscores in node names because PlantUML doesn't like dashes
 		nodeName := strings.ReplaceAll(node.Name, "-", "_")
 		// the root node is the one with no dependencies and that task starts the execution immediately
+		if !node.hasParent {
+			plantuml += fmt.Sprintf("[*] --> %s\n", nodeName)
+		}
 		if len(node.Dependencies) == 0 {
 			plantuml += fmt.Sprintf("%s --> [*]\n", nodeName)
 		}
@@ -156,6 +159,9 @@ func (g *TaskGraph) ToPlantUMLWithTaskRef() string {
 		// Replace dashes with underscores in node names because PlantUML doesn't like dashes
 		nodeName := strings.ReplaceAll(node.Name, "-", "_")
 		// the root node is the one with no dependencies and that task starts the execution immediately
+		if !node.hasParent {
+			plantuml += fmt.Sprintf("[*] --> %s\n", nodeName)
+		}
 		if len(node.Dependencies) == 0 {
 			plantuml += fmt.Sprintf("%s --> [*]\n", nodeName)
 		}

diff --git a/pkg/taskgraph/taskgraph_test.go b/pkg/taskgraph/taskgraph_test.go
@@ -139,6 +139,8 @@ func TestTaskGraphToPlantUML(t *testing.T) {
 	// Test the ToPlantUML method
 	plantuml := graph.ToPlantUML()
 	assert.Contains(t, plantuml, "@startuml\nhide empty description\ntitle test-pipeline\n\n")
+	assert.Contains(t, plantuml, "[*] --> task3\n")
+	assert.Contains(t, plantuml, "[*] --> task4\n")
 	assert.Contains(t, plantuml, "task1 --> [*]\n")
 	assert.Contains(t, plantuml, "task4 --> [*]\n")
 	assert.Contains(t, plantuml, "task2 -down-> task1\n")
@@ -157,6 +159,8 @@ func TestTaskGraphToPlantUMLWithTaskRef(t *testing.T) {
 	assert.Contains(t, plantuml, "@startuml\nhide empty description\ntitle test-pipeline\n\n")
 	assert.Contains(t, plantuml, "task1 --> [*]")
 	assert.Contains(t, plantuml, "task4 --> [*]")
+	assert.Contains(t, plantuml, "[*] --> task3\n")
+	assert.Contains(t, plantuml, "[*] --> task4\n")
 	assert.Contains(t, plantuml, "task2 -down-> task1")
 	assert.Contains(t, plantuml, "task3 -down-> task1")
 	assert.Contains(t, plantuml, "task3 -down-> task2")