selenium - webdriver を使用して画像をキャプチャするソリューション。これに対するより良い解決策はありますか？

Question

Selenium Webdriver のすべてのタグのスクリーン ショットをキャプチャするコードを共有したい

1. リスト内のすべての<img>タグを取得

   List<WebElement> image = driver.findElements(By.tagName("img"));
   int i=1;
   

2. これを使って：

   for(WebElement allImages : image){
   
       //Download image
       File screen=((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
       int width=allImages.getSize().getWidth();
       int height=allImages.getSize().getHeight();
       BufferedImage img=ImageIO.read(screen);
       BufferedImage dest=img.getSubimage(allImages.getLocation().getX(), allImages.getLocation().getY(), width, height);
       ImageIO.write(dest, "png", screen); 
       File file=new File("/Users/h-disk/Desktop/folder-name/abc"+i+".png"); 
   

このコードを使用すると、すべて<img>のをキャプチャできます