from selenium.webdriver.common.keys import Keys
from .utils import shift, cmdtrl


def test_execute_code(notebook):
    browser = notebook.browser
    
    def clear_outputs():
        return notebook.browser.execute_script(
            "Jupyter.notebook.clear_all_output();")

    # Execute cell with Javascript API
    notebook.edit_cell(index=0, content='a=10; print(a)')
    browser.execute_script("Jupyter.notebook.get_cell(0).execute();")
    outputs = notebook.wait_for_cell_output(0)
    assert outputs[0].text == '10'

    # Execute cell with Shift-Enter
    notebook.edit_cell(index=0, content='a=11; print(a)')
    clear_outputs()
    shift(notebook.browser, Keys.ENTER)
    outputs = notebook.wait_for_cell_output(0)
    assert outputs[0].text == '11'
    notebook.delete_cell(index=1)

    # Execute cell with Ctrl-Enter
    notebook.edit_cell(index=0, content='a=12; print(a)')
    clear_outputs()
    cmdtrl(notebook.browser, Keys.ENTER)
    outputs = notebook.wait_for_cell_output(0)
    assert outputs[0].text == '12'

    # Execute cell with toolbar button
    notebook.edit_cell(index=0, content='a=13; print(a)')
    clear_outputs()
    notebook.browser.find_element_by_css_selector(
        "button[data-jupyter-action='jupyter-notebook:run-cell-and-select-next']").click()
    outputs = notebook.wait_for_cell_output(0)
    assert outputs[0].text == '13'

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

    # Default behaviour: stop on error
    clear_outputs()
    browser.execute_script("""
        var cell0 = Jupyter.notebook.get_cell(0);
        var cell1 = Jupyter.notebook.get_cell(1);
        cell0.execute();
        cell1.execute();
    """)
    outputs = notebook.wait_for_cell_output(0)
    assert notebook.get_cell_output(1) == []

    # Execute a cell with stop_on_error=false
    clear_outputs()
    browser.execute_script("""
            var cell0 = Jupyter.notebook.get_cell(0);
            var cell1 = Jupyter.notebook.get_cell(1);
            cell0.execute(false);
            cell1.execute();
        """)
    outputs = notebook.wait_for_cell_output(1)
    assert outputs[0].text == '14'