Automation 的 ReportFlow

ReportFlow:
// click the Grid icon and switch to grid page
public void changeToGrid()
// click the Add/Locate icon in the grid page/or in the controller, and then add investment
public void addInvestmentToGrid(final String fundName, final String ticker, boolean isGridMode)
public void addInvestment(final String fundName, final String ticker, final boolean isGridMode)
// wait for the investment exists in the grid or in the controller
public void waitForInvestmentExistInGrid(final String fundName, final boolean isGridMode)
// click the ColumnManagement icon to enter the ColumnManament popover
public void openColumnSetManagement()
// click the cancel button in the Column Management popover
public void closeColumnSetManagementByCancel()
// remove the Column item in the Column Management popover and then click Done button
public void deleteColumnSet(final String columnName)
// click the posth columnset through clicking the columnset input box in the grid view
public String openColumnset(final int pos)
// click the list input box, then select and click the last investment list in Grid or in Controller
public String openRandomInvestmentList(String testCaseId, boolean isGridMode)
// click the list input box, then select and click the target list in the Grid or in the Controller
public void openInvestmentList(String listName, boolean isGridMode)
// click the checkbox to select it in the save list/column dialog after modifying the list and column
public void selectedCheckBox(WebElement element , boolean checked)
// create new investment list; this method has a wait time after clicking Enter
public void createInvestmentList(final String newName, final boolean isGridMode)

// click the Grid icon and switch to grid page
public void changeToGrid(){
    WebElement gridIcon = reportPage.getICONGrid(); //get the grid icon
    SeleniumUtil.jsClick(driver, gridIcon);
    SeleniumUtil.waitForElementVisible(driver, By.cssSelector("div#grid div#primary-header div.grid-header"),"Fail to switch to Grid."); // wait for the grid header to be visible
    SeleniumUtil.waitForElementVisible(driver, By.cssSelector("div#grid div#grid-content div.grid-view"),"Fail to switch to Grid."); // wait for the grid content/view to be visible
 }

// click the Add/Locate icon in the grid page/or in the controller, and then add investment. This method will //also wait for the investment added/existing in the grid or in the controller
public void addInvestmentToGrid(final String fundName, final String ticker, boolean isGridMode) {
    if(isGridMode){
        reportPage.getAddLocateIconInGridMode().click();
    }
    else {
        reportPage.getAddLocateIcon().click();
    }
    addInvestment(fundName, ticker, isGridMode);
}

public void addInvestment(final String fundName, final String ticker, final boolean isGridMode) {
    WebElement inputEl = reportPage.getAddLocateInput();
    inputEl.clear();
    inputEl.sendKeys(fundName);
    final ReportPage page = reportPage;
    Function<WebDriver, Boolean> waitFn = new Function<WebDriver, Boolean>() {
        @Override
        public Boolean apply(WebDriver driver) {
            try {
                List<WebElement> nameList = page.getAddLocateResultNameList();
                List<WebElement> tickerList = page.getAddLocateResultTickerList();
                int size = nameList.size();
                for(int i = 0; i < size; i++){
                    WebElement nameEl = nameList.get(i);
                    WebElement tickerEl = tickerList.get(i);
                    String nameStr = nameEl.getText().trim();
                    if(ticker == null){
                        if(nameStr.contains(fundName.trim())){
                            nameEl.click();
                            waitForInvestmentExistInGrid(nameStr, isGridMode);
                            return true;
                        }
                    }
                    else {
                        String tickerStr = tickerEl.getText().split("\\|")[0].trim();
                        if(nameStr.contains(fundName.trim()) && tickerStr.equals(ticker)){
                            nameEl.click();
                            waitForInvestmentExistInGrid(nameStr, isGridMode);
                            return true;
                        }
                    }
                }
            } catch (Exception e) {
            }
            return false;
        }
    };
    SeleniumUtil.createWait(driver).withMessage("Fail to add investment.").until(waitFn);
}

// wait for the investment exists in the grid or in the controller
public void waitForInvestmentExistInGrid(final String fundName, final boolean isGridMode) {
    final ReportPage page = reportPage;
    Function<WebDriver, Boolean> waitFn = new Function<WebDriver, Boolean>() {
        @Override
        public Boolean apply(WebDriver driver) {
            try {
                List<WebElement> list;
                if(isGridMode){
                    list = page.getInvestmentListInGridMode(); // get the investment list in the grid
                }
                else {
                    list = page.getInvestmentList(); // get the investment list in the controller
                }
                for(WebElement el : list){
                    if(el.getText().contains(fundName)){
                        return true;
                    }
                }
            } catch (Exception e) {
            }
            return false;
        }
    };
    SeleniumUtil.createWait(driver).withMessage(fundName + " is not in Grid.").until(waitFn);
}

// click the ColumnManagement icon to enter the ColumnManament popover
public void openColumnSetManagement(){
    WebElement iconColumnManagement = reportPage.getICONColumnManagement();
    SeleniumUtil.jsClick(driver, iconColumnManagement);
    SeleniumUtil.sleep(2);
}

// click the cancel button in the Column Management popover
public void closeColumnSetManagementByCancel(){
    WebElement cancleBtn = reportPage.getCancleBtn();
    SeleniumUtil.jsClick(driver, cancleBtn);
    SeleniumUtil.sleep(2);
}

