tests: fuzzer: fix train schedule request format

Author: eckter

tests/fuzzer/fuzzer.py

@@ -148,7 +148,7 @@ def test_new_train(
             schedule_payload=schedule_payload,
         )
 
-    schedule_id = r.json()["ids"][0]
+    schedule_id = r.json()[0]
     r = get_with_timeout(f"{editoast_url}train_schedule/{schedule_id}/result/")
     if r.status_code // 100 != 2:
         make_error(
@@ -551,7 +551,7 @@ def make_payload_schedule(scenario: Scenario, path: int, rolling_stock: int, pat
                 "departure_time": random.randint(0, 3600 * 24),
                 "allowances": make_random_allowances(path_length),
                 "initial_speed": 0,
-                "rolling_stock": rolling_stock,
+                "rolling_stock_id": rolling_stock,
                 "speed_limit_category": "foo",
             }
         ],

tests/tests/test_regressions.py

@@ -35,7 +35,7 @@ def _schedule_with_payload(editoast_url, payload, accept_400):
         if r.status_code // 100 == 4 and accept_400:
             return None
         raise RuntimeError(f"Schedule error {r.status_code}: {r.content}")
-    return r.json()["ids"][0]
+    return r.json()[0]
 
 
 def _stdcm_with_payload(editoast_url: str, payload):
@@ -57,7 +57,7 @@ def _reproduce_test(path_to_json: Path, scenario: Scenario, rolling_stock_id: in
     stop_after_pathfinding = fuzzer_output["error_type"] == "PATHFINDING"
     stop_after_schedule = fuzzer_output["error_type"] == "SCHEDULE"
 
-    assert "small_infra" == fuzzer_output["infra_name"]
+    assert fuzzer_output["infra_name"] in ["small_infra", "Small Infra"]
     timetable = scenario.timetable
     path_id = _pathfinding_with_payload(
         EDITOAST_URL, fuzzer_output["path_payload"], scenario.infra, stop_after_pathfinding