co.verisoft.fw.objectrepository.ObjectReporsitoryFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of Selenium4-Junit5 Show documentation
VeriSoft framework for testing web and mobile applications. Selenium Module
There is a newer version: 2.3.9
package co.verisoft.fw.objectrepository;

import co.verisoft.fw.report.observer.Report;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.extern.slf4j.Slf4j;
import org.jetbrains.annotations.Nullable;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.WrapsElement;
import org.openqa.selenium.interactions.Locatable;

import java.io.File;
import java.io.IOException;
import java.lang.reflect.Field;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Proxy;
import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;
import java.util.stream.Stream;

@Slf4j
public class ObjectReporsitoryFactory {
    protected static ObjectRepository repository;

    static ObjectRepository retrieveObjectRepository(String objectRepositoryFilePath) {
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            File file = new File(objectRepositoryFilePath);
            return objectMapper.readValue(file, ObjectRepository.class);
        } catch (NullPointerException e) {
            throw new RuntimeException("Property 'object.repository.path' is not defined in root.config.properties or default.config.properties file");
        } catch (IOException e) {
            log.warn(String.format("Object repository file not found (%s): ", objectRepositoryFilePath) + e);
            return new ObjectRepository();
        }
    }
    private static Field[] getAllFields(Object page) {
        List fields = new ArrayList<>();
        Class currentClass = page.getClass();
        while (currentClass != null) {
            fields.addAll(Stream.of(currentClass.getDeclaredFields()).collect(Collectors.toList()));
            currentClass = currentClass.getSuperclass();
        }
        return fields.toArray(new Field[0]);
    }
    public static void initObjects(WebDriver driver, Object page, String objectRepositoryFilePath) {
        repository = retrieveObjectRepository(objectRepositoryFilePath);
        @Nullable String pageName = getPageName(page);

        Field[] allFields = getAllFields(page);
        for (Field field : allFields) {
            field.setAccessible(true);

            if (field.getType().equals(WebElement.class) &&
                    field.getAnnotation(ObjectRepositoryItem.class) != null) {
                try {
                    field.set(page, createWebElementProxy(driver, field.getAnnotation(ObjectRepositoryItem.class).id(), pageName));
                } catch (Exception e) {
                    Report.error("Could not proxy object from object repository. Message is " + e.getMessage());
                    throw new RuntimeException(e);
                }
            } else if (isListOfWebElements(field) &&
                    field.getAnnotation(ObjectRepositoryItem.class) != null) {
                try {
                    field.set(page, createListWebElementProxy(driver, field.getAnnotation(ObjectRepositoryItem.class).id(), pageName));
                } catch (Exception e) {
                    Report.error("Could not proxy list of objects from object repository. Message is " + e.getMessage());
                    throw new RuntimeException(e);
                }
            }
        }
    }

    private static boolean isListOfWebElements(Field field) {
        Type genericType = field.getGenericType();
        if (genericType instanceof ParameterizedType) {
            Type[] typeArguments = ((ParameterizedType) genericType).getActualTypeArguments();
            if (typeArguments.length == 1 && typeArguments[0] == WebElement.class) {
                return field.getType() == List.class;
            }
        }
        return false;
    }

    private static Object createWebElementProxy(WebDriver driver, String elementObjectId, String pageName) {
        return Proxy.newProxyInstance(
                WebElement.class.getClassLoader(),
                new Class[]{WebElement.class, WrapsElement.class, Locatable.class},
                new DynamicWebElement(driver, repository, elementObjectId, pageName));
    }

    private static Object createListWebElementProxy(WebDriver driver, String elementObjectId, String pageName) {
        return Proxy.newProxyInstance(
                List.class.getClassLoader(),
                new Class[]{List.class},
                (proxy, method, args) -> {
                    List elements = new DynamicWebElements(driver, repository, elementObjectId, pageName).resolveElementsFromRepository(proxy);
                    return method.invoke(elements, args);
                });
    }

    private static @Nullable String getPageName(Object page) {
        if (Objects.nonNull(page.getClass().getAnnotation(PageObjectName.class)))
            return page.getClass().getSimpleName();
        else
            return null;
    }
}