JaninoConnectionTest.java

/*
 * Copyright 2006-2012 The Scriptella Project Team.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package scriptella.driver.janino;

import org.codehaus.commons.compiler.CompileException;
import scriptella.AbstractTestCase;
import scriptella.configuration.MockConnectionEl;
import scriptella.configuration.StringResource;
import scriptella.spi.ConnectionParameters;
import scriptella.spi.MockDriverContext;
import scriptella.spi.MockParametersCallbacks;
import scriptella.spi.ParametersCallback;
import scriptella.spi.ProviderException;
import scriptella.spi.QueryCallback;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * Tests Janino connection class.
 *
 * @author Fyodor Kupolov
 * @version 1.0
 */
public class JaninoConnectionTest extends AbstractTestCase {
    public static int field; //Field to be filled by a script

    /**
     * This test creates a Janino connection that executes simple valid and invalid scripts.
     */
    public void testScript() {
        JaninoConnection c = new JaninoConnection(new ConnectionParameters(new MockConnectionEl(), MockDriverContext.INSTANCE));
        field = 0;
        c.executeScript(new StringResource(JaninoConnectionTest.class.getName() + ".field=1;"), null);
        try {
            c.executeScript(new StringResource(JaninoConnectionTest.class.getName() + ".nosuchfield=1;"), null);
            fail("This script should fail");
        } catch (ProviderException e) {
            Throwable ne = e.getNativeException();
            assertTrue(ne instanceof CompileException);
            //OK
        }
        c.close();
        assertEquals(1, field);
    }

    public void testQuery() {
        JaninoConnection c = new JaninoConnection(new ConnectionParameters(new MockConnectionEl(), MockDriverContext.INSTANCE));
        field = 0;
        final List<String> rows = new ArrayList<String>();

        c.executeQuery(new StringResource(
                "set(\"p\", \"//\"+get(\"p\") );" +
                "next();" +
                "next(new String[] {\"p\"}, new Object[] {\"v2\"});"),
                MockParametersCallbacks.SIMPLE, new QueryCallback() {
            public void processRow(final ParametersCallback parameters) {
                rows.add(parameters.getParameter("p").toString());
            }
        });
        c.close();
        List<String> expected = Arrays.asList("//*p*", "v2");
        assertEquals(expected, rows);
    }

    public void testErrorSourceCode() {
        JaninoConnection c = new JaninoConnection(new ConnectionParameters(new MockConnectionEl(), MockDriverContext.INSTANCE));
        //test compilation errors
        try {
            c.executeScript(new StringResource("int b=1;\na='"), MockParametersCallbacks.NULL);
            fail("Error statements must be recognized");
        } catch (JaninoProviderException e) {
            assertEquals("a='", e.getErrorStatement());
        }
        //test execution errors
        //Enable debug info for Janino
        System.setProperty("org.codehaus.janino.source_debugging.enable", "true");
        try {
            c.executeScript(new StringResource("int b=1;\nObject a=get(\"1\");"), MockParametersCallbacks.UNSUPPORTED);
            fail("Error statements must be recognized");
        } catch (JaninoProviderException e) {
            assertTrue("UnsupportedOperationException is expected to be thrown by get() method", e.getCause() instanceof UnsupportedOperationException);
            // Error statement cann be determined because line info is available
            assertEquals("Object a=get(\"1\");", e.getErrorStatement());
        }
        //Disable debug info for Janino
        System.getProperties().remove("org.codehaus.janino.source_debugging.enable");
        try {
            c.executeScript(new StringResource("int b=1;\nObject a=get(\"1\");"), MockParametersCallbacks.UNSUPPORTED);
            fail("Error statements must be recognized");
        } catch (JaninoProviderException e) {
            assertTrue("UnsupportedOperationException is expected to be thrown by get() method", e.getCause() instanceof UnsupportedOperationException);
            // Error statement cannot be determined because line info is not available
            assertNull(e.getErrorStatement());
        }
    }
}