tests: adapt fuzzer for timetable v2

Author: eckter

tests/conftest.py

@@ -10,7 +10,7 @@
 from tests.scenario import Scenario
 from tests.services import EDITOAST_URL
 from tests.test_e2e import FAST_ROLLING_STOCK_JSON_PATH, TestRollingStock
-from tests.utils.timetable import create_scenario
+from tests.utils.timetable import create_scenario, create_scenario_v2
 
 
 def _load_generated_infra(name: str) -> int:
@@ -80,6 +80,12 @@ def small_scenario(small_infra: Infra, foo_project_id: int, foo_study_id: int) -
     yield Scenario(foo_project_id, foo_study_id, scenario_id, small_infra.id, timetable_id)
 
 
+@pytest.fixture
+def small_scenario_v2(small_infra: Infra, foo_project_id: int, foo_study_id: int) -> Iterator[Scenario]:
+    scenario_id, timetable_id = create_scenario_v2(EDITOAST_URL, small_infra.id, foo_project_id, foo_study_id)
+    yield Scenario(foo_project_id, foo_study_id, scenario_id, small_infra.id, timetable_id)
+
+
 def get_rolling_stock(editoast_url: str, rolling_stock_name: str) -> int:
     """
     Returns the ID corresponding to the rolling stock name, if available.

tests/fuzzer/fuzzer.py

@@ -83,6 +83,7 @@ def make_error(
     error = "" if response.content is None else response.content.decode("utf-8")
     raise FailedTest(
         {
+            "timetable_version": 1,
             "error_type": error_type.value,
             "code": response.status_code,
             "error": error,
@@ -282,7 +283,7 @@ def run(
                 print(e)
                 log_folder.mkdir(exist_ok=True)
                 with open(str(log_folder / f"{i}.json"), "w") as f:
-                    print(json.dumps(e.args[0], indent=4), file=f)
+                    print(json.dumps(e.args[0], indent=4, default=lambda o: "<not serializable>"), file=f)
 
         # Let's reset the scenario (empty timetable) so we can keep a
         # manageable/reproducible state.