001/*
002 * Copyright (c) 2014, Oracle and/or its affiliates. All rights reserved.
003 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
004 *
005 * This code is free software; you can redistribute it and/or modify it
006 * under the terms of the GNU General Public License version 2 only, as
007 * published by the Free Software Foundation.
008 *
009 * This code is distributed in the hope that it will be useful, but WITHOUT
010 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
011 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
012 * version 2 for more details (a copy is included in the LICENSE file that
013 * accompanied this code).
014 *
015 * You should have received a copy of the GNU General Public License version
016 * 2 along with this work; if not, write to the Free Software Foundation,
017 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
018 *
019 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
020 * or visit www.oracle.com if you need additional information or have any
021 * questions.
022 */
023package com.oracle.graal.truffle.test.builtins;
024
025import com.oracle.truffle.api.*;
026import com.oracle.truffle.api.dsl.*;
027import com.oracle.truffle.api.frame.*;
028import com.oracle.truffle.api.nodes.*;
029import com.oracle.truffle.api.source.*;
030import com.oracle.truffle.sl.*;
031import com.oracle.truffle.sl.nodes.*;
032import com.oracle.truffle.sl.runtime.*;
033
034/**
035 * Returns all functions that start with test.
036 */
037@NodeInfo(shortName = "callFunctionsWith")
038public abstract class SLCallFunctionsWithBuiltin extends SLGraalRuntimeBuiltin {
039
040    @Child private IndirectCallNode indirectCall = Truffle.getRuntime().createIndirectCallNode();
041
042    @Specialization
043    public SLNull runTests(VirtualFrame frame, String startsWith, SLFunction harness) {
044        SLContext context = ((SLRootNode) getRootNode()).getSLContext();
045        boolean found = false;
046        for (SLFunction function : context.getFunctionRegistry().getFunctions()) {
047            if (function.getName().startsWith(startsWith) && getSource(function) == getSource(harness) && getSource(function) != null) {
048                indirectCall.call(frame, harness.getCallTarget(), new Object[]{function});
049                found = true;
050            }
051        }
052        if (!found) {
053            throw new SLAssertionError("No tests found to execute.");
054        }
055        return SLNull.SINGLETON;
056    }
057
058    private static Source getSource(SLFunction function) {
059        SourceSection section = function.getCallTarget().getRootNode().getSourceSection();
060        if (section != null) {
061            return section.getSource();
062        }
063        return null;
064    }
065}