IPython/html/tests/launchnotebook.py

"""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 = 1234

    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()
        time.sleep(3.0)

    def base_url(self):
        return 'http://localhost:%i/' % self.port