Fix test for check_disk_quota

Signed-off-by: Fabrice Normandin <[email protected]>
mila-iqia · Apr 19, 2024 · f50da46 · f50da46
1 parent a2efc28
commit f50da46
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/integration/test_code.py b/tests/integration/test_code.py
@@ -35,7 +35,8 @@
     reason="Graham doesn't use a lustre filesystem for $HOME.",
     strict=True,
 )
-def test_check_disk_quota(
+@pytest.mark.asyncio
+async def test_check_disk_quota(
     login_node_v2: RemoteV2,
     capsys: pytest.LogCaptureFixture,
     caplog: pytest.LogCaptureFixture,
@@ -44,7 +45,7 @@ def test_check_disk_quota(
         pytest.skip(reason="Test doesn't work on localhost.")
 
     with caplog.at_level(logging.DEBUG):
-        check_disk_quota(remote=login_node_v2)
+        await check_disk_quota(remote=login_node_v2)
     # TODO: Maybe figure out a way to actually test this, (apart from just running it
     # and expecting no errors).