elyra-extension/tests/integration/codesnippetfromselectedcells.ts

/*
 * Copyright 2018-2022 Elyra Authors
 *
 * 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.
 */

const cellSelector =
  'div.CodeMirror-lines[role="presentation"] > div[role="presentation"]';

describe('Code snippet from cells tests', () => {
  beforeEach(() => {
    cy.resetJupyterLab();

    // Create new python notebook
    cy.get(
      '.jp-LauncherCard[data-category="Notebook"][title="Python 3 (ipykernel)"]'
    ).click();

    cy.wait(2000);
  });

  it('test empty cell', () => {
    cy.get(cellSelector)
      .first()
      .rightclick();

    cy.wait(2000);

    cy.get(
      'li.lm-Menu-item[data-command="codesnippet:save-as-snippet"]'
    ).should('have.class', 'p-mod-disabled');
  });

  it('test 1 cell', () => {
    // Create new cell
    cy.get(
      '.jp-NotebookPanel-toolbar > div:nth-child(2) > button:nth-child(1)'
    ).click();

    cy.wait(2000);

    populateCells();

    cy.get(cellSelector)
      .first()
      .rightclick({
        force: true
      });

    cy.wait(2000);

    cy.get(
      'li.lm-Menu-item[data-command="codesnippet:save-as-snippet"]'
    ).click();

    cy.wait(2000);

    // Verify snippet editor contents
    cy.get('span[role="presentation"]:visible').should(
      'have.text',
      'print("test")'
    );
  });

  it('test 2 cells', () => {
    // Create new cells
    cy.get(
      '.jp-NotebookPanel-toolbar > div:nth-child(2) > button:nth-child(1)'
    ).click();

    cy.wait(2000);

    populateCells();

    // Select all cells
    cy.get(
      ':nth-child(1) > .jp-Cell-inputWrapper > .jp-InputArea > .jp-InputPrompt'
    )
      .first()
      .click({
        shiftKey: true
      });

    cy.get('div.lm-Widget.p-Widget.jp-InputPrompt.jp-InputArea-prompt:visible')
      .first()
      .rightclick({
        force: true
      });

    cy.wait(2000);

    cy.get(
      'li.lm-Menu-item[data-command="codesnippet:save-as-snippet"]'
    ).click();

    cy.wait(2000);

    // Verify snippet editor contents
    cy.get(
      '.elyra-form-code > .CodeMirror > .CodeMirror-scroll span[role="presentation"]:contains("test")'
    ).should('have.length', 2);
  });
});

// ------------------------------
// ----- Utility Functions
// ------------------------------

// Populate cells
const populateCells = (): void => {
  cy.get('span[role="presentation"]').each(cell => {
    cy.get(cell).type('print("test")');
    cy.dismissAssistant('notebook');
  });
};