org.rstudio.studio.selenium.SourceInteraction.java Source code

Introduction

Here is the source code for org.rstudio.studio.selenium.SourceInteraction.java

Source

/*
 * SourceInteraction.java
 *
 * Copyright (C) 2009-13 by RStudio, Inc.
 *
 * Unless you have received this program directly from RStudio pursuant
 * to the terms of a commercial license agreement with RStudio, then
 * this program is licensed to you under the terms of version 3 of the
 * GNU Affero General Public License. This program is distributed WITHOUT
 * ANY EXPRESS OR IMPLIED WARRANTY, INCLUDING THOSE OF NON-INFRINGEMENT,
 * MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE. Please refer to the
 * AGPL (http://www.gnu.org/licenses/agpl-3.0.txt) for more details.
 *
 */
package org.rstudio.studio.selenium;

import java.util.List;

import org.junit.AfterClass;
import org.junit.Assert;
import org.junit.BeforeClass;
import org.junit.Test;
import org.openqa.selenium.By;
import org.openqa.selenium.Keys;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.interactions.Actions;
import org.openqa.selenium.support.ui.ExpectedCondition;
import org.openqa.selenium.support.ui.ExpectedConditions;
import org.openqa.selenium.support.ui.WebDriverWait;
import org.rstudio.core.client.ElementIds;

public class SourceInteraction {
    @BeforeClass
    public static void setUpBeforeClass() throws Exception {
        driver_ = RStudioWebAppDriver.start();

        (new WebDriverWait(driver_, 10))
                .until(ExpectedConditions.presenceOfElementLocated(By.className("gwt-MenuBar")));
    }

    @AfterClass
    public static void tearDownAfterClass() throws Exception {
        RStudioWebAppDriver.stop();
    }

    @Test
    public void createAndSourceRFile() {
        createRFile();

        // Type some code into the file. Note that the matching brace is auto 
        // completed.
        Actions a = new Actions(driver_);
        a.sendKeys("f <- function() {" + Keys.ENTER);
        a.sendKeys(Keys.TAB + "42");
        a.perform();

        // Source the entire file
        WebElement sourceMenuEntry = MenuNavigator.getMenuItem(driver_, "Code", "Source");
        sourceMenuEntry.click();

        // Wait for the console to contain the string "source"
        ConsoleTestUtils.waitForConsoleContainsText(driver_, "source(");

        closeUnsavedRFile();
    }

    @Test
    public void findAndReplace() {
        createRFile();

        // Type some code into the file
        String preReplaceCode = "foo <- 'bar'";
        Actions a = new Actions(driver_);
        a.sendKeys(preReplaceCode + Keys.ENTER);
        a.perform();

        // Find the ACE editor instance that the code appears in. (CONSIDER: 
        // This is not the best way to find the code editor instance.)
        WebElement editor = null;
        List<WebElement> editors = driver_.findElements(By.className("ace_content"));
        for (WebElement e : editors) {
            if (e.getText().contains(preReplaceCode)) {
                editor = e;
                break;
            }
        }
        Assert.assertNotNull(editor);

        // Invoke find and replace
        WebElement findMenuEntry = MenuNavigator.getMenuItem(driver_, "Edit", "Find...");
        findMenuEntry.click();

        // Wait for the find and replace panel to come up
        (new WebDriverWait(driver_, 2)).until(ExpectedConditions
                .presenceOfElementLocated(By.id(ElementIds.getElementId(ElementIds.FIND_REPLACE_BAR))));

        // Type the text and the text to be replaced (replace 'bar' with 'foo')
        Actions rep = new Actions(driver_);
        rep.sendKeys("bar" + Keys.TAB + "foo" + Keys.ENTER);
        rep.perform();

        DialogTestUtils.respondToModalDialog(driver_, "OK");

        Actions dismiss = new Actions(driver_);
        dismiss.sendKeys(Keys.ESCAPE);
        dismiss.perform();

        // Ensure that the source has been updated
        Assert.assertTrue(editor.getText().contains("foo <- 'foo'"));

        closeUnsavedRFile();
    }

    private void createRFile() {
        WebElement newRScriptMenuEntry = MenuNavigator.getMenuItem(driver_, "File", "New File", "R Script");
        newRScriptMenuEntry.click();

        // Wait for the "Untitled" buffer to appear
        (new WebDriverWait(driver_, 5)).until(new ExpectedCondition<Boolean>() {
            public Boolean apply(WebDriver d) {
                List<WebElement> elements = driver_.findElements(By.className("gwt-TabLayoutPanelTab-selected"));
                for (WebElement e : elements) {
                    if (e.getText().startsWith("Untitled")) {
                        return true;
                    }
                }
                return false;
            }
        });
    }

    private void closeUnsavedRFile() {
        WebElement closeEntry = MenuNavigator.getMenuItem(driver_, "File", "Close");
        closeEntry.click();
        DialogTestUtils.respondToModalDialog(driver_, "Don't Save");
    }

    private static WebDriver driver_;
}