// remove the Column item in the Column Management popover and then click Done button
public void deleteColumnSet(final String columnName) {
    openColumnSetManagement();  // enter into ColumnManament popover
    SeleniumUtil.sleep(2);
    List<WebElement> itemList = reportPage.getColumnManagementItemList();
    for(WebElement el : itemList){
        WebElement nameEl = el.findElement(By.cssSelector("div.name"));
        if(nameEl.getText().equals(columnName)){
            el.findElement(By.cssSelector("span.removeMe")).click();
            break;
        }
    }
    Function<WebDriver, Boolean> waitFn = new Function<WebDriver, Boolean>() {
        @Override
        public Boolean apply(WebDriver driver) {
            try{
                List<String> list = getColumnNamesFromColumnset();
                return !list.contains(columnName);
            }
            catch(Exception e){
            }
            return false;
        }
    };
    SeleniumUtil.createWait(driver).withMessage("can't delete column:" + columnName).until(waitFn);
    reportPage.getBtnDone().click();
}

// click the posth columnset through clicking the columnset input box in the grid view
public String openColumnset(final int pos) {
    WebElement columnsetEl = reportPage.getTXTColumnset();
    columnsetEl.click();
    Function<WebDriver, List<WebElement>> waitFn = new Function<WebDriver, List<WebElement>>() {
        @Override
        public List<WebElement> apply(WebDriver driver) {
            try {
                List<WebElement> columnsetList = reportPage.getDLGColumnsetList();
                if (columnsetList.size() > pos) {
                    return columnsetList;
                }
                return null;
            } catch (Exception e) {
                return null;
            }
        }
    };
    List<WebElement> list = SeleniumUtil.createWait(driver).until(waitFn);
    String columnsetName = list.get(pos).getText();
    list.get(pos).click();
    return columnsetName;
}

// click the list input box, then select and click the last investment list in Grid or in Controller
public String openRandomInvestmentList(String testCaseId, boolean isGridMode) {
    logger.info("start:openRandomInvestmentList");

    WebElement nameInputEl;
    if(isGridMode){
        nameInputEl = reportPage.getListNameDisplayInGridMode();
    }
    else {
        nameInputEl = reportPage.getListNameDisplay();
    }
    SeleniumUtil.sleep(3);
    SeleniumUtil.jsClick(driver, nameInputEl); 

    List<WebElement> list = SeleniumUtil.waitForAllElementsVisible(driver, By.cssSelector("div.popover-list div.entity-row"));
    WebElement targetEl = list.get(list.size() -1); //select the last investment list
    SeleniumUtil.scrollIntoView(driver, targetEl);
    String curListName = targetEl.getAttribute("name");
    if(!targetEl.isDisplayed()){
        SeleniumUtil.scrollIntoView(driver, targetEl);
    }
    SeleniumUtil.jsClick(driver, targetEl);
    return curListName;
}

// click the list input box, then select and click the target list in the Grid or in the Controller,
// you should pass the list name as the target
public void openInvestmentList(String listName, boolean isGridMode){
    WebElement nameInputEl;
    if(isGridMode){
        nameInputEl = reportPage.getListNameInputInGridMode();
    }
    else {
        nameInputEl = reportPage.getListNameInput();
    } 

    SeleniumUtil.sleep(3);
    if(!SeleniumUtil.isElementVisible(driver, By.cssSelector("div[name='" + listName + "']"))){
        if(SeleniumUtil.getBrowserName(driver).equals("firefox")){
            SeleniumUtil.waitForPageToLoad(driver);
            JavascriptExecutor js = ((JavascriptExecutor) driver);
            js.executeScript("$('div.controller-list>input').click();");
        }else{
            SeleniumUtil.jsClick(driver, nameInputEl);
        }
    }
    WebElement targetEl = SeleniumUtil.waitForElementVisible(driver, By.cssSelector("div[name='" + listName + "']"));
    targetEl.click();
}

// click the checkbox to select it in the save list/column dialog after modifying the list and column
public void selectedCheckBox(WebElement element , boolean checked){
    String classValue = element.getAttribute("class");
    if(classValue.contains("selected") && checked){
        return;
    }
    else{
        element.click();
    }
}

// create new investment list; this method has a wait time after clicking Enter
public void createInvestmentList(final String newName, final boolean isGridMode) {
    WebElement saveEl = reportPage.getBTNSave();
    SeleniumUtil.jsClick(driver, saveEl);
    SeleniumUtil.sleep(2);
    WebElement nameInputEl;
    if(isGridMode){
        nameInputEl = reportPage.getListNameInputInGridMode();
    }
    else {
        nameInputEl = reportPage.getListNameInput();
    }

    nameInputEl.clear();
    nameInputEl.sendKeys(newName);
    nameInputEl.sendKeys(Keys.ENTER);

    Function<WebDriver, Boolean> waitFn = new Function<WebDriver, Boolean>() {
    @Override
        public Boolean apply(WebDriver driver) {
            try {
                WebElement nameInputEl;
                if(isGridMode){
                    nameInputEl = reportPage.getListNameInputInGridMode();
                }
                else {
                    nameInputEl = reportPage.getListNameInput();
                }
                return newName.equals(nameInputEl.getAttribute("value"));
            } catch (Exception e) {
                return false;
            }
        }
    };
    // wait for the list is added successfully
    SeleniumUtil.createWait(driver).withMessage("Fail to input name.").until(waitFn);
    // wait for the message displayed
    SeleniumUtil.waitForElementNotVisible(driver, By.cssSelector("span.message-content"));
}

------------------------------------------------
ReportPage

// the input box in the top-left of the grid view
public WebElement getListNameInputInGridMode(); 

// the list name showing in the input box in the top-left of the grid view
public WebElement getListNameDisplayInGridMode()

// the differences between getListNameInputInGridMode and getListNameDisplayInGridMode are:
// getListNameInputInGridMode().value == getListNameDisplayInGridMode()