net.n2oapp.framework.autotest.ComponentFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of n2o-autotest Show documentation
The newest version!
package net.n2oapp.framework.autotest;

import com.codeborne.selenide.ElementsCollection;
import com.codeborne.selenide.SelenideElement;
import net.n2oapp.framework.autotest.api.ComponentLibrary;
import net.n2oapp.framework.autotest.api.collection.ComponentsCollection;
import net.n2oapp.framework.autotest.api.component.Component;

import java.util.Arrays;
import java.util.HashSet;
import java.util.Optional;
import java.util.Set;

/**
 * Фабрика реализаций компонентов для автотестирования
 */
public class ComponentFactory {
    private Set> components = new HashSet<>();
    private Set> collections = new HashSet<>();

    @SuppressWarnings("unchecked")
    public  T produce(SelenideElement element,
                                           Class componentClass) {
        T component = (T) findAndProduce(componentClass, components);
        component.setElement(element);
        return component;
    }

    @SuppressWarnings("unchecked")
    public  T produce(ElementsCollection elements,
                                                      Class collectionClass) {
        T collection = (T) findAndProduce(collectionClass, collections);
        collection.setElements(elements);
        return collection;
    }

    @SafeVarargs
    public final ComponentFactory addComponents(Class... components) {
        this.components.addAll(Arrays.asList(components));
        return this;
    }

    @SafeVarargs
    public final ComponentFactory addCollections(Class... collections) {
        this.collections.addAll(Arrays.asList(collections));
        return this;
    }

    public ComponentFactory addLibrary(ComponentLibrary library) {
        components.addAll(library.components());
        collections.addAll(library.collections());
        return this;
    }

    private Object findAndProduce(Class targetClass, Set> candidates) {
        Optional> found = find(targetClass, candidates);
        if (!found.isPresent())
            throw new IllegalArgumentException("Implementation not found for " + targetClass);
        try {
            return found.get().getConstructor().newInstance();
        } catch (Exception e) {
            throw new IllegalStateException(e);
        }
    }

    private Optional> find(Class targetClass, Set> candidates) {
        if (targetClass.isInterface()) {
            for (Class candidate : candidates) {
                if (targetClass.isAssignableFrom(candidate))
                    return Optional.of(candidate);
            }
        } else {
            return Optional.of(targetClass);
        }
        return Optional.empty();
    }
}