fix agent tests

blacklanternsecurity · Sep 25, 2023 · 20edb21 · 20edb21
1 parent 1d92059
commit 20edb21
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/bbot/test/test_step_1/test_cli.py b/bbot/test/test_step_1/test_cli.py
@@ -53,7 +53,10 @@ async def test_cli(monkeypatch, bbot_config):
     task = asyncio.create_task(cli._main())
     await asyncio.sleep(2)
     task.cancel()
-    await task
+    try:
+        await task
+    except asyncio.CancelledError:
+        pass
 
     # no args
     monkeypatch.setattr("sys.argv", ["bbot"])