org.jacoco.core.runtime.OfflineInstrumentationAccessGeneratorTest.java Source code

Introduction

Here is the source code for org.jacoco.core.runtime.OfflineInstrumentationAccessGeneratorTest.java

Source

/*******************************************************************************
 * Copyright (c) 2009, 2013 Mountainminds GmbH & Co. KG and Contributors
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *    Marc R. Hoffmann - initial API and implementation
 *    
 *******************************************************************************/
package org.jacoco.core.runtime;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertSame;

import org.jacoco.core.JaCoCo;
import org.jacoco.core.instr.MethodRecorder;
import org.jacoco.core.internal.instr.InstrSupport;
import org.jacoco.core.test.TargetLoader;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;
import org.objectweb.asm.ClassWriter;
import org.objectweb.asm.Opcodes;
import org.objectweb.asm.Type;
import org.objectweb.asm.commons.GeneratorAdapter;
import org.objectweb.asm.commons.Method;

/**
 * Unit tests for {@link OfflineInstrumentationAccessGenerator}.
 */
public class OfflineInstrumentationAccessGeneratorTest {

    private IExecutionDataAccessorGenerator generator;

    private static boolean[] probes;

    // runtime stub
    public static boolean[] getProbes(final long classid, final String classname, final int probecount) {
        return probes;
    }

    @BeforeClass
    public static void setupClass() {
        probes = new boolean[3];
    }

    @Before
    public void setup() {
        String name = getClass().getName().replace('.', '/');
        generator = new OfflineInstrumentationAccessGenerator(name);
    }

    @Test
    public void testRuntimeAccess() throws Exception {
        ITarget target = generateAndInstantiateClass(123);
        assertSame(probes, target.get());
    }

    @Test
    public void testRuntimeClassName() throws Exception {
        generator = new OfflineInstrumentationAccessGenerator();
        MethodRecorder actual = new MethodRecorder();
        generator.generateDataAccessor(987654321, "foo/Bar", 17, actual.getVisitor());

        MethodRecorder expected = new MethodRecorder();
        expected.getVisitor().visitLdcInsn(Long.valueOf(987654321));
        expected.getVisitor().visitLdcInsn("foo/Bar");
        expected.getVisitor().visitIntInsn(Opcodes.BIPUSH, 17);
        String rtname = JaCoCo.RUNTIMEPACKAGE.replace('.', '/') + "/Offline";
        expected.getVisitor().visitMethodInsn(Opcodes.INVOKESTATIC, rtname, "getProbes",
                "(JLjava/lang/String;I)[Z");

        assertEquals(expected, actual);
    }

    /**
     * Creates a new class with the given id, loads this class and instantiates
     * it. The constructor of the generated class will request the probe array
     * from the access generator under test.
     */
    private ITarget generateAndInstantiateClass(int classid) throws InstantiationException, IllegalAccessException {

        final String className = "org/jacoco/test/targets/RuntimeTestTarget_" + classid;
        Type classType = Type.getObjectType(className);

        final ClassWriter writer = new ClassWriter(0);
        writer.visit(Opcodes.V1_5, Opcodes.ACC_PUBLIC, className, null, "java/lang/Object",
                new String[] { Type.getInternalName(ITarget.class) });

        writer.visitField(InstrSupport.DATAFIELD_ACC, InstrSupport.DATAFIELD_NAME, InstrSupport.DATAFIELD_DESC,
                null, null);

        // Constructor
        GeneratorAdapter gen = new GeneratorAdapter(
                writer.visitMethod(Opcodes.ACC_PUBLIC, "<init>", "()V", null, new String[0]), Opcodes.ACC_PUBLIC,
                "<init>", "()V");
        gen.visitCode();
        gen.loadThis();
        gen.invokeConstructor(Type.getType(Object.class), new Method("<init>", "()V"));
        gen.loadThis();
        final int size = generator.generateDataAccessor(classid, className, 2, gen);
        gen.putStatic(classType, InstrSupport.DATAFIELD_NAME, Type.getObjectType(InstrSupport.DATAFIELD_DESC));
        gen.returnValue();
        gen.visitMaxs(size + 1, 0);
        gen.visitEnd();

        // get()
        gen = new GeneratorAdapter(writer.visitMethod(Opcodes.ACC_PUBLIC, "get", "()[Z", null, new String[0]),
                Opcodes.ACC_PUBLIC, "get", "()[Z");
        gen.visitCode();
        gen.getStatic(classType, InstrSupport.DATAFIELD_NAME, Type.getObjectType(InstrSupport.DATAFIELD_DESC));
        gen.returnValue();
        gen.visitMaxs(1, 0);
        gen.visitEnd();

        writer.visitEnd();

        final TargetLoader loader = new TargetLoader(className.replace('/', '.'), writer.toByteArray());
        return (ITarget) loader.newTargetInstance();
    }

    /**
     * With this interface access read coverage data of the generated class.
     */
    public interface ITarget {

        /**
         * Returns a reference to the probe array.
         * 
         * @return the probe array
         */
        boolean[] get();

    }

}