Fix tests

Author: krassowski

tests/test_curve.py

@@ -180,7 +180,7 @@ def _make_app(tmp_path, *, enable_curve=False, **kwargs):
     if enable_curve:
         # Populate the Curve keys into the connection file
         km = KernelManager(connection_file=connection_file_path)
-        km.transport_encryption = True
+        km.transport_encryption = "enabled"
         km.pre_start_kernel()
 
     app = IPKernelApp(connection_file=connection_file_path, **kwargs)

tests/test_kernelapp.py

@@ -6,6 +6,7 @@
 
 import pytest
 from jupyter_client import KernelManager
+from jupyter_client.kernelspec import KernelSpecManager
 from jupyter_core.paths import secure_write
 from traitlets.config.loader import Config
 
@@ -133,19 +134,38 @@ def test_trio_loop():
 
 
 def test_init_sockets_curve_enabled_logs_debug(tmp_path):
+    kernel_name = "curve-test"
+    kernels_dir = tmp_path / "kernels"
+    kernel_dir = kernels_dir / kernel_name
+    kernel_dir.mkdir(parents=True)
+    with (kernel_dir / "kernel.json").open("w") as f:
+        json.dump(
+            {
+                "argv": ["python", "-m", "ipykernel_launcher", "-f", "{connection_file}"],
+                "display_name": "curve-test",
+                "language": "python",
+                "metadata": {"supported_encryption": "curve"},
+            },
+            f,
+        )
+
     connection_file = str(tmp_path / "kernel.json")
-    km = KernelManager(connection_file=connection_file)
-    km.transport_encryption = True
+    km = KernelManager(
+        connection_file=connection_file,
+        kernel_name=kernel_name,
+        kernel_spec_manager=KernelSpecManager(kernel_dirs=[str(kernels_dir)]),
+    )
+    km.transport_encryption = "enabled"
     km.pre_start_kernel()
 
     app = IPKernelApp(connection_file=connection_file)
     super(IPKernelApp, app).initialize(argv=[""])
     app.init_connection_file()
-    with patch.object(app.log, "debug") as mock_debug:
+    with patch.object(app.log, "info") as mock_info:
         app.init_sockets()
     app.cleanup_connection_file()
     app.close()
-    messages = [str(call) for call in mock_debug.call_args_list]
+    messages = [str(call) for call in mock_info.call_args_list]
     assert any("Detected CurveZMQ secret key; using transport encryption" in m for m in messages), (
         "Expected a debug log mentioning CurveZMQ when keys are provided"
     )