info.novatec.testit.webtester.junit5.extensions.browsers.ManagedBrowserExtension Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of webtester-support-junit5 Show documentation
There is a newer version: 2.10.3
Show newest version
package info.novatec.testit.webtester.junit5.extensions.browsers;

import java.lang.reflect.Field;
import java.lang.reflect.Modifier;
import java.lang.reflect.UndeclaredThrowableException;
import java.util.ArrayList;
import java.util.List;
import java.util.function.Predicate;

import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Nested;
import org.junit.jupiter.api.extension.AfterAllCallback;
import org.junit.jupiter.api.extension.AfterEachCallback;
import org.junit.jupiter.api.extension.BeforeAllCallback;
import org.junit.jupiter.api.extension.BeforeEachCallback;
import org.junit.jupiter.api.extension.Extension;
import org.junit.jupiter.api.extension.ExtensionContext;

import lombok.extern.slf4j.Slf4j;

import info.novatec.testit.webtester.browser.Browser;
import info.novatec.testit.webtester.browser.BrowserFactory;
import info.novatec.testit.webtester.browser.proxy.ProxyConfiguration;
import info.novatec.testit.webtester.junit5.extensions.BaseExtension;


/**
 * This {@link Extension} is used to manage and inject {@link Browser} instances into static and non static fields of a test
 * class. This injection is executed according to the field's modifier.
 * 
 * Static browser fields are initialized before the first {@link BeforeAll} method is executed and the browser will be
 * automatically closed after the last {@link AfterAll} method was executed.
 * Instance browser fields are initialized before the first {@link BeforeEach} method is executed and the browser will
 * be automatically closed after the last {@link AfterEach} method was executed.
 * 
 * In order to decide how the browser instances are created either the test class must be annotated with {@link
 * CreateBrowsersUsing} or the field with {@link CreateUsing}.
 * 
 * In case more than one browser is managed, each has to have a unique name which can be defined by setting the {@link
 * Managed#value()} property.
 * 

 * Example:
 * 
 * @EnableWebTesterExtensions
 * @CreateBrowsersUsing(DefaultFactory.class)
 * public class FooTest {
 *
 *     @Managed
 *     static Browser browser;
 *
 * }
 *
 * @EnableWebTesterExtensions
 * @CreateBrowsersUsing(DefaultFactory.class)
 * public class BarTest {
 *
 *     @Managed("browser-1")
 *     static Browser browser1;
 *     @Managed("browser-2")
 *     @CreateUsing(SpecialFactory.class)
 *     Browser browser2;
 *
 * }
 * 
 *
 * @see Managed
 * @see Browser
 * @see CreateBrowsersUsing
 * @see CreateUsing
 * @since 2.1
 */
@Slf4j
@SuppressWarnings({ "PMD.SignatureDeclareThrowsException", "PMD.AvoidCatchingGenericException" })
public class ManagedBrowserExtension extends BaseExtension
    implements BeforeAllCallback, AfterAllCallback, BeforeEachCallback, AfterEachCallback {

    private final Predicate isStaticField = (field) -> Modifier.isStatic(field.getModifiers());
    private final Predicate isInstanceField = isStaticField.negate();

    @Override
    public void beforeAll(ExtensionContext context) throws Exception {
        executeHandlingUndeclaredThrowables(context, this::initializeAndInjectStaticBrowsers);
    }

    @Override
    public void beforeEach(ExtensionContext context) throws Exception {
        executeHandlingUndeclaredThrowables(context, this::initializeAndInjectInstanceBrowsers);
    }

    @Override
    public void afterEach(ExtensionContext context) {
        getManagedInstanceBrowsers(context).forEach(this::closeAndLogErrors);
    }

    @Override
    public void afterAll(ExtensionContext context) {
        getManagedStaticBrowsers(context).forEach(this::closeAndLogErrors);
    }

    private void closeAndLogErrors(Browser browser) {
        try {
            log.debug("closing managed browser: {}", browser);
            browser.close();
        } catch (RuntimeException e) {
            log.warn("error while closing managed browser instance:", e);
        }
    }

    private void initializeAndInjectStaticBrowsers(ExtensionContext context) {
        List managedBrowsers = getManagedStaticBrowsers(context);
        getModel(context).getBrowserFields().stream().filter(isStaticField).forEach(field -> {
            Browser browser = createBrowserFor(field);
            managedBrowsers.add(browser);
            setValue(field, null, browser);
        });
    }

    private void initializeAndInjectInstanceBrowsers(ExtensionContext context) {
        Object testInstance = context.getRequiredTestInstance();
        List managedBrowsers = getManagedInstanceBrowsers(context);
        getModel(context).getBrowserFields().stream().filter(isInstanceField).forEach(field -> {
            Browser browser = createBrowserFor(field);
            managedBrowsers.add(browser);
            setValue(field, testInstance, browser);
        });
    }

    private Browser createBrowserFor(Field field) {

        CreateUsing createUsing = getFieldLevelAnnotation(field);
        if (createUsing != null) {
            Class proxyConfigurationClass = createUsing.proxy();
            Class factoryClass = createUsing.value();
            return createBrowser(factoryClass, proxyConfigurationClass);
        }

        CreateBrowsersUsing createBrowsersUsing = getClassLevelAnnotation(field);
        if (createBrowsersUsing != null) {
            Class proxyConfigurationClass = createBrowsersUsing.proxy();
            Class factoryClass = createBrowsersUsing.value();
            return createBrowser(factoryClass, proxyConfigurationClass);
        }

        throw new NoBrowserFactoryException();

    }

    private CreateUsing getFieldLevelAnnotation(Field field) {
        return field.getAnnotation(CreateUsing.class);
    }

    private CreateBrowsersUsing getClassLevelAnnotation(Field field) {
        Class testClass = field.getDeclaringClass();
        CreateBrowsersUsing isPresent = testClass.getAnnotation(CreateBrowsersUsing.class);
        if (isPresent == null && testClass.isAnnotationPresent(Nested.class)) {
            Class declaringClass = testClass.getDeclaringClass();
            while (declaringClass != null && isPresent == null) {
                isPresent = declaringClass.getAnnotation(CreateBrowsersUsing.class);
                declaringClass = declaringClass.getDeclaringClass();
            }
        }
        return isPresent;
    }

    private Browser createBrowser(Class factoryClass,
        Class proxyConfigurationClass) {
        try {
            ProxyConfiguration proxyConfiguration = proxyConfigurationClass.newInstance();
            return factoryClass.newInstance().withProxyConfiguration(proxyConfiguration).createBrowser();
        } catch (InstantiationException | IllegalAccessException e) {
            throw new UndeclaredThrowableException(e, "error while creating browser factory");
        }
    }

    @SuppressWarnings("unchecked")
    private List getManagedInstanceBrowsers(ExtensionContext context) {
        return ( List ) context.getStore(BaseExtension.NAMESPACE)
            .getOrComputeIfAbsent("managed-instance-browsers", s -> new ArrayList());
    }

    @SuppressWarnings("unchecked")
    private List getManagedStaticBrowsers(ExtensionContext context) {
        return ( List ) context.getStore(BaseExtension.NAMESPACE)
            .getOrComputeIfAbsent("managed-static-browsers", s -> new ArrayList());
    }

}