com.github.tankist88.carpenter.generator.command.CreateInitMockMethodCommand Maven / Gradle / Ivy

Go to download
package com.github.tankist88.carpenter.generator.command;

import com.github.tankist88.carpenter.core.dto.unit.field.FieldProperties;
import com.github.tankist88.carpenter.core.dto.unit.method.MethodCallInfo;
import com.github.tankist88.carpenter.generator.TestGenerator;
import com.github.tankist88.carpenter.generator.dto.unit.method.MethodExtInfo;

import java.lang.reflect.Modifier;
import java.util.Collections;
import java.util.List;

import static com.github.tankist88.carpenter.core.property.AbstractGenerationProperties.TAB;

public class CreateInitMockMethodCommand extends AbstractReturnClassInfoCommand {
    public static final String INIT_METHOD = "init()";

    private StringBuilder builder;

    private MethodCallInfo callInfo;

    private List methods;

    public CreateInitMockMethodCommand(MethodCallInfo callInfo) {
        this.callInfo = callInfo;
        this.builder = new StringBuilder();
    }

    @Override
    public void execute() {
        super.execute();
        createInitMockMethod();
    }

    @Override
    public List returnResult() {
        return methods;
    }

    private void createInitMockMethod() {
        builder.append(TAB + "@BeforeMethod\n")
               .append(TAB + "public void ")
               .append(INIT_METHOD)
               .append(" {\n")
               .append(TAB + TAB + "initMocks(this);\n");

        for (FieldProperties f : callInfo.getServiceFields()) {
            if (f.getDeclaringClass().equals(callInfo.getClassName()) && !Modifier.isPrivate(f.getModifiers())) {
                builder.append(TAB + TAB + TestGenerator.TEST_INST_VAR_NAME).append(".")
                       .append(f.getUnitName()).append(" = ").append(f.getUnitName()).append(";\n");
            }
        }

        builder.append(TAB + "}\n");
        methods = Collections.singletonList(new MethodExtInfo(callInfo.getClassName(), INIT_METHOD, builder.toString()));
    }
}