beakerx.po.js

/*
 *  Copyright 2017 TWO SIGMA OPEN SOURCE, LLC
 *
 *  Licensed under the Apache 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.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */

var BeakerXPageObject = function () {

  this.baseDocURL = 'http://127.0.0.1:8888/tree/doc/contents';
  this.baseTestURL = 'http://127.0.0.1:8888/tree/test/notebooks';
  this.kernelIdleIcon = $('i.kernel_idle_icon');

  this.runNotebookByName = function(name, done, subDir){
    browser
      .url(subDir === undefined ? this.baseDocURL : this.baseDocURL + '/' + subDir)
      .call(done);
    browser.waitForEnabled('=' + name);
    browser.click('=' + name);
    browser.window(browser.windowHandles().value[1]);
  }

  this.runNotebookByUrl = function(url){
    browser.url('http://127.0.0.1:8888' + url);
    this.kernelIdleIcon.waitForEnabled();
  }

  this.clickRunCell = function () {
    var cssRunCell = 'button[data-jupyter-action="jupyter-notebook:run-cell-and-select-next"]';
    browser.waitForEnabled(cssRunCell);
    browser.click(cssRunCell);
    this.kernelIdleIcon.waitForEnabled();
  }

  this.clickSaveNotebook = function () {
    browser.click('button[data-jupyter-action="jupyter-notebook:save-notebook"]');
  }

  this.closeAndHaltNotebook = function () {
    this.clickSaveNotebook();
    browser.click('=File');
    browser.waitForEnabled('=Close and Halt');
    browser.click('=Close and Halt');
    browser.endAll();
  }

  this.getCodeCellByIndex = function (index) {
    return $$('div.code_cell')[index];
  }

  this.runCodeCellByIndex = function (index) {
    var codeCell = this.getCodeCellByIndex(index);
    codeCell.scroll();
    codeCell.click();
    this.clickRunCell();
    this.kernelIdleIcon.waitForEnabled();
    return codeCell;
  }

  this.runCellToGetDtContainer = function(index){
    this.kernelIdleIcon.waitForEnabled();
    var codeCell = this.runCodeCellByIndex(index);
    return codeCell.$('div.dtcontainer');
  }

  this.runCellToGetSvgElement = function(index){
    this.kernelIdleIcon.waitForEnabled();
    var codeCell = this.runCodeCellByIndex(index);
    return codeCell.$('#svgg');
  }

  this.runCellAndCheckOutputText = function(index, expectedText){
    var resultTest;
    try{
      resultTest = this.runCellToGetOutputTextElement(index).getText();
    }catch(e){
      console.log(expectedText + ' --- ' + e.toString());
      resultTest = this.runCellToGetOutputTextElement(index).getText();
    }
    expect(resultTest).toMatch(expectedText);
  }

  this.runCellToGetOutputTextElement = function(index){
    var codeCell = this.runCodeCellByIndex(index);
    return codeCell.$('div.output_subarea.output_text');
  }

  this.plotLegendContainerIsEnabled = function(dtcontainer){
    var plotLegendContainer = dtcontainer.$('#plotLegendContainer');
    plotLegendContainer.waitForEnabled();
  }

  this.dataTablesIsEnabled = function(dtcontainer){
    var dataTables = dtcontainer.$('.dataTables_scroll');
    dataTables.waitForEnabled();
  }

  this.runCellToGetWidgetElement = function(index){
    this.kernelIdleIcon.waitForEnabled();
    var codeCell = this.runCodeCellByIndex(index);
    return codeCell.$('div.jupyter-widgets');
  }

  this.runCellToGetEasyForm = function(index){
    this.kernelIdleIcon.waitForEnabled();
    var codeCell = this.runCodeCellByIndex(index);
    return codeCell.$('div.beaker-easyform-container');
  }

  this.runCellToGetTableElement = function(index){
    this.kernelIdleIcon.waitForEnabled();
    var codeCell = this.runCodeCellByIndex(index);
    return codeCell.$('div.dataTables_scrollBody');
  }

  this.checkCellOutputText = function(index, expectedText){
    var codeCell = this.getCodeCellByIndex(index);
    codeCell.scroll();
    var resultTest;
    try{
      resultTest = codeCell.$('div.output_subarea.output_text').getText();
    }catch(e){
      console.log(expectedText + ' --- ' + e.toString());
      resultTest = this.runCellToGetOutputTextElement(index).getText();
    }
    expect(resultTest).toMatch(expectedText);
  }

};
module.exports = BeakerXPageObject;