Test SequenceDiagram.Link failures

gregoryv · Aug 19, 2019 · 30d23fd · 30d23fd
1 parent d2f3d0e
commit 30d23fd
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/link_test.go b/link_test.go
@@ -20,3 +20,17 @@ func TestLink_class(t *testing.T) {
 		assert().Equals(got, c.exp)
 	}
 }
+
+func TestSequenceDiagram_Link_missing_from_column(t *testing.T) {
+	dia := &SequenceDiagram{}
+	defer func() { recover() }()
+	dia.Link("a", "b", "..")
+	t.Fail()
+}
+
+func TestSequenceDiagram_Link_missing_to_column(t *testing.T) {
+	dia := &SequenceDiagram{columns: []string{"a"}}
+	defer func() { recover() }()
+	dia.Link("a", "b", "..")
+	t.Fail()
+}