ResiLien
/
terraform
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

digraph: double-quote dot graph elements

This commit is contained in:
Mitchell Hashimoto 2014-07-01 09:54:40 -07:00
parent 29d6a4242e
commit 83795bc54d
2 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
digraph/graphviz.go
View File

@ -11,10 +11,10 @@ func GenerateDot(nodes []Node, w io.Writer) {
w.Write([]byte("digraph {\n")) w.Write([]byte("digraph {\n"))
defer w.Write([]byte("}\n")) defer w.Write([]byte("}\n"))
for _, n := range nodes { for _, n := range nodes {
w.Write([]byte(fmt.Sprintf("\t%s;\n", n))) w.Write([]byte(fmt.Sprintf("\t\"%s\";\n", n)))
for _, edge := range n.Edges() { for _, edge := range n.Edges() {
target := edge.Tail() target := edge.Tail()
line := fmt.Sprintf("\t%s -> %s [label=\"%s\"];\n", line := fmt.Sprintf("\t\"%s\" -> \"%s\" [label=\"%s\"];\n",
n, target, edge) n, target, edge)
w.Write([]byte(line)) w.Write([]byte(line))
} }

18
digraph/graphviz_test.go
View File

@ -27,31 +27,31 @@ b -> e
if !strings.HasSuffix(out, "\n}\n") { if !strings.HasSuffix(out, "\n}\n") {
t.Fatalf("bad: %v", out) t.Fatalf("bad: %v", out)
} }
if !strings.Contains(out, "\n\ta;\n") { if !strings.Contains(out, "\n\t\"a\";\n") {
t.Fatalf("bad: %v", out) t.Fatalf("bad: %v", out)
} }
if !strings.Contains(out, "\n\tb;\n") { if !strings.Contains(out, "\n\t\"b\";\n") {
t.Fatalf("bad: %v", out) t.Fatalf("bad: %v", out)
} }
if !strings.Contains(out, "\n\tc;\n") { if !strings.Contains(out, "\n\t\"c\";\n") {
t.Fatalf("bad: %v", out) t.Fatalf("bad: %v", out)
} }
if !strings.Contains(out, "\n\td;\n") { if !strings.Contains(out, "\n\t\"d\";\n") {
t.Fatalf("bad: %v", out) t.Fatalf("bad: %v", out)
} }
if !strings.Contains(out, "\n\te;\n") { if !strings.Contains(out, "\n\t\"e\";\n") {
t.Fatalf("bad: %v", out) t.Fatalf("bad: %v", out)
} }
if !strings.Contains(out, "\n\ta -> b [label=\"foo\"];\n") { if !strings.Contains(out, "\n\t\"a\" -> \"b\" [label=\"foo\"];\n") {
t.Fatalf("bad: %v", out) t.Fatalf("bad: %v", out)
} }
if !strings.Contains(out, "\n\ta -> c [label=\"Edge\"];\n") { if !strings.Contains(out, "\n\t\"a\" -> \"c\" [label=\"Edge\"];\n") {
t.Fatalf("bad: %v", out) t.Fatalf("bad: %v", out)
} }
if !strings.Contains(out, "\n\tb -> d [label=\"Edge\"];\n") { if !strings.Contains(out, "\n\t\"b\" -> \"d\" [label=\"Edge\"];\n") {
t.Fatalf("bad: %v", out) t.Fatalf("bad: %v", out)
} }
if !strings.Contains(out, "\n\tb -> e [label=\"Edge\"];\n") { if !strings.Contains(out, "\n\t\"b\" -> \"e\" [label=\"Edge\"];\n") {
t.Fatalf("bad: %v", out) t.Fatalf("bad: %v", out)
} }
} }