Ejemplo n.º 1
0
  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;
    }
  }