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

Introduction

Here is the source code for com.denimgroup.threadfix.selenium.pages.ApiKeysIndexPage.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.ArrayList;
import java.util.List;

import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;

public class ApiKeysIndexPage extends BasePage {

    private List<WebElement> notes = new ArrayList<WebElement>();
    //private List<WebElement> restrictedBoxes = new ArrayList<WebElement>();
    private WebElement createNewKeyLink;

    public ApiKeysIndexPage(WebDriver webdriver) {
        super(webdriver);
        createNewKeyLink = driver.findElementByLinkText("Create New Key");
        for (int i = 1; i <= getNumRows(); i++) {
            notes.add(driver.findElementById("note" + i));
        }
    }

    public int getNumRows() {
        List<WebElement> bodyRows = driver.findElementsByClassName("bodyRow");

        if (bodyRows != null && bodyRows.size() == 1 && bodyRows.get(0).getText().contains("No keys found.")) {
            return 0;
        }

        return driver.findElementsByClassName("bodyRow").size();
    }

    public int getIndex(String roleName) {
        int i = -1;
        for (WebElement note : notes) {
            i++;
            String text = note.getText().trim();
            if (text.equals(roleName.trim())) {
                return i;
            }
        }
        return -1;
    }

    public String getKeyText(String note) {
        return driver.findElementById("key" + getIndex(note)).getText();
    }

    public ApiKeysIndexPage clickEdit(String note) {
        driver.findElementById("editKey" + (getIndex(note) + 1)).click();
        waitForElement(driver.findElementsByClassName("modal").get(getIndex(note)));
        return new ApiKeysIndexPage(driver);
    }

    public ApiKeysIndexPage clickNewLink() {
        createNewKeyLink.click();
        waitForElement(driver.findElementById("newKeyModalDiv"));
        return new ApiKeysIndexPage(driver);
    }

    public ApiKeysIndexPage clickDelete(String note) {
        clickEdit(note);
        driver.findElementsById("deleteButton").get(getIndex(note)).click();
        handleAlert();

        return new ApiKeysIndexPage(driver);
    }

    public ApiKeysIndexPage clickSubmitButton(String oldNote) {
        int oldCnt = getNumRows();
        int timer = 0;
        if (oldNote == null) {
            driver.findElementById("submitKeyModalCreate").click();
            while (getNumRows() != (oldCnt + 1)) {
                if (timer >= 10) {
                    break;
                }
                timer++;
                sleep(100);
            }
        } else {
            driver.findElementsById("submitKeyModal").get(getIndex(oldNote)).click();
            sleep(1000);
        }
        return new ApiKeysIndexPage(driver);
    }

    public ApiKeysIndexPage setNote(String newNote, String oldNote) {
        if (oldNote == null) {
            driver.findElementsById("note").get(getNumRows()).clear();
            driver.findElementsById("note").get(getNumRows()).sendKeys(newNote);
        } else {
            driver.findElementsById("note").get(getIndex(oldNote)).clear();
            driver.findElementsById("note").get(getIndex(oldNote)).sendKeys(newNote);
        }
        return this;
    }

    public ApiKeysIndexPage setRestricted(String oldNote) {
        if (oldNote == null) {
            driver.findElementById("isRestrictedKey" + (getNumRows() + 1)).click();
        } else {
            driver.findElementById("isRestrictedKey" + (getIndex(oldNote) + 1)).click();
        }
        return new ApiKeysIndexPage(driver);
    }

    public ApiKeysIndexPage waitModalDisappear() {
        waitForInvisibleElement(driver.findElementById("newKeyModalDiv"));
        return new ApiKeysIndexPage(driver);
    }

    public boolean isCreateValidationPresent() {
        return driver.findElementByClassName("alert-success").getText()
                .contains("API key was successfully created.");
    }

    public boolean isEditValidationPresent() {
        return driver.findElementByClassName("alert-success").getText()
                .contains("API key was successfully edited.");
    }

    public boolean isDeleteValidationPresent() {
        return driver.findElementByClassName("alert-success").getText()
                .contains("API key was successfully deleted.");
    }

    public boolean isNotePresent(String note) {
        return getIndex(note) != -1;
    }

    public boolean isRestricted(String note) {
        return driver.findElementById("restricted" + (getIndex(note) + 1)).getText().trim().contains("true");
    }

    public boolean isCorrectLength(String note) {
        return notes.get(getIndex(note)).getText().trim().length() <= 255;
    }

}