mockit.integration.junit4.FakeRunNotifier Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2006 JMockit developers
 * This file is subject to the terms of the MIT license (see LICENSE.txt).
 */
package mockit.integration.junit4;

import edu.umd.cs.findbugs.annotations.NonNull;

import mockit.Invocation;
import mockit.Mock;
import mockit.MockUp;
import mockit.coverage.testRedundancy.JUnitListener;
import mockit.coverage.testRedundancy.TestCoverage;
import mockit.integration.TestRunnerDecorator;
import mockit.internal.faking.FakeInvocation;
import mockit.internal.state.TestRun;

import org.junit.runner.Description;
import org.junit.runner.Result;
import org.junit.runner.notification.RunNotifier;

/**
 * Startup fake which works in conjunction with {@link JUnit4TestRunnerDecorator} to provide JUnit 4.5+ integration.
 * 
 * This class is not supposed to be accessed from user code. JMockit will automatically load it at startup.
 */
public final class FakeRunNotifier extends MockUp {
    @Mock
    public static void fireTestRunStarted(Invocation invocation, Description description) {
        RunNotifier it = invocation.getInvokedInstance();
        assert it != null;

        TestCoverage testCoverage = TestCoverage.INSTANCE;

        if (testCoverage != null) {
            it.addListener(new JUnitListener(testCoverage));
        }

        prepareToProceed(invocation);
        it.fireTestRunStarted(description);
    }

    private static void prepareToProceed(@NonNull Invocation invocation) {
        ((FakeInvocation) invocation).prepareToProceedFromNonRecursiveMock();
    }

    @Mock
    public static void fireTestStarted(Invocation invocation, Description description) {
        Class currentTestClass = TestRun.getCurrentTestClass();

        if (currentTestClass != null) {
            Class newTestClass = description.getTestClass();

            if (newTestClass == null || !currentTestClass.isAssignableFrom(newTestClass)) {
                TestRunnerDecorator.cleanUpMocksFromPreviousTestClass();
            }
        }

        prepareToProceed(invocation);

        RunNotifier it = invocation.getInvokedInstance();
        assert it != null;
        it.fireTestStarted(description);
    }

    @Mock
    public static void fireTestRunFinished(Invocation invocation, Result result) {
        TestRun.enterNoMockingZone();

        try {
            TestRunnerDecorator.cleanUpAllMocks();

            prepareToProceed(invocation);

            RunNotifier it = invocation.getInvokedInstance();
            assert it != null;
            it.fireTestRunFinished(result);
        } finally {
            TestRun.exitNoMockingZone();
        }
    }
}