Selenium在失败时采取屏幕

Question

您好，我使用selelium和spock编写烟雾测试，我想截取测试失败的截图。尝试这样的：

public class ScreenshotTestRule implements MethodRule { 
    public Statement apply(final Statement statement, final FrameworkMethod frameworkMethod, final Object o) { 
     return new Statement() { 
      @Override 
      public void evaluate() throws Throwable { 
       try { 
        statement.evaluate() 
       } catch (Throwable t) { 
        captureScreenshot(frameworkMethod.getName().replaceAll(" ", "-")) 
        throw t 
       } 
      } 

      public void captureScreenshot(String fileName) { 
       try { 
        new File("target/surefire-reports/").mkdirs() 
        File screenshot = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE) 
        Files.copy(screenshot, new File("/target/surefire-reports/"+fileName+".png")) 
       } catch (Exception e) { 
        print "Error while creating screenshot " + fileName 
        throw new RuntimeException(e) 
       } 
      } 
     } 
    } 
} 

，但我收到此错误：

java.lang.RuntimeException: groovy.lang.MissingFieldException: No such field: driver for class: lt.inventi.apollo.system.test.SmokeTest$ScreenshotTestRule 

Answer 1

private class ScreenshotTestRule implements MethodRule { 

    WebDriver testDriver 

    public ScreenshotTestRule(WebDriver driver) { 
     this.testDriver = driver 
    } 

    public Statement apply(final Statement statement, final FrameworkMethod frameworkMethod, final Object o) { 
     return new Statement() { 
      @Override 
      public void evaluate() throws Throwable { 
       try { 
        statement.evaluate() 
       } catch (Throwable t) { 
        captureScreenshot(frameworkMethod.getName().replaceAll(" ", "-")) 
        throw t 
       } 
      } 

      public void captureScreenshot(String fileName) { 
       try { 
        new File("/target/test-screenshots/").mkdirs() 
        File screenshot = ((TakesScreenshot) testDriver).getScreenshotAs(OutputType.FILE) 
        String imagePath = "target/test-screenshots/" + fileName + "-screenshot.png"; 
        Files.copy(screenshot, new File(imagePath)) 
       } catch (Exception e) { 
        print "Error while creating screenshot " + fileName 
       } 
      } 
     } 
    } 
} 

简单的通过它扔构造