/*
 * Copyright 2021 Google Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
 * in compliance with the License. You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License
 * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 * or implied. See the License for the specific language governing permissions and limitations under
 * the License.
 */

package com.google.testing.junit.testparameterinjector;

import com.google.common.collect.ImmutableList;
import java.util.List;
import org.junit.rules.TestRule;
import org.junit.runner.Description;
import org.junit.runners.model.InitializationError;
import org.junit.runners.model.Statement;

/**
 * A JUnit test runner which knows how to instantiate and run test classes where each test case may
 * be parameterized with its own unique set of test parameters (as opposed to {@link
 * org.junit.runners.Parameterized} where each test case in a test class is invoked with the exact
 * same set of parameters).
 */
public final class TestParameterInjector extends PluggableTestRunner {

  public TestParameterInjector(Class<?> testClass) throws InitializationError {
    super(testClass);
  }

  @Override
  protected List<TestRule> getInnerTestRules() {
    return ImmutableList.of(new TestNamePrinterRule());
  }

  @Override
  protected List<TestMethodProcessor> createTestMethodProcessorList() {
    return TestMethodProcessors.createNewParameterizedProcessorsWithLegacyFeatures(getTestClass());
  }

  /** A {@link TestRule} that prints the current test name before and after the test. */
  private static final class TestNamePrinterRule implements TestRule {

    @Override
    public Statement apply(final Statement originalStatement, final Description testDescription) {
      return new Statement() {
        @Override
        public void evaluate() throws Throwable {
          String testName =
              testDescription.getTestClass().getSimpleName()
                  + "."
                  + testDescription.getMethodName();
          System.out.println("\n\nBeginning test: " + testName);
          try {
            originalStatement.evaluate();
          } finally {
            System.out.println("\nEnd of test: " + testName);
          }
        }
      };
    }
  }
}

@@ -651,140 +651,22 @@
 ort 
-com.google.common.collect.ImmutableList;%0Aimport java.util.List;%0Aimport org.junit.rules.TestRule;%0Aimport org.junit.runner.Description
+java.util.List
 ;%0Aim
@@ -718,50 +718,8 @@
 ror;
-%0Aimport org.junit.runners.model.Statement;
 %0A%0A/*
@@ -1219,130 +1219,8 @@
  %7D%0A%0A
-  @Override%0A  protected List%3CTestRule%3E getInnerTestRules() %7B%0A    return ImmutableList.of(new TestNamePrinterRule());%0A  %7D%0A%0A
   @O
@@ -1406,795 +1406,7 @@
 %0A  %7D
-%0A%0A  /** A %7B@link TestRule%7D that prints the current test name before and after the test. */%0A  private static final class TestNamePrinterRule implements TestRule %7B%0A%0A    @Override%0A    public Statement apply(final Statement originalStatement, final Description testDescription) %7B%0A      return new Statement() %7B%0A        @Override%0A        public void evaluate() throws Throwable %7B%0A          String testName =%0A              testDescription.getTestClass().getSimpleName()%0A                  + %22.%22%0A                  + testDescription.getMethodName();%0A          System.out.println(%22%5Cn%5CnBeginning test: %22 + testName);%0A          try %7B%0A            originalStatement.evaluate();%0A          %7D finally %7B%0A            System.out.println(%22%5CnEnd of test: %22 + testName);%0A          %7D%0A        %7D%0A      %7D;%0A    %7D%0A  %7D
 %0A%7D%0A

Remove the test rule that prints the test name