JavaFixture/src/main/java/com/github/nylle/javafixture/SpecimenFactory.java at 9ee576851a444617c8df0c6738e42001e17818a8 · Nylle/JavaFixture · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
package com.github.nylle.javafixture;

import com.github.nylle.javafixture.specimen.AbstractSpecimen;
import com.github.nylle.javafixture.specimen.ArraySpecimen;
import com.github.nylle.javafixture.specimen.CollectionSpecimen;
import com.github.nylle.javafixture.specimen.EnumSpecimen;
import com.github.nylle.javafixture.specimen.GenericSpecimen;
import com.github.nylle.javafixture.specimen.InterfaceSpecimen;
import com.github.nylle.javafixture.specimen.MapSpecimen;
import com.github.nylle.javafixture.specimen.ObjectSpecimen;
import com.github.nylle.javafixture.specimen.PredefinedSpecimen;
import com.github.nylle.javafixture.specimen.PrimitiveSpecimen;
import com.github.nylle.javafixture.specimen.SpecialSpecimen;
import com.github.nylle.javafixture.specimen.TimeSpecimen;

import java.util.List;

public class SpecimenFactory {
    private final static List<ISpecimen.IMeta> specimenCandidates = List.of(
            SpecialSpecimen.meta(),
            PrimitiveSpecimen.meta(),
            EnumSpecimen.meta(),
            CollectionSpecimen.meta(),
            MapSpecimen.meta(),
            GenericSpecimen.meta(),
            ArraySpecimen.meta(),
            TimeSpecimen.meta(),
            InterfaceSpecimen.meta(),
            AbstractSpecimen.meta()
    );

    private final Context context;

    public SpecimenFactory(Context context) {
        this.context = context;
    }

    public <T> ISpecimen<T> build(SpecimenType<T> type) {

        if (context.isCached(type)) {
            return new PredefinedSpecimen<>(type, context);
        }

        return specimenCandidates.stream()
                .filter(x -> x.supports(type))
                .map(x -> x.create(type, context, this))
                .findFirst()
                .orElseGet(() -> new ObjectSpecimen<>(type, context, this));
    }
}