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

Introduction

Here is the source code for com.denimgroup.threadfix.selenium.pages.WafRulesPage.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 org.openqa.selenium.Alert;
import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.support.ui.Select;

public class WafRulesPage extends BasePage {

    private WebElement nameText;
    private WebElement wafTypeText;
    private WebElement editLink;
    private WebElement lastItemFoundInApplicationsTableBodyLink;
    private WebElement deleteButton;

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

        nameText = driver.findElementById("nameText");
        wafTypeText = driver.findElementById("wafTypeText");
    }

    public String getNameText() {
        return nameText.getText();
    }

    public String getWafTypeText() {
        return wafTypeText.getText();
    }

    public WafEditPage clickEditLink() {
        editLink.click();
        return new WafEditPage(driver);
    }

    public WafRulesPage clickGenerateWafRulesButton() {
        driver.findElementById("generateWafRulesButton").click();
        return new WafRulesPage(driver);
    }

    public TeamIndexPage clickDownloadWafRulesButton() {
        driver.findElementById("downloadWafRulesButton").click();
        return new TeamIndexPage(driver);
    }

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

        return false;
    }

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

    public WafIndexPage clickDeleteButton() {
        deleteButton.click();

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

        return new WafIndexPage(driver);
    }

    public WafRulesPage clickDeleteButtonInvalid() {
        deleteButton.click();

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

        return new WafRulesPage(driver);
    }

    public String getWafDirectiveSelect() {
        return new Select(driver.findElementById("wafDirectiveSelect")).getFirstSelectedOption().getText();
    }

    public WafRulesPage setWafDirectiveSelect(String code) {
        new Select(driver.findElementById("wafDirectiveSelect")).selectByVisibleText(code);
        return new WafRulesPage(driver);
    }

}