Added new arguments added in #5103

golemfactory · Mar 2, 2020 · 8d13135 · 8d13135
1 parent 6f4df24
commit 8d13135
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/scripts/task_api_tests/basic_integration.py b/scripts/task_api_tests/basic_integration.py
@@ -38,9 +38,12 @@ async def test_task(
 ) -> None:
 
     env_prerequisites = app_definition.requestor_prereq
-    app_manager = appmanager.AppManager()
-    app_manager.register_app(app_definition)
-    app_manager.set_enabled(app_definition.id, True)
+    app_manager = appmanager.AppManager(work_dir / 'apps', False)
+    app_id = app_definition.id
+    if not app_manager.registered(app_id):
+        app_manager.register_app(app_definition)
+    if not app_manager.enabled(app_id):
+        app_manager.set_enabled(app_id, True)
 
     runtime_logs_dir = work_dir / 'runtime_logs'
     runtime_logs_dir.mkdir()
@@ -73,7 +76,7 @@ async def test_task(
     output_dir = work_dir / 'output'
     output_dir.mkdir()
     golem_params = requestedtaskmanager.CreateTaskParams(
-        app_id=app_definition.id,
+        app_id=app_id,
         name='testtask',
         task_timeout=TASK_TIMEOUT,
         subtask_timeout=SUBTASK_TIMEOUT,