mermaid: Use stateDiagram-v2 to beautify state diagram. (#65)

atlas-comstock · web-flow · commit 8729525eaa45 · 2021-01-12T16:39:11.000+01:00
diff --git a/mermaid_visualizer.go b/mermaid_visualizer.go
@@ -34,7 +34,7 @@ func visualizeForMermaidAsStateDiagram(fsm *FSM) string {
 
 	sortedTransitionKeys := getSortedTransitionKeys(fsm.transitions)
 
-	buf.WriteString("stateDiagram\n")
+	buf.WriteString("stateDiagram-v2\n")
 	buf.WriteString(fmt.Sprintln(`    [*] -->`, fsm.current))
 
 	for _, k := range sortedTransitionKeys {
diff --git a/mermaid_visualizer_test.go b/mermaid_visualizer_test.go
@@ -22,7 +22,7 @@ func TestMermaidOutput(t *testing.T) {
 		t.Errorf("got error for visualizing with type MERMAID: %s", err)
 	}
 	wanted := `
-stateDiagram
+stateDiagram-v2
     [*] --> closed
     closed --> open: open
     intermediate --> closed: part-close

Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@ func TestMermaidOutput(t *testing.T) {`
`22`	`22`	`t.Errorf("got error for visualizing with type MERMAID: %s", err)`
`23`	`23`	`}`
`24`	`24`	wanted := `
`25`		`-stateDiagram`
	`25`	`+stateDiagram-v2`
`26`	`26`	`[*] --> closed`
`27`	`27`	`closed --> open: open`
`28`	`28`	`intermediate --> closed: part-close`