fix(api): fix failed tests (#1646)

AntaresSimulatorTeam · Jul 28, 2023 · 7dc032b · 7dc032b
1 parent f2afc37
commit 7dc032b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/tests/storage/integration/test_exporter.py b/tests/storage/integration/test_exporter.py
@@ -121,6 +121,7 @@ def test_export_flat(
     export_study_flat(
         path_studies / "STA-mini",
         export_path / "STA-mini-export",
+        Mock(),
         outputs,
         output_list,
     )

diff --git a/tests/variantstudy/model/test_variant_model.py b/tests/variantstudy/model/test_variant_model.py
@@ -15,6 +15,7 @@
     StudyAdditionalData,
 )
 from antarest.study.storage.variantstudy.command_factory import CommandFactory
+from antarest.study.storage.variantstudy.model.dbmodel import VariantStudy
 from antarest.study.storage.variantstudy.model.model import (
     CommandDTO,
     GenerationResultInfoDTO,
@@ -39,7 +40,12 @@
 )
 
 
-def test_commands_service(tmp_path: Path, command_factory: CommandFactory):
+@patch(
+    "antarest.study.storage.variantstudy.variant_study_service.export_study_flat"
+)
+def test_commands_service(
+    mock_export, tmp_path: Path, command_factory: CommandFactory
+):
     engine = create_engine(
         "sqlite:///:memory:",
         echo=False,
@@ -52,6 +58,7 @@ def test_commands_service(tmp_path: Path, command_factory: CommandFactory):
         custom_engine=engine,
         session_args={"autocommit": False, "autoflush": False},
     )
+
     repository = VariantStudyRepository(LocalCache())
     service = VariantStudyService(
         raw_study_service=Mock(),
@@ -214,6 +221,7 @@ def test_smart_generation(
     def export_flat(
         path_study: Path,
         dest: Path,
+        study_factory: VariantStudy,
         outputs: bool = True,
         denormalize: bool = True,
     ) -> None: