mark test_async_main_runs_fully as asyncio

escapewindow · Jun 1, 2018 · 31d37ab · 31d37ab
1 parent 7f3c774
commit 31d37ab
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scriptworker/test/test_client.py b/scriptworker/test/test_client.py
@@ -207,8 +207,9 @@ def test_bad_artifact_url(valid_artifact_rules, valid_artifact_task_ids, url):
         client.validate_artifact_url(valid_artifact_rules, valid_artifact_task_ids, url)
 
 
+@pytest.mark.asyncio
 @pytest.mark.parametrize('should_validate_task', (True, False))
-def test_sync_main_runs_fully(config, event_loop, should_validate_task):
+async def test_sync_main_runs_fully(config, event_loop, should_validate_task):
     copyfile(BASIC_TASK, os.path.join(config['work_dir'], 'task.json'))
     async_main_calls = []
     run_until_complete_calls = []