|
6 | 6 | from tornado import gen
|
7 | 7 | from tornado.httpclient import HTTPRequest, HTTPResponse, HTTPError
|
8 | 8 | from traitlets.config import Config
|
9 |
| -from .launchnotebook import NotebookTestBase |
10 |
| -from notebook.gateway.managers import GatewayClient |
| 9 | +from .launchserver import ServerTestBase |
| 10 | +from jupyter_server.gateway.managers import GatewayClient |
11 | 11 |
|
12 | 12 | try:
|
13 | 13 | from unittest.mock import patch, Mock
|
@@ -133,10 +133,10 @@ def mock_gateway_request(url, **kwargs):
|
133 | 133 | raise HTTPError(404, message='Kernel does not exist: %s' % requested_kernel_id)
|
134 | 134 |
|
135 | 135 |
|
136 |
| -mocked_gateway = patch('notebook.gateway.managers.gateway_request', mock_gateway_request) |
| 136 | +mocked_gateway = patch('jupyter_server.gateway.managers.gateway_request', mock_gateway_request) |
137 | 137 |
|
138 | 138 |
|
139 |
| -class TestGateway(NotebookTestBase): |
| 139 | +class TestGateway(ServerTestBase): |
140 | 140 |
|
141 | 141 | mock_gateway_url = 'http://mock-gateway-server:8889'
|
142 | 142 | mock_http_user = 'alice'
|
@@ -165,17 +165,17 @@ def get_argv(cls):
|
165 | 165 | return argv
|
166 | 166 |
|
167 | 167 | def test_gateway_options(self):
|
168 |
| - nt.assert_equal(self.notebook.gateway_config.gateway_enabled, True) |
169 |
| - nt.assert_equal(self.notebook.gateway_config.url, TestGateway.mock_gateway_url) |
170 |
| - nt.assert_equal(self.notebook.gateway_config.http_user, TestGateway.mock_http_user) |
171 |
| - nt.assert_equal(self.notebook.gateway_config.connect_timeout, self.notebook.gateway_config.connect_timeout) |
172 |
| - nt.assert_equal(self.notebook.gateway_config.connect_timeout, 44.4) |
| 168 | + nt.assert_equal(self.server.gateway_config.gateway_enabled, True) |
| 169 | + nt.assert_equal(self.server.gateway_config.url, TestGateway.mock_gateway_url) |
| 170 | + nt.assert_equal(self.server.gateway_config.http_user, TestGateway.mock_http_user) |
| 171 | + nt.assert_equal(self.server.gateway_config.connect_timeout, self.server.gateway_config.connect_timeout) |
| 172 | + nt.assert_equal(self.server.gateway_config.connect_timeout, 44.4) |
173 | 173 |
|
174 | 174 | def test_gateway_class_mappings(self):
|
175 | 175 | # Ensure appropriate class mappings are in place.
|
176 |
| - nt.assert_equal(self.notebook.kernel_manager_class.__name__, 'GatewayKernelManager') |
177 |
| - nt.assert_equal(self.notebook.session_manager_class.__name__, 'GatewaySessionManager') |
178 |
| - nt.assert_equal(self.notebook.kernel_spec_manager_class.__name__, 'GatewayKernelSpecManager') |
| 176 | + nt.assert_equal(self.server.kernel_manager_class.__name__, 'GatewayKernelManager') |
| 177 | + nt.assert_equal(self.server.session_manager_class.__name__, 'GatewaySessionManager') |
| 178 | + nt.assert_equal(self.server.kernel_spec_manager_class.__name__, 'GatewayKernelSpecManager') |
179 | 179 |
|
180 | 180 | def test_gateway_get_kernelspecs(self):
|
181 | 181 | # Validate that kernelspecs come from gateway.
|
@@ -249,11 +249,11 @@ def test_gateway_kernel_lifecycle(self):
|
249 | 249 | self.assertFalse(self.is_kernel_running(kernel_id))
|
250 | 250 |
|
251 | 251 | def create_session(self, kernel_name):
|
252 |
| - """Creates a session for a kernel. The session is created against the notebook server |
| 252 | + """Creates a session for a kernel. The session is created against the server |
253 | 253 | which then uses the gateway for kernel management.
|
254 | 254 | """
|
255 | 255 | with mocked_gateway:
|
256 |
| - nb_path = os.path.join(self.notebook_dir, 'testgw.ipynb') |
| 256 | + nb_path = os.path.join(self.root_dir, 'testgw.ipynb') |
257 | 257 | kwargs = dict()
|
258 | 258 | kwargs['json'] = {'path': nb_path, 'type': 'notebook', 'kernel': {'name': kernel_name}}
|
259 | 259 |
|
|
0 commit comments