package core.listener;

import core.annotation_handler.CoreStepPageHandler;
import core.services.SelenideImplPage;
import core.utils.ThreadLocalPageContainer;
import org.testng.ITestContext;
import org.testng.ITestListener;
import org.testng.ITestResult;

public class TestListener implements ITestListener {

    @Override
    public void onTestStart(ITestResult result) {
        initCurrentTestPage(result);
    }

    @Override
    public void onTestSuccess(ITestResult result) {

    }

    @Override
    public void onTestFailure(ITestResult result) {
        ThreadLocalPageContainer.getThreadLocalPage().attachScreenShot();
    }

    @Override
    public void onTestSkipped(ITestResult result) {

    }

    @Override
    public void onTestFailedButWithinSuccessPercentage(ITestResult result) {
    }


    @Override
    public void onStart(ITestContext context) {

    }

    @Override
    public void onFinish(ITestContext context) {
        ThreadLocalPageContainer.getThreadLocalPage().quitWebDriver();
    }

    private void initCurrentTestPage(ITestResult result){
        Object currentTest = result.getInstance();
        SelenideImplPage page = CoreStepPageHandler.handle(currentTest);
        ThreadLocalPageContainer.setThreadLocalPage(page);
        page.setTestNameLogger(result.getName().toUpperCase());
    }

}