com.aventstack.extentreports.service.ExtentTestManager Maven / Gradle / Ivy

Go to download
package com.aventstack.extentreports.service;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
import java.util.Optional;

import org.testng.ITestResult;

import com.aventstack.extentreports.ExtentTest;
import com.aventstack.extentreports.model.Test;
import com.aventstack.extentreports.testng.listener.commons.ExtentTestCommons;

public class ExtentTestManager {

    private static Map classTestMap = new HashMap<>();
    private static ThreadLocal methodTest = new ThreadLocal<>();
    private static ThreadLocal dataProviderTest = new ThreadLocal<>();

    public static synchronized ExtentTest getTest() {
        return methodTest.get();
    }

    public static synchronized ExtentTest getTest(ITestResult result) {
        ExtentTest t = result.getParameters() != null && result.getParameters().length > 0 
                ? dataProviderTest.get()
                : methodTest.get();
        return t;
    }

    public static synchronized ExtentTest createMethod(ITestResult result, Boolean createAsChild) {
        if (!createAsChild)
            return createMethod(result);

        String className = result.getInstance().getClass().getSimpleName();
        String methodName = result.getMethod().getMethodName();
        ExtentTest classTest;

        if (classTestMap.containsKey(className)) {
            classTest = classTestMap.get(className);
        } else {
            classTest = ExtentService.getInstance().createTest(className);
            classTestMap.put(className, classTest);
        }

        Optional test = classTest.getModel().getNodeContext().getAll().stream()
                .filter(x -> x.getName().equals(methodName)).findFirst();

        if (result.getParameters().length > 0) {
            if (!test.isPresent()) {
                createTest(result, classTest);
            }
            String paramName = Arrays.asList(result.getParameters()).toString();
            ExtentTest paramTest = methodTest.get().createNode(paramName);
            dataProviderTest.set(paramTest);
        } else {
            dataProviderTest.set(null);
            createTest(result, classTest);
        }

        return methodTest.get();
    }

    public static synchronized ExtentTest createMethod(ITestResult result) {
        String methodName = result.getMethod().getMethodName();
        if (result.getParameters().length > 0) {
            if (methodTest.get() != null && methodTest.get().getModel().getName().equals(methodName)) {
            } else {
                createTest(result, null);
            }
            String paramName = Arrays.asList(result.getParameters()).toString();
            ExtentTest paramTest = methodTest.get().createNode(paramName);
            dataProviderTest.set(paramTest);
        } else {
            createTest(result, null);
        }
        return methodTest.get();
    }

    private static synchronized ExtentTest createTest(ITestResult result, ExtentTest classTest) {
        String methodName = result.getMethod().getMethodName();
        ExtentTest test;
        if (classTest != null) {
            test = classTest.createNode(methodName, result.getMethod().getDescription());
        } else {
            test = ExtentService.getInstance().createTest(methodName, result.getMethod().getDescription());
        }
        methodTest.set(test);
        String[] groups = result.getMethod().getGroups();
        ExtentTestCommons.assignGroups(test, groups);
        return test;
    }

}