From 259c560b357132d0961f036d08e8d8ccd33c4025 Mon Sep 17 00:00:00 2001
From: Dan Kortschak <dan@kortschak.io>
Date: Mon, 16 Jul 2018 08:28:00 +0930
Subject: [PATCH] graph/encoding/dot: add test for chained edge attributes

---
 graph/encoding/dot/decode_test.go | 129 ++++++++++++++++++++++++++++++
 1 file changed, 129 insertions(+)

diff --git a/graph/encoding/dot/decode_test.go b/graph/encoding/dot/decode_test.go
index 13f4002a..dbf0e5a8 100644
--- a/graph/encoding/dot/decode_test.go
+++ b/graph/encoding/dot/decode_test.go
@@ -164,6 +164,135 @@ const undirectedWithPorts = `graph {
 	E:_ -- F:c;
 }`
 
+func TestChainedEdgeAttributes(t *testing.T) {
+	golden := []struct {
+		in, want string
+		directed bool
+	}{
+		{
+			in:       directedChained,
+			want:     directedNonchained,
+			directed: true,
+		},
+		{
+			in:       undirectedChained,
+			want:     undirectedNonchained,
+			directed: false,
+		},
+	}
+	for i, g := range golden {
+		var dst encoding.Builder
+		if g.directed {
+			dst = newDotDirectedGraph()
+		} else {
+			dst = newDotUndirectedGraph()
+		}
+		data := []byte(g.in)
+		if err := Unmarshal(data, dst); err != nil {
+			t.Errorf("i=%d: unable to unmarshal DOT graph; %v", i, err)
+			continue
+		}
+		buf, err := Marshal(dst, "", "", "\t", false)
+		if err != nil {
+			t.Errorf("i=%d: unable to marshal graph; %v", i, dst)
+			continue
+		}
+		got := string(buf)
+		if got != g.want {
+			t.Errorf("i=%d: graph content mismatch; want:\n%s\n\ngot:\n%s", i, g.want, got)
+			continue
+		}
+	}
+}
+
+const directedChained = `digraph {
+	graph [
+		outputorder=edgesfirst
+	];
+	node [
+		shape=circle
+		style=filled
+	];
+	edge [
+		penwidth=5
+		color=gray
+	];
+
+	// Node definitions.
+	A [label="foo 2"];
+	B [label="bar 2"];
+
+	// Edge definitions.
+	A -> B -> A [label="baz 2"];
+}`
+
+const directedNonchained = `digraph {
+	graph [
+		outputorder=edgesfirst
+	];
+	node [
+		shape=circle
+		style=filled
+	];
+	edge [
+		penwidth=5
+		color=gray
+	];
+
+	// Node definitions.
+	A [label="foo 2"];
+	B [label="bar 2"];
+
+	// Edge definitions.
+	A -> B [label="baz 2"];
+	B -> A [label="baz 2"];
+}`
+
+const undirectedChained = `graph {
+	graph [
+		outputorder=edgesfirst
+	];
+	node [
+		shape=circle
+		style=filled
+	];
+	edge [
+		penwidth=5
+		color=gray
+	];
+
+	// Node definitions.
+	A [label="foo 2"];
+	B [label="bar 2"];
+	C [label="bif 2"];
+
+	// Edge definitions.
+	A -- B -- C [label="baz 2"];
+}`
+
+const undirectedNonchained = `graph {
+	graph [
+		outputorder=edgesfirst
+	];
+	node [
+		shape=circle
+		style=filled
+	];
+	edge [
+		penwidth=5
+		color=gray
+	];
+
+	// Node definitions.
+	A [label="foo 2"];
+	B [label="bar 2"];
+	C [label="bif 2"];
+
+	// Edge definitions.
+	A -- B [label="baz 2"];
+	B -- C [label="baz 2"];
+}`
+
 // Below follows a minimal implementation of a graph capable of validating the
 // round-trip encoding and decoding of DOT graphs with nodes and edges
 // containing DOT attributes.