first commit

.venv/Lib/site-packages/notebook/tests/selenium/test_buffering.py

@@ -0,0 +1,50 @@
+"""Tests buffering of execution requests."""
+
+from .utils import wait_for_selector
+
+
+def wait_for_cell_text_output(notebook, index):
+    cell = notebook.cells[index]
+    output = wait_for_selector(cell, ".output_text", single=True)
+    return output.text
+
+
+def wait_for_kernel_ready(notebook):
+    wait_for_selector(notebook.browser, ".kernel_idle_icon")
+
+
+def test_kernels_buffer_without_conn(prefill_notebook):
+    """Test that execution request made while disconnected is buffered."""
+    notebook = prefill_notebook(["print(1 + 2)"])
+
+    wait_for_kernel_ready(notebook)
+    notebook.browser.execute_script("IPython.notebook.kernel.stop_channels();")
+    notebook.execute_cell(0)
+    notebook.browser.execute_script("IPython.notebook.kernel.reconnect();")
+    wait_for_kernel_ready(notebook)
+
+    assert wait_for_cell_text_output(notebook, 0) == "3"
+
+
+def test_buffered_cells_execute_in_order(prefill_notebook):
+    """Test that buffered requests execute in order."""
+    notebook = prefill_notebook(['', 'k=1', 'k+=1', 'k*=3', 'print(k)'])
+
+    # Repeated execution of cell queued up in the kernel executes
+    # each execution request in order.
+    wait_for_kernel_ready(notebook)
+    notebook.browser.execute_script("IPython.notebook.kernel.stop_channels();")
+    # k == 1
+    notebook.execute_cell(1)
+    # k == 2
+    notebook.execute_cell(2)
+    # k == 6
+    notebook.execute_cell(3)
+    # k == 7
+    notebook.execute_cell(2)
+    notebook.execute_cell(4)
+    notebook.browser.execute_script("IPython.notebook.kernel.reconnect();")
+    wait_for_kernel_ready(notebook)
+
+    # Check that current value of k is 7
+    assert wait_for_cell_text_output(notebook, 4) == "7"