com.flextrade.jfixture.JFixture Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jfixture Show documentation
JFixture is an open source library based on the popular .NET library, AutoFixture
There is a newer version: 2.7.2
package com.flextrade.jfixture;

import com.flextrade.jfixture.behaviours.specimentype.SpecimenTypeInjectorBehaviour;
import com.flextrade.jfixture.builders.CompositeBuilder;
import com.flextrade.jfixture.builders.DefaultEngineParts;
import com.flextrade.jfixture.customisation.AutoPropertyCustomisation;
import com.flextrade.jfixture.customisation.Customisation;
import com.flextrade.jfixture.customisation.ThrowOnNoResolutionCustomisation;
import com.flextrade.jfixture.customisation.ThrowingRecursionCustomisation;
import com.flextrade.jfixture.extensions.CreateExtensions;
import com.flextrade.jfixture.extensions.CreateExtensionsImpl;

import java.lang.reflect.Type;
import java.util.Collection;

public class JFixture implements SpecimenCreator, SpecimenBuilderPipeline, BehaviourProvider, CustomisationContainer {

    private final CompositeBuilder engine;
    private final FinalBuilderContainer finalBuilderContainer;
    private final FixtureBehaviours behaviours;
    private final FixtureCollections collections;
    private final FluentCustomisation customisation;
    private final CreateExtensions createExtensions;

    public JFixture() {
        DefaultEngineParts defaultEngineParts = new DefaultEngineParts();
        MultipleCount multipleCount = defaultEngineParts.multipleCount();
        Collection builders = defaultEngineParts.builders();

        this.engine = new CompositeBuilder(builders);
        this.finalBuilderContainer = new FinalBuilderContainer();
        this.behaviours = new FixtureBehaviours(this.finalBuilderContainer, this.engine);
        this.collections = new FixtureCollections(this, multipleCount);
        this.customisation = new DefaultFluentCustomisation(this, multipleCount);
        this.createExtensions = new CreateExtensionsImpl(this.finalBuilderContainer);

        this.applyDefaultCustomisations();
    }

    public FluentCustomisation customise() {
        return this.customisation;
    }

    public FixtureCollections collections() {
        return this.collections;
    }

    public BehavioursContainer behaviours() {
        return this.behaviours;
    }

    public JFixture customise(Customisation customisation) {
        customisation.customise(this);
        return this;
    }

    public JFixture addBuilderToStartOfPipeline(SpecimenBuilder builder) {
        this.engine.add(0, builder);
        return this;
    }

    public JFixture addBuilderToEndOfPipeline(SpecimenBuilder builder) {
        this.engine.add(builder);
        return this;
    }

    public CreateExtensions create() {
        return this.createExtensions;
    }

    @SuppressWarnings("unchecked")
    public  T create(Type type) {
        return (T) this.create((Object) type);
    }

    @SuppressWarnings("unchecked")
    public  T create(Class clazz) {
        return (T) this.create((Object) clazz);
    }

    Object create(Object object) {
        Object result = this.finalBuilderContainer.getBuilder().create(object, this.finalBuilderContainer.getContext());
        if (result instanceof NoSpecimen) return null;
        return result;
    }

    private void applyDefaultCustomisations() {
        this.customise(new AutoPropertyCustomisation());
        this.customise(new ThrowingRecursionCustomisation());
        this.customise(new ThrowOnNoResolutionCustomisation());
        this.behaviours().add(new SpecimenTypeInjectorBehaviour());
    }
}