Mock cannot be used with await, so provide a fake coroutine

cylc · Oct 29, 2020 · 338b973 · 338b973
1 parent d3648e7
commit 338b973
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/cylc/uiserver/tests/test_workflows_mgr.py b/cylc/uiserver/tests/test_workflows_mgr.py
@@ -444,7 +444,13 @@ async def test_connect(
     mocker.patch('cylc.flow.network.client.SuiteRuntimeClient.get_header')
     mocker.patch('cylc.uiserver.data_store_mgr.DataStoreMgr.'
                  'start_subscription')
-    mocker.patch('cylc.uiserver.data_store_mgr.DataStoreMgr.sync_workflow')
+
+    async def my_sync_workflow(*args, **kwargs):
+        return True
+    mocker.patch(
+        'cylc.uiserver.data_store_mgr.DataStoreMgr.sync_workflow',
+        side_effect=my_sync_workflow
+    )
 
     await uiserver.workflows_mgr.update()