如何使用Cucumber和JUnit获取屏幕截图并将其附加到Allure报告中?

时间:2017-04-10 04:57:35

标签: java selenium junit cucumber

我在自动化测试项目中使用Cucumber,Selenium,Java,Maven和JUnit堆栈。

目标是在失败和破坏的测试中截取屏幕截图。我找到了Java / Maven / JUnit堆栈的解决方案:

MyIntA<MyIntB>

但是,当然它在使用Cucumber时不起作用,因为它不使用任何@Test方法。

所以,我决定将@Rule更改为@ClassRule,让它听取任何失败,所以这里是:

@Rule
public TestWatcher screenshotOnFailure = new TestWatcher() {
    @Override
    protected void failed(Throwable e, Description description) {
        makeScreenshotOnFailure();
    }

    @Attachment("Screenshot on failure")
    public byte[] makeScreenshotOnFailure() {
        return ((TakesScreenshot) driver).getScreenshotAs(OutputType.BYTES);
    }
};

这个解决方案对我没有帮助。

所以,问题是:“当我在我的测试项目中使用Java / Selenium / Cucumber / JUnit / Maven时,如何在失败时附加屏幕截图?”

3 个答案:

答案 0 :(得分:1)

解决方案只是将以下代码添加到定义类中:

@After
public void embedScreenshot(Scenario scenario) {
    if (scenario.isFailed()) {
        try {
            byte[] screenshot = ((TakesScreenshot) Application.getInstance().getWebDriver())
                    .getScreenshotAs(OutputType.BYTES);
            scenario.embed(screenshot, "image/png");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

答案 1 :(得分:0)

在GlobalGlue中

public class GlobalGlue {

  @Before
  public void before(Scenario scenario) throws Exception {
    CONTEXT.setScenario(scenario);
  }

  @After
  public void after() {
    WebDriverUtility.after(getDriver(), CONTEXT.getScenario());
  }

}

创建另一个类WebDriverUtility,并使用add方法:

public static void after(WebDriver driver, Scenario scenario) {
  getScreenshot(driver, scenario);
  driver.close();
}

public static void getScreenshot(WebDriver driver, Scenario scenario) {
if (scenario.isFailed()) {
  final byte[] screenshot = ((TakesScreenshot) driver).getScreenshotAs(OutputType.BYTES);
  scenario.embed(screenshot, "image/png");
  log.info("Thread: " + Thread.currentThread().getId() + " :: "
      + "Screenshot taken and inserted in scenario report");
}

}

主要部分是你需要在场景失败的情况下嵌入屏幕截图:

 final byte[] screenshot = ((TakesScreenshot) driver).getScreenshotAs(OutputType.BYTES);
 scenario.embed(screenshot, "image/png");

ExecutionContext.java

import com.google.common.cache.CacheBuilder;
import com.google.common.cache.CacheLoader;
import com.google.common.cache.LoadingCache;
import cucumber.api.Scenario;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.openqa.selenium.WebDriver;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ExecutionException;

/**
 * Maintains one webdriver per scenario and one scenario per thread.
 * Can be used for parallel execution.
 * Assumes that scenarios within one feature are not parallel.
 * Can be rewritten using <code>ThreadLocal</code>.
 *
 * @author dkhvatov
 */
public enum ExecutionContext {

  CONTEXT;

  private static final Logger log = LogManager.getLogger(ExecutionContext.class);

  private final LoadingCache<Scenario, WebDriver> webDrivers =
    CacheBuilder.newBuilder()
      .build(CacheLoader.from(scenario ->
        WebDriverUtility.createDriver()));

  private final Map<String, Scenario> scenarios = new ConcurrentHashMap<>();


  /**
   * Lazily gets a webdriver for the current scenario.
   *
   * @return webdriver
   */
  public WebDriver getDriver() {
    try {
      Scenario scenario = getScenario();
      if (scenario == null) {
        throw new IllegalStateException("Scenario is not set for context. " +
          "Please verify your glue settings. Either use GlobalGlue, or set " +
          "scenario manually: CONTEXT.setScenario(scenario)");
      }
      return webDrivers.get(scenario);
    } catch (ExecutionException e) {
      log.error("Unable to start webdriver", e);
      throw new RuntimeException(e);
    }
  }

  /**
   * Gets scenario for a current thread.
   *
   * @return scenario
   */
  public Scenario getScenario() {
    return scenarios.get(Thread.currentThread().getName());
  }

  /**
   * Sets current scenario. Overwrites current scenario in a map.
   *
   * @param scenario scenario
   */
  public void setScenario(Scenario scenario) {
    scenarios.put(Thread.currentThread().getName(), scenario);
  }

}

答案 2 :(得分:0)

这样,您可以将屏幕附加到您的吸引力报告中。 还可以在pom文件中使用最新的诱惑版本

768

Pom文件:

import com.google.common.io.Files;
import io.qameta.allure.Attachment;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import java.io.File;
import java.io.IOException;

public class ScreenshotUtils {
    @Attachment(type = "image/png")
    public static byte[] screenshot(WebDriver driver)/* throws IOException */ {
        try {
            File screen = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
            return Files.toByteArray(screen);
        } catch (IOException e) {
            return null;
        }
    }
}
相关问题