How to wrap this "TakesScreenshot" function into method?

Question

I am using this code to capture screenshot in selenium webdriver with JAVA:

  File screenshot = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
    
    
    try {
        FileUtils.copyFile(screenshot, new File("C:\\projectScreenshots\\homePageScreenshot.png"));
    } catch (IOException e) {
        System.out.println(e.getMessage());
    }

But when I wrap this code into function like below, then it throws me NullpointerException.

public static void TakesScreenshot()
{
            

File screenshot = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
     try {
            FileUtils.copyFile(screenshot, new File("C:\\projectScreenshots\\homePageScreenshot.png"));
        } catch (IOException e) {
            System.out.println(e.getMessage());
        }
}

How does `TakesScreenshot` (bad name for a method, by the way) know what `driver` is? Where is it declared? You could pass it as a parameter to the method. — Federico klez Culloca, Jun 24 '22 at 08:19
@FedericoklezCulloca I answered my question using your comment. Please free to edit my comment if it contains errors. — Jackie M, Jun 24 '22 at 13:18

score 0 · Accepted Answer · edited Jun 24 '22 at 18:29

I solved NullpointerException by using this below function:

public static void takeScreenshot(String file, WebDriver driver) {
    try{
        File scrFile=((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
        FileUtils.copyFile(scrFile, new File("C:\\Users\\"+file+".png"));
    } catch(IOException ioe) {
        ioe.getStackTrace();
    }
}

To call the above function, we need to use this line in the script:

takeScreenshot("Screenshot", driver);

How to wrap this "TakesScreenshot" function into method?

1 Answers1