public static List<Image> getPictureElements( WebDriver driver, WebElement webElementSet, int componentNo) { List<WebElement> elements = new ArrayList<WebElement>(); for (String selector : imageCssLocators) { elements.addAll( WebPageHelpers.createWebElementsList(driver, webElementSet, By.cssSelector(selector))); } // Add to the list if the webElementSet is an image it's self if (webElementSet.getTagName().equals("img")) { elements.add(0, webElementSet); } List<Image> Images = new ArrayList<Image>(); // Iterate elements list to populate imageBeans list ImageBeans if (WebPageHelpers.testWebElementListNotEmpty(elements)) { for (WebElement element : elements) { Image image = new Image( element.getAttribute("src"), element.getAttribute("width"), element.getAttribute("height")); Images.add(image); } } if (Images.isEmpty()) { return null; } else { return Images; } }