mirror of
https://github.com/jupyter/notebook.git
synced 2024-12-27 04:20:22 +08:00
41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
"""Base class for notebook tests."""
|
|
|
|
import sys
|
|
import time
|
|
from subprocess import Popen, PIPE
|
|
from unittest import TestCase
|
|
|
|
from IPython.utils.tempdir import TemporaryDirectory
|
|
|
|
|
|
class NotebookTestBase(TestCase):
|
|
"""A base class for tests that need a running notebook.
|
|
|
|
This creates an empty profile in a temp ipython_dir
|
|
and then starts the notebook server with a separate temp notebook_dir.
|
|
"""
|
|
|
|
port = 12342
|
|
|
|
def setUp(self):
|
|
self.ipython_dir = TemporaryDirectory()
|
|
self.notebook_dir = TemporaryDirectory()
|
|
notebook_args = [
|
|
sys.executable, '-c',
|
|
'from IPython.html.notebookapp import launch_new_instance; launch_new_instance()',
|
|
'--port=%d' % self.port,
|
|
'--no-browser',
|
|
'--ipython-dir=%s' % self.ipython_dir.name,
|
|
'--notebook-dir=%s' % self.notebook_dir.name
|
|
]
|
|
self.notebook = Popen(notebook_args, stdout=PIPE, stderr=PIPE)
|
|
time.sleep(3.0)
|
|
|
|
def tearDown(self):
|
|
self.notebook.terminate()
|
|
self.ipython_dir.cleanup()
|
|
self.notebook_dir.cleanup()
|
|
|
|
def base_url(self):
|
|
return 'http://localhost:%i/' % self.port
|