|
|
"""Tests for the notebook kernel and session manager."""
|
|
|
|
|
|
from subprocess import PIPE
|
|
|
import time
|
|
|
from unittest import TestCase
|
|
|
|
|
|
from IPython.config.loader import Config
|
|
|
from IPython.frontend.html.notebook.kernelmanager import MultiKernelManager
|
|
|
from IPython.zmq.kernelmanager import KernelManager
|
|
|
|
|
|
class TestKernelManager(TestCase):
|
|
|
|
|
|
def _get_tcp_km(self):
|
|
|
return MultiKernelManager()
|
|
|
|
|
|
def _get_ipc_km(self):
|
|
|
c = Config()
|
|
|
c.KernelManager.transport = 'ipc'
|
|
|
c.KernelManager.ip = 'test'
|
|
|
km = MultiKernelManager(config=c)
|
|
|
return km
|
|
|
|
|
|
def _run_lifecycle(self, km):
|
|
|
kid = km.start_kernel(stdout=PIPE, stderr=PIPE)
|
|
|
self.assertTrue(kid in km)
|
|
|
self.assertTrue(kid in km.list_kernel_ids())
|
|
|
self.assertEqual(len(km),1)
|
|
|
km.restart_kernel(kid)
|
|
|
self.assertTrue(kid in km.list_kernel_ids())
|
|
|
# We need a delay here to give the restarting kernel a chance to
|
|
|
# restart. Otherwise, the interrupt will kill it, causing the test
|
|
|
# suite to hang. The reason it *hangs* is that the shutdown
|
|
|
# message for the restart sometimes hasn't been sent to the kernel.
|
|
|
# Because linger is oo on the shell channel, the context can't
|
|
|
# close until the message is sent to the kernel, which is not dead.
|
|
|
time.sleep(1.0)
|
|
|
km.interrupt_kernel(kid)
|
|
|
k = km.get_kernel(kid)
|
|
|
self.assertTrue(isinstance(k, KernelManager))
|
|
|
km.shutdown_kernel(kid)
|
|
|
self.assertTrue(not kid in km)
|
|
|
|
|
|
def _run_cinfo(self, km, transport, ip):
|
|
|
kid = km.start_kernel(stdout=PIPE, stderr=PIPE)
|
|
|
k = km.get_kernel(kid)
|
|
|
cinfo = km.get_connection_info(kid)
|
|
|
self.assertEqual(transport, cinfo['transport'])
|
|
|
self.assertEqual(ip, cinfo['ip'])
|
|
|
self.assertTrue('stdin_port' in cinfo)
|
|
|
self.assertTrue('iopub_port' in cinfo)
|
|
|
self.assertTrue('shell_port' in cinfo)
|
|
|
self.assertTrue('hb_port' in cinfo)
|
|
|
km.shutdown_kernel(kid)
|
|
|
|
|
|
def test_tcp_lifecycle(self):
|
|
|
km = self._get_tcp_km()
|
|
|
self._run_lifecycle(km)
|
|
|
|
|
|
def test_tcp_cinfo(self):
|
|
|
km = self._get_tcp_km()
|
|
|
self._run_cinfo(km, 'tcp', '127.0.0.1')
|
|
|
|
|
|
def test_ipc_lifecycle(self):
|
|
|
km = self._get_ipc_km()
|
|
|
self._run_lifecycle(km)
|
|
|
|
|
|
def test_ipc_cinfo(self):
|
|
|
km = self._get_ipc_km()
|
|
|
self._run_cinfo(km, 'ipc', 'test')
|
|
|
|
|
|
|
