Mercurial > hg > truffle
view graal/com.oracle.graal.graph.test/src/com/oracle/graal/graph/test/TypedNodeIteratorTest2.java @ 19526:8fc336a04d77
Create TYPE fields for LIRInstruction and CompositeValue. Renaming NodeClass#get to NodeClass#create.
author | Thomas Wuerthinger <thomas.wuerthinger@oracle.com> |
---|---|
date | Fri, 20 Feb 2015 22:22:55 +0100 |
parents | 880717e44675 |
children |
line wrap: on
line source
/* * Copyright (c) 2013, 2015, Oracle and/or its affiliates. All rights reserved. * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. * * This code is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License version 2 only, as * published by the Free Software Foundation. * * This code is distributed in the hope that it will be useful, but WITHOUT * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License * version 2 for more details (a copy is included in the LICENSE file that * accompanied this code). * * You should have received a copy of the GNU General Public License version * 2 along with this work; if not, write to the Free Software Foundation, * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. * * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA * or visit www.oracle.com if you need additional information or have any * questions. */ package com.oracle.graal.graph.test; import static org.junit.Assert.*; import org.junit.*; import com.oracle.graal.graph.*; import com.oracle.graal.nodeinfo.*; public class TypedNodeIteratorTest2 { @NodeInfo static class NodeA extends Node implements TestNodeInterface { public static final NodeClass<NodeA> TYPE = NodeClass.create(NodeA.class); protected final String name; public NodeA(String name) { this(TYPE, name); } protected NodeA(NodeClass<? extends NodeA> c, String name) { super(c); this.name = name; } public String getName() { return name; } } @NodeInfo static class NodeB extends NodeA implements IterableNodeType { public static final NodeClass<NodeB> TYPE = NodeClass.create(NodeB.class); public NodeB(String name) { this(TYPE, name); } protected NodeB(NodeClass<? extends NodeB> c, String name) { super(c, name); } } @NodeInfo static class NodeC extends NodeB { public static final NodeClass<NodeC> TYPE = NodeClass.create(NodeC.class); public NodeC(String name) { this(TYPE, name); } protected NodeC(NodeClass<? extends NodeC> c, String name) { super(c, name); } } @NodeInfo static final class NodeD extends NodeC { public static final NodeClass<NodeD> TYPE = NodeClass.create(NodeD.class); public NodeD(String name) { super(TYPE, name); } } @Test public void simpleSubclassTest() { Graph graph = new Graph(); graph.add(new NodeB("b")); graph.add(new NodeD("d")); Assert.assertEquals("bd", TypedNodeIteratorTest.toString(graph.getNodes(NodeB.TYPE))); Assert.assertEquals("d", TypedNodeIteratorTest.toString(graph.getNodes(NodeD.TYPE))); } @Test public void addingNodeDuringIterationTest() { Graph graph = new Graph(); graph.add(new NodeB("b1")); NodeD d1 = graph.add(new NodeD("d1")); StringBuilder sb = new StringBuilder(); for (NodeB tn : graph.getNodes(NodeB.TYPE)) { if (tn == d1) { graph.add(new NodeB("b2")); } sb.append(tn.getName()); } assertEquals("b1d1b2", sb.toString()); for (NodeB tn : graph.getNodes(NodeB.TYPE)) { if (tn == d1) { graph.add(new NodeB("b3")); } assertNotNull(tn); } assertEquals(4, graph.getNodes(NodeB.TYPE).count()); assertEquals(1, graph.getNodes(NodeD.TYPE).count()); } }