net.serenitybdd.core.photography.SoundEngineer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of serenity-core Show documentation

There is a newer version: 4.2.8

package net.serenitybdd.core.photography;

import net.thucydides.model.domain.TestResult;
import net.thucydides.model.util.EnvironmentVariables;
import org.openqa.selenium.WebDriver;

import static net.serenitybdd.core.photography.StoreHTML.ALWAYS;
import static net.serenitybdd.core.photography.StoreHTML.NEVER;
import static net.thucydides.model.ThucydidesSystemProperty.SERENITY_STORE_HTML;

public class SoundEngineer {

    private final EnvironmentVariables environmentVariables;
    private boolean recordPageSource = true;

    public SoundEngineer(EnvironmentVariables environmentVariables) {
        this.environmentVariables = environmentVariables;
    }

    public SoundEngineer ifRequiredForResult(TestResult result) {
        StoreHTML storeHTML = StoreHTML.valueOf(SERENITY_STORE_HTML.optionalFrom(environmentVariables).orElse("FAILURES"));
        if (storeHTML == NEVER) {
            recordPageSource = false;
        } else if (storeHTML == ALWAYS) {
            recordPageSource = true;
        } else {
            recordPageSource = (result == TestResult.FAILURE || result == TestResult.ERROR);
        }
        return this;
    }

    public PageSourceRecorder recordPageSourceUsing(WebDriver driver) {
        return (recordPageSource) ? new PageSourceRecorder(driver) : new DisabledPageSourceRecorder(driver);
    }
}