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

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jmockit Show documentation

JMockit is a Java toolkit for automated developer testing. It contains mocking/faking APIs and a code coverage tool, supporting both JUnit and TestNG. The mocking APIs allow all kinds of Java code, without testability restrictions, to be tested in isolation from selected dependencies.

There is a newer version: 1.49

Show newest version

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

import javax.annotation.*;

import org.junit.runner.*;
import org.junit.runner.notification.*;

import mockit.*;
import mockit.coverage.*;
import mockit.coverage.testRedundancy.*;
import mockit.integration.*;
import mockit.internal.faking.*;
import mockit.internal.state.TestRun;

/**
 * Startup mock 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 RunNotifierDecorator extends MockUp
{
   @Mock
   public static void fireTestRunStarted(Invocation invocation, Description description) {
      RunNotifier it = invocation.getInvokedInstance();
      assert it != null;

      if (TestCoverage.INSTANCE != null && CodeCoverage.active()) {
         it.addListener(new JUnitListener());
      }

      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();
      }
   }
}