You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
81 lines
3.1 KiB
81 lines
3.1 KiB
from notebook.tests.selenium.utils import wait_for_selector, Notebook
|
|
from selenium.webdriver.common.keys import Keys
|
|
from selenium.webdriver.support.ui import WebDriverWait
|
|
|
|
|
|
promise_js = """
|
|
var done = arguments[arguments.length - 1];
|
|
(%s).then(
|
|
data => { done(["success", data]); },
|
|
error => { done(["error", error]); }
|
|
);
|
|
"""
|
|
|
|
def execute_promise(js, browser):
|
|
state, data = browser.execute_async_script(promise_js % js)
|
|
if state == 'success':
|
|
return data
|
|
raise Exception(data)
|
|
|
|
def wait_for_rename(browser, nbname, timeout=10):
|
|
wait = WebDriverWait(browser, timeout)
|
|
def notebook_renamed(browser):
|
|
elem = browser.find_element_by_id('notebook_name')
|
|
current_name = browser.execute_script('return arguments[0].innerText', elem)
|
|
return current_name == nbname
|
|
return wait.until(notebook_renamed)
|
|
|
|
def save_as(nb):
|
|
JS = 'Jupyter.notebook.save_notebook_as()'
|
|
return nb.browser.execute_script(JS)
|
|
|
|
def get_notebook_name(nb):
|
|
JS = 'return Jupyter.notebook.notebook_name'
|
|
return nb.browser.execute_script(JS)
|
|
|
|
def refresh_notebook(nb):
|
|
nb.browser.refresh()
|
|
nb.__init__(nb.browser)
|
|
|
|
def test_save_readonly_notebook_as(notebook):
|
|
# Make notebook read-only
|
|
notebook.edit_cell(index=0, content='import os\nimport stat\nos.chmod("'
|
|
+ notebook.browser.current_url.split('?')[0].split('/')[-1] + '", stat.S_IREAD)\nprint(0)')
|
|
notebook.browser.execute_script("Jupyter.notebook.get_cell(0).execute();")
|
|
notebook.wait_for_cell_output(0)
|
|
refresh_notebook(notebook)
|
|
# Test that the notebook is read-only
|
|
assert notebook.browser.execute_script('return Jupyter.notebook.writable') == False
|
|
|
|
# Add some content
|
|
test_content_0 = "print('a simple')\nprint('test script')"
|
|
notebook.edit_cell(index=0, content=test_content_0)
|
|
|
|
# Wait for Save As modal, save
|
|
save_as(notebook)
|
|
wait_for_selector(notebook.browser, '.save-message')
|
|
inp = notebook.browser.find_element_by_xpath('//input[@data-testid="save-as"]')
|
|
inp.send_keys('writable_notebook.ipynb')
|
|
inp.send_keys(Keys.RETURN)
|
|
wait_for_rename(notebook.browser, "writable_notebook")
|
|
# Test that the name changed
|
|
assert get_notebook_name(notebook) == "writable_notebook.ipynb"
|
|
# Test that address bar was updated (TODO: get the base url)
|
|
assert "writable_notebook.ipynb" in notebook.browser.current_url
|
|
# Test that it is no longer read-only
|
|
assert notebook.browser.execute_script('return Jupyter.notebook.writable') == True
|
|
|
|
# Add some more content
|
|
test_content_1 = "print('a second simple')\nprint('script to test save feature')"
|
|
notebook.add_and_execute_cell(content=test_content_1)
|
|
# and save the notebook
|
|
execute_promise("Jupyter.notebook.save_notebook()", notebook.browser)
|
|
|
|
# Test that it still contains the content
|
|
assert notebook.get_cell_contents(index=0) == test_content_0
|
|
assert notebook.get_cell_contents(index=1) == test_content_1
|
|
# even after a refresh
|
|
refresh_notebook(notebook)
|
|
assert notebook.get_cell_contents(index=0) == test_content_0
|
|
assert notebook.get_cell_contents(index=1) == test_content_1
|