pytest.raises

dask · Oct 5, 2022 · 74f71c5 · 74f71c5
1 parent 549c902
commit 74f71c5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/distributed/tests/test_worker.py b/distributed/tests/test_worker.py
@@ -1698,7 +1698,8 @@ async def test_pip_install_fails(c, s, a, b):
         with mock.patch(
             "distributed.diagnostics.plugin.subprocess.Popen", return_value=mocked
         ) as Popen:
-            await c.register_worker_plugin(PipInstall(packages=["not-a-package"]))
+            with pytest.raises(RuntimeError):
+                await c.register_worker_plugin(PipInstall(packages=["not-a-package"]))
 
             assert Popen.call_count == 1
             logs = logger.getvalue()