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

Introduction

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

Source

/*
 * WorkbenchTests.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.io.File;

import org.junit.AfterClass;
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.ExpectedConditions;
import org.openqa.selenium.support.ui.WebDriverWait;

public class WorkbenchTests {
    @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 testWorkbenchPersistance() throws Exception {
        clearWorkspace();

        // Add a variable to the workspace
        (new Actions(driver_)).sendKeys(Keys.ESCAPE + "selenium <- function() { 42 }" + Keys.ENTER).perform();

        // Save the workspace 
        WebElement saveItem = MenuNavigator.getMenuItem(driver_, "Session", "Save Workspace As...");
        saveItem.click();

        WebElement saveDialog = DialogTestUtils.waitForModalToAppear(driver_);
        DialogTestUtils.waitForFocusedInput(driver_, saveDialog);

        File tempFile = File.createTempFile("rstudio-selenium-workspace-", ".RData");
        String workspaceFilePath = tempFile.getAbsolutePath();
        tempFile.delete();

        (new Actions(driver_)).sendKeys(workspaceFilePath + Keys.ENTER).perform();

        DialogTestUtils.waitForModalToDisappear(driver_);

        // Clear the workspace and load it again
        clearWorkspace();

        WebElement loadItem = MenuNavigator.getMenuItem(driver_, "Session", "Load Workspace...");
        loadItem.click();

        WebElement loadDialog = DialogTestUtils.waitForModalToAppear(driver_);
        DialogTestUtils.waitForFocusedInput(driver_, loadDialog);

        (new Actions(driver_)).sendKeys(workspaceFilePath + Keys.ENTER).perform();

        DialogTestUtils.waitForModalToDisappear(driver_);

        // List the workspace and see if the variable is present
        ConsoleTestUtils.beginConsoleInteraction(driver_);
        (new Actions(driver_)).sendKeys(Keys.chord(Keys.CONTROL + "l")).sendKeys(Keys.ESCAPE + "ls()" + Keys.ENTER)
                .perform();

        ConsoleTestUtils.waitForConsoleContainsText(driver_, "selenium");
    }

    private void clearWorkspace() {
        // Clear out the workspace and make sure it's clear
        WebElement menuItem = MenuNavigator.getMenuItem(driver_, "Session", "Clear Workspace...");
        menuItem.click();
        DialogTestUtils.respondToModalDialog(driver_, "Yes");

        ConsoleTestUtils.beginConsoleInteraction(driver_);
        (new Actions(driver_)).sendKeys(Keys.ESCAPE + "ls()" + Keys.ENTER).perform();

        ConsoleTestUtils.waitForConsoleContainsText(driver_, "character(0)");
    }

    private static WebDriver driver_;
}