1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 package org.grouplens.grapht.graph;
21
22 import org.apache.commons.lang3.tuple.Pair;
23 import org.junit.Test;
24
25 import static org.hamcrest.Matchers.equalTo;
26 import static org.hamcrest.Matchers.hasSize;
27 import static org.hamcrest.Matchers.nullValue;
28 import static org.hamcrest.Matchers.contains;
29 import static org.junit.Assert.assertThat;
30
31
32
33
34 public class DAGNodeBuilderTest {
35 @Test
36 public void testGetLabel() {
37 DAGNodeBuilder<String,String> bld = DAGNode.newBuilder();
38 assertThat(bld.getLabel(), nullValue());
39 bld.setLabel("foo");
40 assertThat(bld.getLabel(), equalTo("foo"));
41 }
42
43 @Test
44 public void testGetEdges() {
45 DAGNodeBuilder<String,String> bld = DAGNode.newBuilder("foo");
46 assertThat(bld.getEdges(), hasSize(0));
47 DAGNode<String,String> nbr = DAGNode.singleton("bar");
48 bld.addEdge(nbr, "piper");
49 assertThat(bld.getEdges(),
50 contains(Pair.of(nbr, "piper")));
51 DAGNode<String,String> node = bld.build();
52 assertThat(node.getLabel(), equalTo("foo"));
53 assertThat(node.getAdjacentNodes(), contains(nbr));
54 }
55 }