All Downloads are FREE. Search and download functionalities are using the official Maven repository.

io.github.bonigarcia.seljup.SelenideHandler Maven / Gradle / Ivy

The newest version!
/*
 * (C) Copyright 2022 Boni Garcia (https://bonigarcia.github.io/)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */
package io.github.bonigarcia.seljup;

import static java.lang.invoke.MethodHandles.lookup;
import static org.slf4j.LoggerFactory.getLogger;

import java.lang.reflect.Parameter;
import java.util.Optional;

import org.openqa.selenium.WebDriver;
import org.slf4j.Logger;

import com.codeborne.selenide.SelenideConfig;

/**
 * Logic for creating instances of SelenideDriver.
 *
 * @author Boni Garcia
 * @since 4.1.0
 */
public class SelenideHandler {

    final Logger log = getLogger(lookup().lookupClass());

    static final String SELENIDE_DRIVER_CLASS = "com.codeborne.selenide.SelenideDriver";
    static final String SELENIDE_CONFIG_INTERFACE = "com.codeborne.selenide.Config";
    static final String SELENIDE_CONFIG_CLASS = "com.codeborne.selenide.SelenideConfig";
    static final String SELENIDE_PROXY_CLASS = "com.codeborne.selenide.proxy.SelenideProxyServer";

    AnnotationsReader annotationsReader;

    public SelenideHandler(AnnotationsReader annotationsReader) {
        this.annotationsReader = annotationsReader;
    }

    public boolean isSelenide(Class type) {
        return type.getCanonicalName().equals(SELENIDE_DRIVER_CLASS);
    }

    public boolean useCustomSelenideConfig(Parameter parameter,
            Optional testInstance) {
        SelenideConfiguration selenideConfiguration = parameter
                .getAnnotation(SelenideConfiguration.class);
        SelenideConfig globalConfig = annotationsReader.getFromAnnotatedField(
                testInstance, SelenideConfiguration.class,
                SelenideConfig.class);
        return selenideConfiguration != null || globalConfig != null;
    }

    public Object createSelenideDriver(WebDriver driver, Parameter parameter,
            Optional testInstance) {
        Object object = null;
        try {
            Object config = getSelenideConfig(testInstance, parameter);
            if (driver == null) {
                object = Class.forName(SELENIDE_DRIVER_CLASS)
                        .getDeclaredConstructor(
                                Class.forName(SELENIDE_CONFIG_INTERFACE))
                        .newInstance(config);
            } else {
                object = Class.forName(SELENIDE_DRIVER_CLASS)
                        .getDeclaredConstructor(
                                Class.forName(SELENIDE_CONFIG_INTERFACE),
                                WebDriver.class,
                                Class.forName(SELENIDE_PROXY_CLASS))
                        .newInstance(config, driver, null);
            }

        } catch (Exception e) {
            log.warn("Exception creating SelenideDriver object", e);
        }

        return object;
    }

    public Object getSelenideConfig(Optional testInstance,
            Parameter parameter) {
        Object config = null;
        try {
            config = Class.forName(SELENIDE_CONFIG_CLASS)
                    .getDeclaredConstructor().newInstance();

            if (parameter != null) {
                SelenideConfiguration selenideConfiguration = parameter
                        .getAnnotation(SelenideConfiguration.class);

                // @SelenideConfiguration as parameter
                if (selenideConfiguration != null) {
                    Class.forName(SELENIDE_CONFIG_CLASS)
                            .getDeclaredMethod("browser", String.class)
                            .invoke(config, selenideConfiguration.browser());
                    Class.forName(SELENIDE_CONFIG_CLASS)
                            .getDeclaredMethod("headless", boolean.class)
                            .invoke(config, selenideConfiguration.headless());
                    Class.forName(SELENIDE_CONFIG_CLASS)
                            .getDeclaredMethod("browserBinary", String.class)
                            .invoke(config,
                                    selenideConfiguration.browserBinary());
                }

                // @SelenideConfiguration as field
                SelenideConfig globalConfig = annotationsReader
                        .getFromAnnotatedField(testInstance,
                                SelenideConfiguration.class,
                                SelenideConfig.class);
                if (globalConfig != null) {
                    config = globalConfig;
                }
            }

        } catch (Exception e) {
            log.warn("Exception getting Selenide Config", e);
        }

        return config;
    }

}