com.github.tankist88.carpenter.generator.TestGenerator Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of carpenter-generator Show documentation
Carpenter is a framework for unit test generation from runtime data.
There is a newer version: 0.0.10
package com.github.tankist88.carpenter.generator;

import com.github.tankist88.carpenter.core.dto.unit.method.MethodCallInfo;
import com.github.tankist88.carpenter.core.property.GenerationProperties;
import com.github.tankist88.carpenter.generator.builder.TestBuilder;
import com.github.tankist88.carpenter.generator.service.LoadDataService;

import java.lang.reflect.Modifier;

import static com.github.tankist88.carpenter.core.property.GenerationPropertiesFactory.loadProps;
import static com.github.tankist88.carpenter.generator.util.GenerateUtil.allowedPackage;
import static com.github.tankist88.object2source.util.GenerationUtil.isAnonymousClass;

public class TestGenerator {
    public static final String GENERATED_TEST_CLASS_POSTFIX = "GeneratedTest";
    public static final String TEST_INST_VAR_NAME = "testInstance";

    private GenerationProperties props;
    private LoadDataService loadDataService;

    private TestGenerator() {
        this.props = loadProps();
        this.loadDataService = new LoadDataService();
    }
    
    public static boolean isCreateMockFields() {
        return loadProps().isCreateMockFields();
    }

    public static boolean isUsePowermock() {
        return loadProps().isUsePowermock();
    }

    private boolean skipTestMethod(MethodCallInfo callInfo) {
        return skipTestMethod(callInfo, props);
    }

    private static boolean skipTestMethod(MethodCallInfo callInfo, GenerationProperties props) {
        boolean deniedModifier = Modifier.isPrivate(callInfo.getMethodModifiers());
        boolean deniedDeclarationPlace = !callInfo.getClassName().equals(callInfo.getDeclaringTypeName());
        boolean deniedClassType = callInfo.isMemberClass() && !Modifier.isStatic(callInfo.getClassModifiers());
        boolean anonymousClass = isAnonymousClass(callInfo.getClassName());
        boolean skipNoZeroArgConst = !callInfo.isClassHasZeroArgConstructor() && !props.isNoZeroArgConstructorTestAllowed();
        return  !allowedPackage(callInfo.getClassName(), props) ||
                deniedModifier ||
                deniedClassType ||
                deniedDeclarationPlace ||
                anonymousClass ||
                skipNoZeroArgConst;
    }

    private int generate() {
        TestBuilder testBuilder = new TestBuilder();
        testBuilder.appendPreviousGenerated();
        for (MethodCallInfo callInfo : loadDataService.loadObjectDump()) {
            if(skipTestMethod(callInfo)) continue;
            if (isCreateMockFields()) {
                testBuilder
                        .appendMockTestField(callInfo)
                        .appendMockField(callInfo)
                        .appendInitMock(callInfo);
            }
            testBuilder.appendTestMethod(callInfo);
        }
        return testBuilder.build();
    }

    static int runGenerator() {
        return (new TestGenerator()).generate();
    }

    public static void main(String args[]) {
        int generatedTests = runGenerator();
        System.out.println("Generated tests count: " + generatedTests);
        System.out.println("Object dumps folder: " + loadProps().getObjectDumpDir());
        System.out.println("Destination folder: " + loadProps().getUtGenDir());
    }
}