From 515400cb709d7e98af02a8ef9c53706db35eb91b Mon Sep 17 00:00:00 2001
From: Thomas Kluyver <thomas@kluyver.me.uk>
Date: Tue, 5 Apr 2022 17:16:42 +0100
Subject: [PATCH] Make monkeypatching of function temporary
---
tests/test_webservice.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/tests/test_webservice.py b/tests/test_webservice.py
index 4c8caa7b1..65ac0546c 100644
--- a/tests/test_webservice.py
+++ b/tests/test_webservice.py
@@ -144,13 +144,15 @@ async def test_wait_on_transfer_exceptions(
('sim', ['DARK', '1', '2', '3', '4'], 1, "success: simulated"),
],
)
-async def test_run_action(mode, cmd, retcode, expected):
+async def test_run_action(mode, cmd, retcode, expected, monkeypatch):
job_db = mock.Mock()
async def mock_run_proc_async(*args):
return retcode, b'Submitted job: 42'
- webservice.webservice.run_proc_async = mock_run_proc_async
+ monkeypatch.setattr(
+ webservice.webservice, 'run_proc_async', mock_run_proc_async
+ )
ret = await run_action(job_db, cmd, mode, 1, 1, 1)
assert ret.lower().startswith(expected)
--
GitLab