فهرست منبع

fix: failed rest tests

trgiangdo 1 ماه پیش
والد
کامیت
78cf90fdc4
4فایلهای تغییر یافته به همراه16 افزوده شده و 5 حذف شده
  1. 1 1
      tests/rest/conftest.py
  2. 3 0
      tests/rest/test_datanode.py
  3. 8 4
      tests/rest/test_job.py
  4. 4 0
      tests/rest/test_task.py

+ 1 - 1
tests/rest/conftest.py

@@ -292,7 +292,7 @@ def default_cycle():
 def __create_job():
     task_manager = _TaskManager
     task = __default_task()
-    task_manager._repository._save(task)
+    task_manager._create(task)
     submit_id = f"SUBMISSION_{str(uuid.uuid4())}"
     return Job(id=JobId(f"JOB_{uuid.uuid4()}"), task=task, submit_id=submit_id, submit_entity_id=task.id)
 

+ 3 - 0
tests/rest/test_datanode.py

@@ -13,6 +13,8 @@ from unittest import mock
 
 from flask import url_for
 
+from taipy.core.data._data_manager_factory import _DataManagerFactory
+
 
 def test_get_datanode(client, default_datanode):
     # test 404
@@ -94,6 +96,7 @@ def test_read_datanode(client, default_df_datanode):
 
 
 def test_write_datanode(client, default_datanode):
+    _DataManagerFactory._build_manager()._repository._save(default_datanode)
     with mock.patch("taipy.core.data._data_manager._DataManager._get") as config_mock:
         config_mock.return_value = default_datanode
         # Get DataNode

+ 8 - 4
tests/rest/test_job.py

@@ -13,6 +13,9 @@ from unittest import mock
 
 from flask import url_for
 
+from taipy.core._orchestrator._orchestrator_factory import _OrchestratorFactory
+from taipy.core.job._job_manager_factory import _JobManagerFactory
+
 
 def test_get_job(client, default_job):
     # test 404
@@ -34,8 +37,9 @@ def test_delete_job(client):
     rep = client.get(user_url)
     assert rep.status_code == 404
 
-    with mock.patch("taipy.core.job._job_manager._JobManager._delete"), mock.patch(
-        "taipy.core.job._job_manager._JobManager._get"
+    with (
+        mock.patch("taipy.core.job._job_manager._JobManager._delete"),
+        mock.patch("taipy.core.job._job_manager._JobManager._get"),
     ):
         # test get_job
         rep = client.delete(url_for("api.job_by_id", job_id="foo"))
@@ -66,11 +70,11 @@ def test_get_all_jobs(client, create_job_list):
 
 def test_cancel_job(client, default_job):
     # test 404
-    from taipy.core._orchestrator._orchestrator_factory import _OrchestratorFactory
-
     _OrchestratorFactory._build_orchestrator()
     _OrchestratorFactory._build_dispatcher()
 
+    _JobManagerFactory._build_manager()._repository._save(default_job)
+
     user_url = url_for("api.job_cancel", job_id="foo")
     rep = client.post(user_url)
     assert rep.status_code == 404

+ 4 - 0
tests/rest/test_task.py

@@ -13,6 +13,8 @@ from unittest import mock
 
 from flask import url_for
 
+from taipy.core.task._task_manager_factory import _TaskManagerFactory
+
 
 def test_get_task(client, default_task):
     # test 404
@@ -76,6 +78,8 @@ def test_get_all_tasks(client, task_data, default_task_config_list):
 
 
 def test_execute_task(client, default_task):
+    _TaskManagerFactory._build_manager()._create(default_task)
+
     # test 404
     user_url = url_for("api.task_submit", task_id="foo")
     rep = client.post(user_url)