Merge pull request #42 from qmuntal/graph-tests

graphviz: quote guard functions appropriately

graph.go

@@ -204,7 +204,7 @@ func (g *graph) formatOneTransition(sm *StateMachine, source, destination State,
 		if sb.Len() > 0 {
 			sb.WriteString(" ")
 		}
-		sb.WriteString(fmt.Sprintf("[%s]", esc(info.Description.String(), true)))
+		sb.WriteString(fmt.Sprintf("[%s]", esc(info.Description.String(), false)))
 	}
 	return g.formatOneLine(str(source, true), str(destination, true), sb.String(), lhead, ltail)
 }

graph_test.go

@@ -2,6 +2,7 @@ package stateless_test
 
 import (
 	"bytes"
+	"context"
 	"flag"
 	"os"
 	"reflect"
@@ -40,11 +41,28 @@ func withInitialState() *stateless.StateMachine {
 	return sm
 }
 
+func withGuards() *stateless.StateMachine {
+	sm := stateless.NewStateMachine("B")
+	sm.SetTriggerParameters("X", reflect.TypeOf(0))
+	sm.Configure("A").
+		Permit("X", "D", func(_ context.Context, args ...interface{}) bool {
+			return args[0].(int) == 3
+		})
+
+	sm.Configure("B").
+		SubstateOf("A").
+		Permit("X", "C", func(_ context.Context, args ...interface{}) bool {
+			return args[0].(int) == 2
+		})
+	return sm
+}
+
 func TestStateMachine_ToGraph(t *testing.T) {
 	tests := []func() *stateless.StateMachine{
 		emptyWithInitial,
 		withSubstate,
 		withInitialState,
+		withGuards,
 	}
 	for _, fn := range tests {
 		name := runtime.FuncForPC(reflect.ValueOf(fn).Pointer()).Name()

testdata/golden/withGuards.dot

@@ -0,0 +1,14 @@
+digraph {
+	compound=true;
+	node [shape=Mrecord];
+	rankdir="LR";
+
+	subgraph cluster_A {
+		label="A";
+		B [label="B"];
+	}
+	B -> D [label="X [func1]", ltail="cluster_A"];
+	B -> C [label="X [func2]"];
+	init [label="", shape=point];
+	init -> B
+}