add timeouts

Author: Zsailer

jupyter_server/tests/services/kernels/test_api.py

@@ -12,6 +12,9 @@
 from jupyter_server.utils import url_path_join
 
 
+TEST_TIMEOUT = 20
+
+
 @pytest.fixture
 def pending_kernel_is_ready(jp_serverapp):
     async def _(kernel_id):
@@ -63,6 +66,7 @@ async def test_no_kernels(jp_fetch):
     assert kernels == []
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_default_kernels(jp_fetch, jp_base_url, jp_cleanup_subprocesses):
     r = await jp_fetch("api", "kernels", method="POST", allow_nonstandard_methods=True)
     kernel = json.loads(r.body.decode())
@@ -78,6 +82,7 @@ async def test_default_kernels(jp_fetch, jp_base_url, jp_cleanup_subprocesses):
     await jp_cleanup_subprocesses()
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_main_kernel_handler(
     jp_fetch, jp_base_url, jp_cleanup_subprocesses, jp_serverapp, pending_kernel_is_ready
 ):
@@ -146,6 +151,7 @@ async def test_main_kernel_handler(
     await jp_cleanup_subprocesses()
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_kernel_handler(jp_fetch, jp_cleanup_subprocesses, pending_kernel_is_ready):
     # Create a kernel
     r = await jp_fetch(
@@ -193,6 +199,7 @@ async def test_kernel_handler(jp_fetch, jp_cleanup_subprocesses, pending_kernel_
     await jp_cleanup_subprocesses()
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_kernel_handler_startup_error(
     jp_fetch, jp_cleanup_subprocesses, jp_serverapp, jp_kernelspecs
 ):
@@ -204,6 +211,7 @@ async def test_kernel_handler_startup_error(
         await jp_fetch("api", "kernels", method="POST", body=json.dumps({"name": "bad"}))
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_kernel_handler_startup_error_pending(
     jp_fetch, jp_ws_fetch, jp_cleanup_subprocesses, jp_serverapp, jp_kernelspecs
 ):
@@ -219,6 +227,7 @@ async def test_kernel_handler_startup_error_pending(
         await jp_ws_fetch("api", "kernels", kid, "channels")
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_connection(
     jp_fetch, jp_ws_fetch, jp_http_port, jp_auth_header, jp_cleanup_subprocesses
 ):

jupyter_server/tests/services/sessions/test_api.py

@@ -17,6 +17,9 @@
 from jupyter_server.utils import url_path_join
 
 
+TEST_TIMEOUT = 20
+
+
 j = lambda r: json.loads(r.body.decode())
 
 
@@ -197,6 +200,7 @@ def assert_session_equality(actual, expected):
     assert_kernel_equality(actual["kernel"], expected["kernel"])
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_create(session_client, jp_base_url, jp_cleanup_subprocesses, jp_serverapp):
     # Make sure no sessions exist.
     resp = await session_client.list()
@@ -241,6 +245,7 @@ async def test_create(session_client, jp_base_url, jp_cleanup_subprocesses, jp_s
     await jp_cleanup_subprocesses()
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_create_bad(
     session_client, jp_base_url, jp_cleanup_subprocesses, jp_serverapp, jp_kernelspecs
 ):
@@ -261,6 +266,7 @@ async def test_create_bad(
     await jp_cleanup_subprocesses()
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_create_bad_pending(
     session_client, jp_base_url, jp_ws_fetch, jp_cleanup_subprocesses, jp_serverapp, jp_kernelspecs
 ):
@@ -293,6 +299,7 @@ async def test_create_bad_pending(
     await jp_cleanup_subprocesses()
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_create_file_session(
     session_client, jp_cleanup_subprocesses, jp_serverapp, session_is_ready
 ):
@@ -306,6 +313,7 @@ async def test_create_file_session(
     await jp_cleanup_subprocesses()
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_create_console_session(
     session_client, jp_cleanup_subprocesses, jp_serverapp, session_is_ready
 ):
@@ -320,6 +328,7 @@ async def test_create_console_session(
     await jp_cleanup_subprocesses()
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_create_deprecated(session_client, jp_cleanup_subprocesses, jp_serverapp):
     resp = await session_client.create_deprecated("foo/nb1.ipynb")
     assert resp.code == 201
@@ -332,6 +341,7 @@ async def test_create_deprecated(session_client, jp_cleanup_subprocesses, jp_ser
     await jp_cleanup_subprocesses()
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_create_with_kernel_id(
     session_client, jp_fetch, jp_base_url, jp_cleanup_subprocesses, jp_serverapp
 ):
@@ -363,6 +373,7 @@ async def test_create_with_kernel_id(
     await jp_cleanup_subprocesses()
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_create_with_bad_kernel_id(
     session_client, jp_cleanup_subprocesses, jp_serverapp, session_is_ready
 ):
@@ -378,6 +389,7 @@ async def test_create_with_bad_kernel_id(
     await jp_cleanup_subprocesses()
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_delete(session_client, jp_cleanup_subprocesses, jp_serverapp, session_is_ready):
     resp = await session_client.create("foo/nb1.ipynb")
 
@@ -399,6 +411,7 @@ async def test_delete(session_client, jp_cleanup_subprocesses, jp_serverapp, ses
     await jp_cleanup_subprocesses()
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_modify_path(session_client, jp_cleanup_subprocesses, jp_serverapp, session_is_ready):
     resp = await session_client.create("foo/nb1.ipynb")
     newsession = j(resp)
@@ -413,6 +426,7 @@ async def test_modify_path(session_client, jp_cleanup_subprocesses, jp_serverapp
     await jp_cleanup_subprocesses()
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_modify_path_deprecated(
     session_client, jp_cleanup_subprocesses, jp_serverapp, session_is_ready
 ):
@@ -429,6 +443,7 @@ async def test_modify_path_deprecated(
     await jp_cleanup_subprocesses()
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_modify_type(session_client, jp_cleanup_subprocesses, jp_serverapp, session_is_ready):
     resp = await session_client.create("foo/nb1.ipynb")
     newsession = j(resp)
@@ -443,6 +458,7 @@ async def test_modify_type(session_client, jp_cleanup_subprocesses, jp_serverapp
     await jp_cleanup_subprocesses()
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_modify_kernel_name(
     session_client, jp_fetch, jp_cleanup_subprocesses, jp_serverapp, session_is_ready
 ):
@@ -470,6 +486,7 @@ async def test_modify_kernel_name(
     await jp_cleanup_subprocesses()
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_modify_kernel_id(
     session_client, jp_fetch, jp_cleanup_subprocesses, jp_serverapp, session_is_ready
 ):
@@ -504,6 +521,7 @@ async def test_modify_kernel_id(
     await jp_cleanup_subprocesses()
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
 async def test_restart_kernel(
     session_client, jp_base_url, jp_fetch, jp_ws_fetch, jp_cleanup_subprocesses, session_is_ready
 ):