cn.taketoday.aot.test.generate.TestGenerationContext Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of today-core-test Show documentation
Infra Core Test
There is a newer version: 5.0.0-Draft.1
/*
 * Original Author -> Harry Yang ([email protected]) https://taketoday.cn
 * Copyright © Harry Yang & 2017 - 2023 All Rights Reserved.
 *
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see [http://www.gnu.org/licenses/]
 */

package cn.taketoday.aot.test.generate;

import java.util.function.UnaryOperator;

import cn.taketoday.aot.generate.ClassNameGenerator;
import cn.taketoday.aot.generate.DefaultGenerationContext;
import cn.taketoday.aot.generate.GenerationContext;
import cn.taketoday.aot.generate.InMemoryGeneratedFiles;
import cn.taketoday.core.test.tools.TestCompiler;
import cn.taketoday.javapoet.ClassName;

/**
 * {@link GenerationContext} test implementation that uses
 * {@link InMemoryGeneratedFiles} and can configure a {@link TestCompiler}
 * instance.
 *
 * @author Stephane Nicoll
 * @author Sam Brannen
 * @since 4.0
 */
public class TestGenerationContext extends DefaultGenerationContext implements UnaryOperator {

  /**
   * The default test target {@link ClassName}.
   */
  public static final ClassName TEST_TARGET = ClassName.get("com.example", "TestTarget");

  /**
   * Create an instance using the specified {@link ClassNameGenerator}.
   *
   * @param classNameGenerator the class name generator to use
   */
  public TestGenerationContext(ClassNameGenerator classNameGenerator) {
    super(classNameGenerator, new InMemoryGeneratedFiles());
  }

  /**
   * Create an instance using the specified {@code target}.
   *
   * @param target the default target class name to use
   */
  public TestGenerationContext(ClassName target) {
    this(new ClassNameGenerator(target));
  }

  /**
   * Create an instance using the specified {@code target}.
   *
   * @param target the default target class to use
   */
  public TestGenerationContext(Class target) {
    this(ClassName.get(target));
  }

  /**
   * Create an instance using {@link #TEST_TARGET} as the {@code target}.
   */
  public TestGenerationContext() {
    this(TEST_TARGET);
  }

  @Override
  public InMemoryGeneratedFiles getGeneratedFiles() {
    return (InMemoryGeneratedFiles) super.getGeneratedFiles();
  }

  /**
   * Configure the specified {@link TestCompiler} with the state of this context.
   *
   * @param testCompiler the compiler to configure
   * @return a new {@link TestCompiler} instance configured with the generated files
   * @see TestCompiler#with(UnaryOperator)
   */
  @Override
  public TestCompiler apply(TestCompiler testCompiler) {
    return CompilerFiles.from(getGeneratedFiles()).apply(testCompiler);
  }

}