82 } 83 84 protected NodeC(NodeClass<? extends NodeC> c, String name) { 85 super(c, name); 86 } 87 88 } 89 90 @NodeInfo(cycles = CYCLES_IGNORED, size = SIZE_IGNORED) 91 static final class NodeD extends NodeC { 92 public static final NodeClass<NodeD> TYPE = NodeClass.create(NodeD.class); 93 94 protected NodeD(String name) { 95 super(TYPE, name); 96 } 97 98 } 99 100 @Test 101 public void simpleSubclassTest() { 102 Graph graph = new Graph(getOptions()); 103 graph.add(new NodeB("b")); 104 graph.add(new NodeD("d")); 105 106 Assert.assertEquals("bd", TypedNodeIteratorTest.toString(graph.getNodes(NodeB.TYPE))); 107 Assert.assertEquals("d", TypedNodeIteratorTest.toString(graph.getNodes(NodeD.TYPE))); 108 } 109 110 @Test 111 public void addingNodeDuringIterationTest() { 112 Graph graph = new Graph(getOptions()); 113 graph.add(new NodeB("b1")); 114 NodeD d1 = graph.add(new NodeD("d1")); 115 StringBuilder sb = new StringBuilder(); 116 for (NodeB tn : graph.getNodes(NodeB.TYPE)) { 117 if (tn == d1) { 118 graph.add(new NodeB("b2")); 119 } 120 sb.append(tn.getName()); 121 } 122 assertEquals("b1d1b2", sb.toString()); 123 for (NodeB tn : graph.getNodes(NodeB.TYPE)) { 124 if (tn == d1) { 125 graph.add(new NodeB("b3")); 126 } 127 assertNotNull(tn); 128 } 129 assertEquals(4, graph.getNodes(NodeB.TYPE).count()); 130 assertEquals(1, graph.getNodes(NodeD.TYPE).count()); 131 } 132 | 82 } 83 84 protected NodeC(NodeClass<? extends NodeC> c, String name) { 85 super(c, name); 86 } 87 88 } 89 90 @NodeInfo(cycles = CYCLES_IGNORED, size = SIZE_IGNORED) 91 static final class NodeD extends NodeC { 92 public static final NodeClass<NodeD> TYPE = NodeClass.create(NodeD.class); 93 94 protected NodeD(String name) { 95 super(TYPE, name); 96 } 97 98 } 99 100 @Test 101 public void simpleSubclassTest() { 102 Graph graph = new Graph(getOptions(), getDebug()); 103 graph.add(new NodeB("b")); 104 graph.add(new NodeD("d")); 105 106 Assert.assertEquals("bd", TypedNodeIteratorTest.toString(graph.getNodes(NodeB.TYPE))); 107 Assert.assertEquals("d", TypedNodeIteratorTest.toString(graph.getNodes(NodeD.TYPE))); 108 } 109 110 @Test 111 public void addingNodeDuringIterationTest() { 112 Graph graph = new Graph(getOptions(), getDebug()); 113 graph.add(new NodeB("b1")); 114 NodeD d1 = graph.add(new NodeD("d1")); 115 StringBuilder sb = new StringBuilder(); 116 for (NodeB tn : graph.getNodes(NodeB.TYPE)) { 117 if (tn == d1) { 118 graph.add(new NodeB("b2")); 119 } 120 sb.append(tn.getName()); 121 } 122 assertEquals("b1d1b2", sb.toString()); 123 for (NodeB tn : graph.getNodes(NodeB.TYPE)) { 124 if (tn == d1) { 125 graph.add(new NodeB("b3")); 126 } 127 assertNotNull(tn); 128 } 129 assertEquals(4, graph.getNodes(NodeB.TYPE).count()); 130 assertEquals(1, graph.getNodes(NodeD.TYPE).count()); 131 } 132 |