test a notebook, basically

Author: bollwyvl

atest/acceptance/10_lab/10_notebook.robot

@@ -0,0 +1,18 @@
+*** Settings ***
+Suite Setup       Wait for New Jupyter Server to be Ready
+Test Teardown     Close All Browsers
+Library           JupyterLibrary
+Library           Process
+
+*** Test Cases ***
+IPython Notebook
+    Open JupyterLab
+    Launch a new JupyterLab Document
+    Add and Run Cell    print("hello world")
+    Wait Until Kernel Is Idle
+    Capture Page Screenshot    ipython.png
+
+*** Keywords ***
+Clean Up Everything
+    Close All Browsers
+    Terminate All Processes

src/JupyterLibrary/resources/jupyterlab/Launcher.robot

@@ -3,10 +3,10 @@ Documentation     Interact with the JupyterLab Launcher
 Resource   JupyterLibrary/resources/jupyterlab/Selectors.robot
 
 *** Keywords ***
-Launch a new Document
-    [Arguments]    ${kernel}=Python3    ${category}=Notebook
+Launch a new JupyterLab Document
+    [Arguments]    ${kernel}=Python 3    ${category}=Notebook
     [Documentation]    Use the JupyterLab launcher to launch Notebook or Console
-    Click Element    ${JLAB XP CARD}[@title='${kernel}'][@data-category='${category}']
+    Click Element    xpath:${JLAB XP CARD}[@title='${kernel}'][@data-category='${category}']
     Wait Until Page Does Not Contain Element    css:${JLAB CSS SPINNER}
     Wait Until Page Contains Element    css:${JLAB CSS CELL}
     Sleep    0.1s

src/JupyterLibrary/resources/jupyterlab/Notebook.robot

@@ -0,0 +1,17 @@
+*** Settings ***
+Resource   JupyterLibrary/resources/jupyterlab/Selectors.robot
+
+*** Keywords ***
+Add and Run Cell
+    [Arguments]    ${code}
+    [Documentation]    Add a code cell to the currently active notebook and run it
+    Click Element    css:${JLAB CSS NB TOOLBAR} ${JLAB CSS ICON ADD}
+    Sleep    0.1s
+    Click Element    css:${JLAB CSS CELL}
+    Execute JavaScript    document.querySelector("${JLAB CSS CELL}").CodeMirror.setValue(`${code}`)
+    Click Element    css:${JLAB CSS ICON RUN}
+
+Wait Until Kernel Is Idle
+    [Documentation]    Wait for the kernel to be busy, and then stop being busy
+    Wait Until Page Does Not Contain Element    ${JLAB CSS BUSY KERNEL}
+    Wait Until Page Does Not Contain    ${JLAB TEXT BUSY PROMPT}

src/JupyterLibrary/resources/jupyterlab/Selectors.robot

@@ -9,9 +9,17 @@ ${JLAB CSS CMD ITEM}    .p-CommandPalette-item
 ${JLAB CSS ACCEPT}  .jp-mod-accept
 ${JLAB CSS ACTIVE SIDEBAR}  .jp-SideBar .p-TabBar-tab.p-mod-current
 ${JLAB CSS SIDEBAR TAB}  .jp-SideBar .p-TabBar-tab
+${JLAB CSS BUSY KERNEL}  .jp-Toolbar-kernelStatus.jp-FilledCircleIcon
+
+${JLAB CSS NB TOOLBAR}  .jp-NotebookPanel-toolbar
+
+${JLAB CSS ICON RUN}  .jp-RunIcon
+${JLAB CSS ICON ADD}  .jp-AddIcon
 
 ${JLAB XP TOP}            //div[@id='jp-top-panel']
 ${JLAB XP MENU LABEL}       //div[@class='p-MenuBar-itemLabel']
 ${JLAB XP MENU ITEM LABEL}  //div[@class='p-Menu-itemLabel']
 ${JLAB XP CARD}           //div[@class='jp-LauncherCard']
 ${JLAB XP DOCK}           //div[@id='jp-main-dock-panel']
+
+${JLAB TEXT BUSY PROMPT}  In [*]: