com.context.SessionContext Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of AppiumTestDistribution Show documentation
A tool run Android and iOS test in parallel across devices
There is a newer version: 9.0.1
Show newest version
package com.context;

import com.cucumber.listener.CucumberScenarioReporterListener;
import org.apache.log4j.Logger;
import org.testng.ITestContext;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Optional;
import java.util.Properties;

import static com.appium.utils.OverriddenVariable.getOverriddenStringValue;

public class SessionContext {
    static final String TEST_RUNNER = "testrunner";
    private static final HashMap allTestsExecutionContext;
    private static final Logger LOGGER = Logger.getLogger(SessionContext.class.getName());
    private static final Properties reportPortalProperties;
    private static String reportPortalLaunchURL = "";

    static {
        LOGGER.info("SessionContext default constructor");
        new SessionContext();
        allTestsExecutionContext = new HashMap<>();
        reportPortalProperties = loadReportPortalProperties();
        LOGGER.info("Initialized SessionContext");
    }

    static synchronized void addContext(long threadId, TestExecutionContext testExecutionContext) {
        allTestsExecutionContext.put(String.valueOf(threadId), testExecutionContext);
        LOGGER.info(String.format("Adding context for thread - %s", threadId));
    }

    public static synchronized TestExecutionContext getTestExecutionContext(long threadId) {
        return allTestsExecutionContext.get(String.valueOf(threadId));
    }

    public static synchronized void remove(long threadId) {
        LOGGER.info(String.format("Removing context for thread - %s", threadId));
        allTestsExecutionContext.remove(String.valueOf(threadId));
    }

    private static Properties loadReportPortalProperties() {
        Properties properties = new Properties();
        try {
            String reportPortalPropertiesFile = "src/test/resources/reportportal.properties";
            getOverriddenStringValue("REPORT_PORTAL_FILE", reportPortalPropertiesFile);
            LOGGER.info("Using reportportal.properties file from "
                    + reportPortalPropertiesFile);
            File reportPortalFile = new File(reportPortalPropertiesFile);
            String absolutePath = reportPortalFile.getAbsolutePath();
            if (reportPortalFile.exists()) {
                properties.load(new FileInputStream(absolutePath));
                LOGGER.info("Loaded reportportal.properties file - " + absolutePath);
            } else {
                LOGGER.info("reportportal.properties file NOT FOUND - " + absolutePath);
            }

        } catch (IOException e) {
            LOGGER.info("ERROR in loading reportportal.properties file\n" + e.getMessage());
            throw new RuntimeException(e.getMessage());
        }
        return properties;
    }

    public static void setReportPortalLaunchURL(ITestContext iTestContext) {
        Optional reportPortalListener = Arrays.stream(iTestContext.getSuite().getXmlSuite()
                .getListeners().toArray()).filter(x ->
                x.equals("com.epam.reportportal.testng.ReportPortalTestNGListener"))
                .findFirst();
        if (reportPortalListener.isPresent()) {
            setReportPortalLaunchURL();
        }
    }

    public static String getReportPortalLaunchURL() {
        return reportPortalLaunchURL;
    }

    public static void setReportPortalLaunchURL() {
        boolean isReportPortalEnabledInProperties =
                (null == reportPortalProperties.getProperty("rp.enable")
                 || (reportPortalProperties.getProperty("rp.enable")
                                           .equalsIgnoreCase("true")));
        if (isReportPortalEnabledInProperties) {
            String rpLaunchId = System.getProperty("rp.launch.id");
            LOGGER.debug(String.format("System property: rp.launch.id: '%s'",
                    rpLaunchId));
            reportPortalLaunchURL = String.format("%s/ui/#%s/launches/all/%s",
                                                  reportPortalProperties.getProperty("rp.endpoint"),
                                                  reportPortalProperties.getProperty("rp.project"),
                    rpLaunchId);
            LOGGER.info(String.format(
                    "**** ReportPortal URL - %s ****",
                    reportPortalLaunchURL));
        }
    }
}