org.jacoco.core.internal.flow.ClassProbesAdapterTest.java Source code

Introduction

Here is the source code for org.jacoco.core.internal.flow.ClassProbesAdapterTest.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.internal.flow;

import static org.junit.Assert.assertEquals;

import org.junit.Test;
import org.objectweb.asm.ClassVisitor;
import org.objectweb.asm.Label;
import org.objectweb.asm.MethodVisitor;
import org.objectweb.asm.Opcodes;

/**
 * Unit tests for {@link ClassProbesAdapter}.
 */
public class ClassProbesAdapterTest {

    private static class MockVisitor extends ClassProbesVisitor {

        int count;

        @Override
        public MethodProbesVisitor visitMethod(int access, String name, String desc, String signature,
                String[] exceptions) {
            return null;
        }

        @Override
        public void visitTotalProbeCount(int count) {
            this.count = count;
        }
    }

    @Test
    public void testProbeCounter() {
        final MockVisitor mv = new MockVisitor();
        final ClassProbesAdapter adapter = new ClassProbesAdapter(mv);
        assertEquals(0, adapter.nextId());
        assertEquals(1, adapter.nextId());
        assertEquals(2, adapter.nextId());
        adapter.visitEnd();
        assertEquals(3, mv.count);
    }

    @Test
    public void testVisitClassMethods() {
        final MockVisitor mv = new MockVisitor() {
            @Override
            public MethodProbesVisitor visitMethod(int access, String name, String desc, String signature,
                    String[] exceptions) {
                class MockMethodVisitor extends MethodProbesVisitor {
                    @Override
                    public void visitProbe(int probeId) {
                    }

                    @Override
                    public void visitJumpInsnWithProbe(int opcode, Label label, int probeId) {
                    }

                    @Override
                    public void visitInsnWithProbe(int opcode, int probeId) {
                    }

                    @Override
                    public void visitTableSwitchInsnWithProbes(int min, int max, Label dflt, Label[] labels) {
                    }

                    @Override
                    public void visitLookupSwitchInsnWithProbes(Label dflt, int[] keys, Label[] labels) {
                    }
                }
                return new MockMethodVisitor();
            }
        };
        final ClassProbesAdapter adapter = new ClassProbesAdapter(mv);
        adapter.visit(Opcodes.V1_5, 0, "Foo", null, "java/lang/Object", null);
        writeMethod(adapter);
        writeMethod(adapter);
        writeMethod(adapter);

        assertEquals(0, mv.count);
        adapter.visitEnd();
        assertEquals(3, mv.count);
    }

    @Test
    public void testVisitInterfaceMethod() {
        final MockVisitor mv = new MockVisitor() {
            @Override
            public MethodProbesVisitor visitMethod(int access, String name, String desc, String signature,
                    String[] exceptions) {
                class MockMethodVisitor extends MethodProbesVisitor {
                    @Override
                    public void visitProbe(int probeId) {
                    }

                    @Override
                    public void visitJumpInsnWithProbe(int opcode, Label label, int probeId) {
                    }

                    @Override
                    public void visitInsnWithProbe(int opcode, int probeId) {
                    }

                    @Override
                    public void visitTableSwitchInsnWithProbes(int min, int max, Label dflt, Label[] labels) {
                    }

                    @Override
                    public void visitLookupSwitchInsnWithProbes(Label dflt, int[] keys, Label[] labels) {
                    }
                }
                return new MockMethodVisitor();
            }
        };
        final ClassProbesAdapter adapter = new ClassProbesAdapter(mv);
        adapter.visit(Opcodes.V1_5, Opcodes.ACC_INTERFACE, "Foo", null, "java/lang/Object", null);
        writeMethod(adapter);

        assertEquals(1, mv.count);
        adapter.visitEnd();
        assertEquals(1, mv.count);
    }

    @Test
    public void testVisitMethodNullMethodVisitor() {
        final MockVisitor mv = new MockVisitor();
        final ClassProbesAdapter adapter = new ClassProbesAdapter(mv);
        writeMethod(adapter);
        writeMethod(adapter);
        writeMethod(adapter);
        adapter.visitEnd();
        assertEquals(3, mv.count);
    }

    private void writeMethod(final ClassVisitor cv) {
        MethodVisitor mv = cv.visitMethod(0, "foo", "V()", null, null);
        mv.visitCode();
        mv.visitInsn(Opcodes.RETURN);
        mv.visitMaxs(0, 1);
        mv.visitEnd();
    }
}