Show More
@@ -31,37 +31,30 class TestKernelManager(TestCase): | |||||
31 | km.shutdown_kernel(kid) |
|
31 | km.shutdown_kernel(kid) | |
32 | self.assertTrue(not kid in km) |
|
32 | self.assertTrue(not kid in km) | |
33 |
|
33 | |||
34 | def test_km_tcp(self): |
|
34 | def _run_cinfo(self, km, transport, ip): | |
35 | km = self._get_tcp_km() |
|
|||
36 | self._run_lifecycle(km) |
|
|||
37 |
|
||||
38 | def test_km_ipc(self): |
|
|||
39 | km = self._get_ipc_km() |
|
|||
40 | self._run_lifecycle(km) |
|
|||
41 |
|
||||
42 | def test_tcp_cinfo(self): |
|
|||
43 | km = self._get_tcp_km() |
|
|||
44 | kid = km.start_kernel() |
|
35 | kid = km.start_kernel() | |
45 | k = km.get_kernel(kid) |
|
36 | k = km.get_kernel(kid) | |
46 | cinfo = km.get_connection_info(kid) |
|
37 | cinfo = km.get_connection_info(kid) | |
47 |
self.assertEqual( |
|
38 | self.assertEqual(transport, cinfo['transport']) | |
48 |
self.assertEqual( |
|
39 | self.assertEqual(ip, cinfo['ip']) | |
49 | self.assertTrue('stdin_port' in cinfo) |
|
40 | self.assertTrue('stdin_port' in cinfo) | |
50 | self.assertTrue('iopub_port' in cinfo) |
|
41 | self.assertTrue('iopub_port' in cinfo) | |
51 | self.assertTrue('shell_port' in cinfo) |
|
42 | self.assertTrue('shell_port' in cinfo) | |
52 | self.assertTrue('hb_port' in cinfo) |
|
43 | self.assertTrue('hb_port' in cinfo) | |
53 | km.shutdown_kernel(kid) |
|
44 | km.shutdown_kernel(kid) | |
54 |
|
45 | |||
|
46 | def test_km_tcp(self): | |||
|
47 | km = self._get_tcp_km() | |||
|
48 | self._run_lifecycle(km) | |||
|
49 | ||||
|
50 | def test_tcp_cinfo(self): | |||
|
51 | km = self._get_tcp_km() | |||
|
52 | self._run_cinfo(km, 'tcp', '127.0.0.1') | |||
|
53 | ||||
|
54 | def test_km_ipc(self): | |||
|
55 | km = self._get_ipc_km() | |||
|
56 | self._run_lifecycle(km) | |||
|
57 | ||||
55 | def test_ipc_cinfo(self): |
|
58 | def test_ipc_cinfo(self): | |
56 | km = self._get_ipc_km() |
|
59 | km = self._get_ipc_km() | |
57 | kid = km.start_kernel() |
|
60 | self._run_cinfo(km, 'ipc', 'test') | |
58 | k = km.get_kernel(kid) |
|
|||
59 | cinfo = km.get_connection_info(kid) |
|
|||
60 | self.assertEqual('ipc', cinfo['transport']) |
|
|||
61 | self.assertEqual('test', cinfo['ip']) |
|
|||
62 | self.assertTrue('stdin_port' in cinfo) |
|
|||
63 | self.assertTrue('iopub_port' in cinfo) |
|
|||
64 | self.assertTrue('shell_port' in cinfo) |
|
|||
65 | self.assertTrue('hb_port' in cinfo) |
|
|||
66 | km.shutdown_kernel(kid) |
|
|||
67 |
|
@@ -201,6 +201,7 class ShellChannel(ZMQSocketChannel): | |||||
201 | """The thread's main activity. Call start() instead.""" |
|
201 | """The thread's main activity. Call start() instead.""" | |
202 | self.socket = self.context.socket(zmq.DEALER) |
|
202 | self.socket = self.context.socket(zmq.DEALER) | |
203 | self.socket.setsockopt(zmq.IDENTITY, self.session.bsession) |
|
203 | self.socket.setsockopt(zmq.IDENTITY, self.session.bsession) | |
|
204 | self.socket.linger = 0 | |||
204 | self.socket.connect(self.address) |
|
205 | self.socket.connect(self.address) | |
205 | self.stream = zmqstream.ZMQStream(self.socket, self.ioloop) |
|
206 | self.stream = zmqstream.ZMQStream(self.socket, self.ioloop) | |
206 | self.stream.on_recv(self._handle_recv) |
|
207 | self.stream.on_recv(self._handle_recv) |
General Comments 0
You need to be logged in to leave comments.
Login now