Mercurial > hg > graal-compiler
diff graal/com.oracle.truffle.api.test/src/com/oracle/truffle/api/test/tools/NodeExecCounterTest.java @ 18988:cca166aa28c0
Truffle/Tools: unit tests for new framework and tool instances
author | Michael Van De Vanter <michael.van.de.vanter@oracle.com> |
---|---|
date | Tue, 27 Jan 2015 20:27:25 -0800 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/graal/com.oracle.truffle.api.test/src/com/oracle/truffle/api/test/tools/NodeExecCounterTest.java Tue Jan 27 20:27:25 2015 -0800 @@ -0,0 +1,165 @@ +/* + * Copyright (c) 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.truffle.api.test.tools; + +import static com.oracle.truffle.api.test.tools.TestNodes.*; +import static org.junit.Assert.*; + +import org.junit.*; + +import com.oracle.truffle.api.*; +import com.oracle.truffle.api.instrument.*; +import com.oracle.truffle.api.nodes.*; +import com.oracle.truffle.api.test.tools.TestNodes.TestAddNode; +import com.oracle.truffle.api.test.tools.TestNodes.TestValueNode; +import com.oracle.truffle.api.tools.*; +import com.oracle.truffle.api.tools.NodeExecCounter.NodeExecutionCount; + +public class NodeExecCounterTest { + + @Test + public void testNoExecution() { + final NodeExecCounter tool = new NodeExecCounter(); + assertEquals(tool.getCounts().length, 0); + tool.install(); + assertEquals(tool.getCounts().length, 0); + tool.setEnabled(false); + assertEquals(tool.getCounts().length, 0); + tool.setEnabled(true); + assertEquals(tool.getCounts().length, 0); + tool.reset(); + assertEquals(tool.getCounts().length, 0); + tool.dispose(); + assertEquals(tool.getCounts().length, 0); + } + + @Test + public void testToolCreatedTooLate() { + final CallTarget expr13callTarget = createExpr13TestCallTarget(); + final NodeExecCounter tool = new NodeExecCounter(); + tool.install(); + assertEquals(13, expr13callTarget.call()); + assertEquals(tool.getCounts().length, 0); + tool.dispose(); + } + + @Test + public void testToolInstalledcTooLate() { + final NodeExecCounter tool = new NodeExecCounter(); + final CallTarget expr13callTarget = createExpr13TestCallTarget(); + tool.install(); + assertEquals(13, expr13callTarget.call()); + assertEquals(tool.getCounts().length, 0); + tool.dispose(); + } + + @Test + public void testCountingAll() { + final NodeExecCounter tool = new NodeExecCounter(); + tool.install(); + final CallTarget expr13callTarget = createExpr13TestCallTarget(); + + // execute once + assertEquals(13, expr13callTarget.call()); + final NodeExecutionCount[] count1 = tool.getCounts(); + assertNotNull(count1); + assertEquals(count1.length, 2); + for (NodeExecutionCount count : count1) { + final Class<?> class1 = count.nodeClass(); + final long executionCount = count.executionCount(); + if (class1 == TestAddNode.class) { + assertEquals(executionCount, 1); + } else if (class1 == TestValueNode.class) { + assertEquals(executionCount, 2); + } else { + fail(); + } + } + + // Execute 99 more times + for (int i = 0; i < 99; i++) { + assertEquals(13, expr13callTarget.call()); + } + final NodeExecutionCount[] counts100 = tool.getCounts(); + assertNotNull(counts100); + assertEquals(counts100.length, 2); + for (NodeExecutionCount count : counts100) { + final Class<?> class1 = count.nodeClass(); + final long executionCount = count.executionCount(); + if (class1 == TestAddNode.class) { + assertEquals(executionCount, 100); + } else if (class1 == TestValueNode.class) { + assertEquals(executionCount, 200); + } else { + fail(); + } + } + + tool.dispose(); + } + + @Test + public void testCountingTagged() { + final NodeExecCounter tool = new NodeExecCounter(StandardSyntaxTag.STATEMENT); + tool.install(); + final RootNode expr13rootNode = createExpr13TestRootNode(); + + // Not probed yet. + assertEquals(13, expr13rootNode.execute(null)); + assertEquals(tool.getCounts().length, 0); + + final Node addNode = expr13rootNode.getChildren().iterator().next(); + final Probe probe = addNode.probe(); + + // Probed but not tagged yet. + assertEquals(13, expr13rootNode.execute(null)); + assertEquals(tool.getCounts().length, 0); + + probe.tagAs(StandardSyntaxTag.STATEMENT, "fake statement for testing"); + + // Counting now; execute once + assertEquals(13, expr13rootNode.execute(null)); + final NodeExecutionCount[] counts1 = tool.getCounts(); + assertNotNull(counts1); + assertEquals(counts1.length, 1); + final NodeExecutionCount count1 = counts1[0]; + assertNotNull(count1); + assertEquals(count1.nodeClass(), addNode.getClass()); + assertEquals(count1.executionCount(), 1); + + // Execute 99 more times + for (int i = 0; i < 99; i++) { + assertEquals(13, expr13rootNode.execute(null)); + } + + final NodeExecutionCount[] counts100 = tool.getCounts(); + assertNotNull(counts100); + assertEquals(counts100.length, 1); + final NodeExecutionCount count100 = counts100[0]; + assertNotNull(count100); + assertEquals(count100.nodeClass(), addNode.getClass()); + assertEquals(count100.executionCount(), 100); + + tool.dispose(); + } +}