diff --git a/link_test.go b/link_test.go index 8c77221..6567555 100644 --- 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() +}