com.denimgroup.threadfix.selenium.pages.TeamDetailPage.java Source code

Introduction

Here is the source code for com.denimgroup.threadfix.selenium.pages.TeamDetailPage.java

Source

////////////////////////////////////////////////////////////////////////
//
//     Copyright (c) 2009-2013 Denim Group, Ltd.
//
//     The contents of this file are subject to the Mozilla Public License
//     Version 2.0 (the "License"); you may not use this file except in
//     compliance with the License. You may obtain a copy of the License at
//     http://www.mozilla.org/MPL/
//
//     Software distributed under the License is distributed on an "AS IS"
//     basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
//     License for the specific language governing rights and limitations
//     under the License.
//
//     The Original Code is ThreadFix.
//
//     The Initial Developer of the Original Code is Denim Group, Ltd.
//     Portions created by Denim Group, Ltd. are Copyright (C)
//     Denim Group, Ltd. All Rights Reserved.
//
//     Contributor(s): Denim Group, Ltd.
//
////////////////////////////////////////////////////////////////////////
package com.denimgroup.threadfix.selenium.pages;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.openqa.selenium.Alert;
import org.openqa.selenium.By;
import org.openqa.selenium.TimeoutException;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;

public class TeamDetailPage extends BasePage {

    //   private WebElement orgName;
    private WebElement applicationsTableBody;
    private WebElement lastItemFoundInApplicationsTableBodyLink;
    private WebElement addApplicationLink;

    public TeamDetailPage(WebDriver webdriver) {
        super(webdriver);

        //      orgName = driver.findElementById("name");
        applicationsTableBody = driver.findElementById("applicationsTableBody");
        addApplicationLink = driver.findElementByLinkText("Add Application");
    }

    public String getOrgName() {
        return driver.findElementById("name").getText();
    }

    public TeamDetailPage clickEditOrganizationLink() {
        driver.findElementById("teamModalButton").click();
        waitForElement(driver.findElementById("teamModal"));
        return new TeamDetailPage(driver);
    }

    public TeamDetailPage setNameInput(String editedOrgName) {
        driver.findElementById("teamNameInput").clear();
        driver.findElementById("teamNameInput").sendKeys(editedOrgName);
        return new TeamDetailPage(driver);
    }

    public TeamDetailPage clickUpdateButtonValid() {
        driver.findElementById("submitTeamModal").click();
        try {
            waitForInvisibleElement(driver.findElementById("teamModal"));
        } catch (TimeoutException e) {
            driver.findElementById("submitTeamModal").click();
        }
        return new TeamDetailPage(driver);
    }

    public TeamDetailPage clickUpdateButtonInvalid() {
        driver.findElementById("submitTeamModal").click();
        return new TeamDetailPage(driver);
    }

    public TeamDetailPage clickShowMore() {
        driver.findElementById("showDetailsLink").click();
        return new TeamDetailPage(driver);
    }

    public TeamIndexPage clickDeleteButton() {
        clickEditOrganizationLink();
        sleep(500);
        driver.findElementById("deleteLink").click();

        Alert alert = driver.switchTo().alert();
        alert.accept();

        return new TeamIndexPage(driver);
    }

    public boolean isTextPresentInApplicationsTableBody(String text) {
        for (WebElement element : applicationsTableBody.findElements(By.xpath(".//tr/td/a"))) {
            if (element.getText().contains(text)) {
                lastItemFoundInApplicationsTableBodyLink = element;
                return true;
            }
        }
        return false;
    }

    public TeamIndexPage clickExpandAll() {
        driver.findElementById("expandAllButton").click();
        return new TeamIndexPage(driver);
    }

    public TeamIndexPage clickCollapseAll() {
        driver.findElementById("collapseAllButton").click();
        return new TeamIndexPage(driver);
    }

    public ApplicationDetailPage clickTextLinkInApplicationsTableBody(String text) {
        if (isTextPresentInApplicationsTableBody(text)) {
            lastItemFoundInApplicationsTableBodyLink.click();
            return new ApplicationDetailPage(driver);
        } else {
            return null;
        }
    }

    public Map<String, Integer> getVulnCountForApps() {
        Map<String, Integer> map = new HashMap<String, Integer>();

        // get app names
        List<WebElement> appLinks = applicationsTableBody.findElements(By.xpath(".//tr/td/a"));
        List<WebElement> counts = applicationsTableBody.findElements(By.id("vulnCountCell"));

        if (appLinks.size() != counts.size()) {
            return null;
        }

        for (int i = 0; i < appLinks.size(); i++) {
            try {
                map.put(appLinks.get(i).getText(), Integer.valueOf(counts.get(i).getText()));
            } catch (NumberFormatException e) {
                e.printStackTrace();
            }
        }

        return map;
    }

    public ApplicationAddPage clickAddApplicationLink() {
        addApplicationLink.click();
        return new ApplicationAddPage(driver);
    }

    public String getErrorText() {
        return driver.findElementById("name.errors").getText().trim();
    }

}