Add test for renderable node merging.

render/renderable_node_test.go

@@ -0,0 +1,67 @@
+package render_test
+
+import (
+	"reflect"
+	"testing"
+
+	"github.com/weaveworks/scope/render"
+	"github.com/weaveworks/scope/report"
+)
+
+func TestMergeRenderableNodes(t *testing.T) {
+	nodes1 := render.RenderableNodes{
+		"foo": render.RenderableNode{ID: "foo"},
+		"bar": render.RenderableNode{ID: "bar"},
+	}
+	nodes2 := render.RenderableNodes{
+		"bar": render.RenderableNode{ID: "bar"},
+		"baz": render.RenderableNode{ID: "baz"},
+	}
+
+	want := render.RenderableNodes{
+		"foo": render.RenderableNode{ID: "foo"},
+		"bar": render.RenderableNode{ID: "bar"},
+		"baz": render.RenderableNode{ID: "baz"},
+	}
+	nodes1.Merge(nodes2)
+
+	if !reflect.DeepEqual(want, nodes1) {
+		t.Errorf("want %+v, have %+v", want, nodes1)
+	}
+}
+
+func TestMergeRenderableNode(t *testing.T) {
+	node1 := render.RenderableNode{
+		ID:         "foo",
+		LabelMajor: "",
+		LabelMinor: "minor",
+		Rank:       "",
+		Pseudo:     false,
+		Adjacency:  report.MakeIDList("a1"),
+		Origins:    report.MakeIDList("o1"),
+	}
+	node2 := render.RenderableNode{
+		ID:         "foo",
+		LabelMajor: "major",
+		LabelMinor: "",
+		Rank:       "rank",
+		Pseudo:     false,
+		Adjacency:  report.MakeIDList("a2"),
+		Origins:    report.MakeIDList("o2"),
+	}
+
+	want := render.RenderableNode{
+		ID:         "foo",
+		LabelMajor: "major",
+		LabelMinor: "minor",
+		Rank:       "rank",
+		Pseudo:     false,
+		Adjacency:  report.MakeIDList("a1", "a2"),
+		Origins:    report.MakeIDList("o1", "o2"),
+	}
+	node1.Merge(node2)
+
+	if !reflect.DeepEqual(want, node1) {
+		t.Errorf("want %+v, have %+v", want, node1)
+	}
+}