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

Introduction

Here is the source code for org.jacoco.core.internal.flow.MethodSanitizerTest.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.jacoco.core.instr.MethodRecorder;
import org.junit.Before;
import org.junit.Test;
import org.objectweb.asm.Label;
import org.objectweb.asm.MethodVisitor;
import org.objectweb.asm.Opcodes;
import org.objectweb.asm.tree.MethodNode;

/**
 * Unit tests for {@link MethodSanitizer}.
 */
public class MethodSanitizerTest {

    private MethodNode actual;
    private MethodNode expected;

    private MethodVisitor sanitizer;

    @Before
    public void setup() {
        actual = new MethodNode(0, "test", "()V", null, null);
        expected = new MethodNode(0, "test", "()V", null, null);
        sanitizer = new MethodSanitizer(actual, 0, "test", "()V", null, null);
    }

    @Test
    public void testLocalVariablePositive() {
        Label l1 = new Label();
        Label l2 = new Label();
        sanitizer.visitCode();
        sanitizer.visitLabel(l1);
        sanitizer.visitInsn(Opcodes.RETURN);
        sanitizer.visitLabel(l2);
        sanitizer.visitLocalVariable("x", "I", null, l1, l2, 0);
        sanitizer.visitMaxs(0, 0);
        sanitizer.visitEnd();

        Label m1 = new Label();
        Label m2 = new Label();
        expected.visitLabel(m1);
        expected.visitInsn(Opcodes.RETURN);
        expected.visitLabel(m2);
        expected.visitLocalVariable("x", "I", null, m1, m2, 0);
        expected.visitMaxs(0, 0);
        expected.visitEnd();

        assertOutput();
    }

    @Test
    public void testLocalVariableNegative1() {
        Label l1 = new Label();
        Label l2 = new Label();
        sanitizer.visitCode();
        sanitizer.visitInsn(Opcodes.RETURN);
        sanitizer.visitLabel(l2);
        sanitizer.visitLocalVariable("x", "I", null, l1, l2, 0);
        sanitizer.visitMaxs(0, 0);
        sanitizer.visitEnd();

        Label m2 = new Label();
        expected.visitInsn(Opcodes.RETURN);
        expected.visitLabel(m2);
        expected.visitMaxs(0, 0);
        expected.visitEnd();

        assertOutput();
    }

    @Test
    public void testLocalVariableNegative2() {
        Label l1 = new Label();
        Label l2 = new Label();
        sanitizer.visitCode();
        sanitizer.visitLabel(l1);
        sanitizer.visitInsn(Opcodes.RETURN);
        sanitizer.visitLocalVariable("x", "I", null, l1, l2, 0);
        sanitizer.visitMaxs(0, 0);
        sanitizer.visitEnd();

        Label m1 = new Label();
        expected.visitLabel(m1);
        expected.visitInsn(Opcodes.RETURN);
        expected.visitMaxs(0, 0);
        expected.visitEnd();

        assertOutput();
    }

    @Test
    public void testLineNumberPositive() {
        Label l1 = new Label();
        sanitizer.visitCode();
        sanitizer.visitLabel(l1);
        sanitizer.visitLineNumber(15, l1);
        sanitizer.visitInsn(Opcodes.RETURN);
        sanitizer.visitMaxs(0, 0);
        sanitizer.visitEnd();

        Label m1 = new Label();
        expected.visitLabel(m1);
        expected.visitLineNumber(15, m1);
        expected.visitInsn(Opcodes.RETURN);
        expected.visitMaxs(0, 0);
        expected.visitEnd();

        assertOutput();
    }

    @Test
    public void testLineNumberNegative() {
        Label l1 = new Label();
        sanitizer.visitCode();
        sanitizer.visitLineNumber(15, l1);
        sanitizer.visitInsn(Opcodes.RETURN);
        sanitizer.visitMaxs(0, 0);
        sanitizer.visitEnd();

        expected.visitInsn(Opcodes.RETURN);
        expected.visitMaxs(0, 0);
        expected.visitEnd();

        assertOutput();
    }

    private void assertOutput() {
        assertEquals(dump(expected), dump(actual));
    }

    private MethodRecorder dump(MethodNode node) {
        MethodRecorder rec = new MethodRecorder();
        node.accept(rec.getVisitor());
        return rec;
    }
}