net.serenitybdd.screenplay.playwright.Photographer Maven / Gradle / Ivy

Go to download

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

There is a newer version: 4.2.9

Show newest version

package net.serenitybdd.screenplay.playwright;

import com.microsoft.playwright.Page;
import net.thucydides.core.steps.BaseStepListener;
import net.thucydides.core.steps.StepEventBus;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;

public class Photographer {
    public File takesAScreenshot(Page currentPage) {
        BaseStepListener baseStepListener = StepEventBus.getParallelEventBus().getBaseStepListener();

        byte[] screenshot = currentPage.screenshot(new Page.ScreenshotOptions().setFullPage(true));

        try {
            Path outputDirectory = baseStepListener.getOutputDirectory().toPath();
            Path screenshotFile = Files.createTempFile(outputDirectory, "screenshot", ".png");
            Files.write(screenshotFile, screenshot);

            return screenshotFile.toFile();
        } catch (IOException e) {
            throw new AssertionError("Failed to take Playwright screenshot", e);
        }
    }
}