"""Test basic cell execution methods, related shortcuts, and error modes"""


from .utils import CELL_OUTPUT_SELECTOR, EDITOR_PAGE


def test_execute_code(notebook_frontend):
    # Execute cell with Javascript API
    notebook_frontend.edit_cell(index=0, content='a=10; print(a)')
    notebook_frontend.evaluate("Jupyter.notebook.get_cell(0).execute();", page=EDITOR_PAGE)
    outputs = notebook_frontend.wait_for_cell_output(0)
    outputs.wait_for('visible')
    assert outputs.get_inner_text().strip() == '10'

    # Execute cell with Shift-Enter
    notebook_frontend.edit_cell(index=0, content='a=11; print(a)')
    notebook_frontend.clear_all_output()
    notebook_frontend.press("Enter", EDITOR_PAGE, ["Shift"])
    outputs = notebook_frontend.wait_for_cell_output(0)
    outputs.wait_for('visible')
    assert outputs.get_inner_text().strip() == '11'
    notebook_frontend.delete_cell(1)  # Shift+Enter adds a cell

    # Execute cell with Ctrl-Enter (or equivalent)
    notebook_frontend.edit_cell(index=0, content='a=12; print(a)')
    notebook_frontend.clear_all_output()
    notebook_frontend.press(
        "Enter",
        EDITOR_PAGE,
        modifiers=[notebook_frontend.get_platform_modifier_key()]
    )
    outputs = notebook_frontend.wait_for_cell_output(0)
    outputs.wait_for('visible')
    assert outputs.get_inner_text().strip() == '12'

    # Execute cell with toolbar button
    notebook_frontend.edit_cell(index=0, content='a=13; print(a)')
    notebook_frontend.clear_all_output()
    notebook_frontend.click_toolbar_execute_btn()
    outputs = notebook_frontend.wait_for_cell_output(0)
    outputs.wait_for('visible')
    assert outputs.get_inner_text().strip() == '13'
    notebook_frontend.delete_cell(1)  # Toolbar execute button adds a cell

    # Set up two cells to test stopping on error
    notebook_frontend.type('a', EDITOR_PAGE)
    notebook_frontend.edit_cell(index=0, content='raise IOError')
    notebook_frontend.edit_cell(index=1, content='a=14; print(a)')

    # Default behaviour: stop on error
    notebook_frontend.clear_all_output()
    notebook_frontend.evaluate("""
        var cell0 = Jupyter.notebook.get_cell(0);
        var cell1 = Jupyter.notebook.get_cell(1);
        cell0.execute();
        cell1.execute();
    """, page=EDITOR_PAGE)
    assert notebook_frontend.get_cell_output(0) is None

    # Execute a cell with stop_on_error=false
    # .......................................
    # Make sure the previous eval call finished by checking for kernel_idle_icon
    notebook_frontend.locate(".kernel_idle_icon", EDITOR_PAGE).wait_for('visible')
    notebook_frontend.clear_all_output()
    # Make sure clear cell output call is finished
    notebook_frontend.wait_for_condition(
        lambda: len(
            [item for item in notebook_frontend.locate_all(CELL_OUTPUT_SELECTOR, EDITOR_PAGE)]
        ) == 0, timeout=120, period=5
    )
    notebook_frontend.evaluate("""
        var cell0 = Jupyter.notebook.get_cell(0);
        var cell1 = Jupyter.notebook.get_cell(1);
        cell0.execute(false);
        cell1.execute();
    """, page=EDITOR_PAGE)
    # Make sure the previous eval call finished by checking for kernel_idle_icon
    notebook_frontend.locate(".kernel_idle_icon", EDITOR_PAGE).wait_for('visible')
    if notebook_frontend.locate(".kernel_busy_icon", EDITOR_PAGE).is_visible():
        # If kernel is busy, wait for it to finish
        notebook_frontend.locate(".kernel_idle_icon", EDITOR_PAGE).wait_for('visible')
    # Wait for cell outputs
    notebook_frontend.wait_for_condition(
        lambda: len(
            [item for item in notebook_frontend.locate_all(CELL_OUTPUT_SELECTOR, EDITOR_PAGE)]
        ) >= 2, timeout=120, period=5
    )
    outputs = notebook_frontend.wait_for_cell_output(1)
    assert outputs is not None
    assert outputs.get_inner_text().strip() == '14'