1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123
|
package dot
import (
"fmt"
"testing"
)
func TestSimpleRecord(t *testing.T) {
g := NewGraph(Directed)
rb := newRecordBuilder(g.Node("r"))
rb.Field("a")
rb.Build()
if got, want := flatten(g.String()), `digraph {n1[label="a",shape="record"];}`; got != want {
t.Errorf("got [%v] want [%v]", got, want)
}
}
func TestSimpleMRecordWithFieldID(t *testing.T) {
g := NewGraph(Directed)
rb := newRecordBuilder(g.Node("r"))
rb.MRecord()
rb.FieldWithId("a", "a1")
rb.Build()
if got, want := flatten(g.String()), `digraph {n1[label="<a1> a",shape="mrecord"];}`; got != want {
t.Errorf("got [%v] want [%v]", got, want)
}
}
func TestTwoColumnsRecord(t *testing.T) {
g := NewGraph(Directed)
rb := newRecordBuilder(g.Node("r"))
rb.Field("a").Field("b")
rb.Build()
if got, want := flatten(g.String()), `digraph {n1[label="a|b",shape="record"];}`; got != want {
t.Errorf("got [%v] want [%v]", got, want)
}
}
func TestTwoColumnsNestedRecord(t *testing.T) {
g := NewGraph(Directed)
rb := newRecordBuilder(g.Node("r"))
rb.Field("a")
rb.Nesting(func() {
rb.Field("b")
rb.Field("c")
})
rb.Field("d")
rb.Build()
if got, want := flatten(g.String()), `digraph {n1[label="a|{b|c}|d",shape="record"];}`; got != want {
t.Errorf("got [%v] want [%v]", got, want)
}
}
func TestStack(t *testing.T) {
one := recordLabel{}
two := recordLabel{}
s := new(stack)
s.push(one)
s.push(two)
if got, want := s.pop(), two; &got == &want {
t.Errorf("got [%v]:%T want [%v]:%T", got, got, want, want)
}
if got, want := s.pop(), one; &got == &want {
t.Errorf("got [%v]:%T want [%v]:%T", got, got, want, want)
}
if got, want := len(*s), 0; got != want {
t.Errorf("got [%v]:%T want [%v]:%T", got, got, want, want)
}
}
// https://graphviz.org/doc/info/shapes.html#record
/*
digraph structs {
node [shape=record];
struct1 [label="<f0> left|<f1> mid\ dle|<f2> right"];
struct2 [label="<f0> one|<f1> two"];
struct3 [label="hello\nworld |{ b |{c|<here> d|e}| f}| g | h"];
struct1:f1 -> struct2:f0;
struct1:f2 -> struct3:here;
}
*/
func ExampleNode_NewRecordBuilder() {
g := NewGraph(Directed)
r1 := g.Node("struct1").NewRecordBuilder()
r1.FieldWithId("left", "f0")
r1.FieldWithId("mid\dle", "f1")
r1.FieldWithId("right", "f2")
r1.Build()
r2 := g.Node("struct2").NewRecordBuilder()
r2.FieldWithId("one", "f0")
r2.Build()
r3 := g.Node("struct3").NewRecordBuilder()
r3.Field("hello\world")
r3.Nesting(func() {
r3.Field("b")
r3.Nesting(func() {
r3.Field("c")
r3.FieldWithId("d", "here")
r3.Field("e")
})
r3.Field("f")
})
r3.Field("g")
r3.Field("h")
r3.Build()
g.EdgeWithPorts(g.Node("struct1"), g.Node("struct2"), "f1", "f0")
g.EdgeWithPorts(g.Node("struct1"), g.Node("struct3"), "f2", "here")
fmt.Println(flatten(g.String()))
// Output:digraph {n1[label="<f0> left|<f1> mid\dle|<f2> right",shape="record"];n2[label="<f0> one",shape="record"];n3[label="hello\world|{b|{c|<here> d|e}|f}|g|h",shape="record"];n1:f1->n2:f0;n1:f2->n3:here;}
}
|