/*
 * (C) Copyright 2017 Boni Garcia (http://bonigarcia.github.io/)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */
package io.github.bonigarcia.test.screenshot;

import static io.github.bonigarcia.SeleniumJupiter.getString;
import static java.lang.System.clearProperty;
import static java.lang.System.setProperty;
import static org.hamcrest.CoreMatchers.containsString;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.junit.jupiter.api.TestInstance.Lifecycle.PER_CLASS;

import java.io.File;

import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.TestInstance;
import org.junit.jupiter.api.extension.ExtendWith;
import org.openqa.selenium.chrome.ChromeDriver;

import io.github.bonigarcia.SeleniumExtension;

@ExtendWith(SeleniumExtension.class)
@TestInstance(PER_CLASS)
public class ScreenshotPngTest {

    File imageName;

    @BeforeAll
    void setup() {
        setProperty("sel.jup.screenshot.at.the.end.of.tests", "true");
        setProperty("sel.jup.screenshot.format", "base64andpng");
    }

    @AfterAll
    void teardown() {
        setProperty("sel.jup.screenshot.format", "png");
        clearProperty("sel.jup.screenshot.at.the.end.of.tests");
        assertTrue(imageName.exists());
        imageName.delete();
    }

    @Test
    void screenshotTest(ChromeDriver driver) {
        driver.get("https://bonigarcia.github.io/selenium-jupiter/");
        assertThat(driver.getTitle(),
                containsString("A JUnit 5 extension for Selenium WebDriver"));

        imageName = new File(getString("sel.jup.output.folder"),
                driver.getSessionId() + ".png");
    }

}

@@ -675,70 +675,8 @@
 t;%0A%0A
-import static io.github.bonigarcia.SeleniumJupiter.getString;%0A
 impo
@@ -1611,24 +1611,90 @@
 64andpng%22);%0A
+        setProperty(%22sel.jup.output.folder%22, %22surefire-reports%22);%0A
     %7D%0A%0A    @
@@ -1724,35 +1724,37 @@
 own() %7B%0A        
-set
+clear
 Property(%22sel.ju
@@ -1770,21 +1770,27 @@
 hot.
-format%22, %22png
+at.the.end.of.tests
 %22);%0A
@@ -1789,37 +1789,35 @@
 ests%22);%0A        
-clear
+set
 Property(%22sel.ju
@@ -1829,35 +1829,80 @@
 eenshot.
-at.the.end.of.tests
+format%22, %22png%22);%0A        setProperty(%22sel.jup.output.folder%22, %22.
 %22);%0A    
@@ -2250,42 +2250,107 @@
 ile(
-getString(%22sel.jup.output.folder%22)
+%0A                %22./target/surefire-reports/io.github.bonigarcia.test.screenshot.ScreenshotPngTest%22
 ,%0A  

Enhance screenshot test