fixup! editoast: add train_schedule object to search endpoint

Author: EthanPERRUZZA

editoast/src/views/search.rs

@@ -217,8 +217,6 @@ use editoast_common::geometry::GeoJsonPoint;
 use editoast_derive::EditoastError;
 use editoast_derive::Search;
 use editoast_derive::SearchConfigStore;
-use editoast_schemas::train_schedule::Comfort;
-use editoast_schemas::train_schedule::Distribution;
 use editoast_schemas::train_schedule::Margins;
 use editoast_schemas::train_schedule::PathItem;
 use editoast_schemas::train_schedule::PowerRestrictionItem;
@@ -667,6 +665,7 @@ pub(super) struct SearchResultItemStudy {
 )]
 #[allow(unused)]
 /// A search result item for a query with `object = "scenario"`
+#[derive(serde::Deserialize)]
 pub(super) struct SearchResultItemScenario {
     #[search(sql = "scenario.id")]
     id: u64,
@@ -701,6 +700,7 @@ pub(super) struct SearchResultItemScenario {
 )]
 #[allow(unused)]
 /// A search result item for a query with `object = "trainschedule"`
+#[derive(serde::Deserialize)] 
 pub(super) struct SearchResultItemTrainSchedule {
     #[search(sql = "train_schedule.id")]
     id: u64,
@@ -721,11 +721,11 @@ pub(super) struct SearchResultItemTrainSchedule {
     #[search(sql = "train_schedule.initial_speed")]
     initial_speed: f64,
     #[search(sql = "train_schedule.comfort")]
-    comfort: Comfort,
+    comfort: i64,
     #[search(sql = "train_schedule.path")]
     path: Vec<PathItem>,
     #[search(sql = "train_schedule.constraint_distribution")]
-    constraint_distribution: Distribution,
+    constraint_distribution: i64,
     #[search(sql = "train_schedule.speed_limit_tag")]
     speed_limit_tag: Option<String>,
     #[search(sql = "train_schedule.power_restrictions")]
@@ -746,3 +746,67 @@ pub(super) struct SearchResultItemTrainSchedule {
     object(name = "trainschedule", config = SearchResultItemTrainSchedule),
 )]
 pub struct SearchConfigFinder;
+
+#[cfg(test)]
+pub mod test {
+
+    use axum::http::StatusCode;
+    use pretty_assertions::assert_eq;
+    use rstest::rstest;
+    use serde_json::json;
+
+    use super::*;
+    use crate::models::fixtures::{create_simple_train_schedule, create_timetable};
+    use crate::views::test_app::TestAppBuilder;
+    use crate::models::train_schedule::TrainSchedule;
+
+    #[rstest]
+    async fn search_trainschedule_post() {
+        let app = TestAppBuilder::default_app();
+        let pool = app.db_pool();
+
+        // Create the timetable in the database
+        let timetable = create_timetable(&mut pool.get_ok()).await;
+        let timetable_id = timetable.id;
+
+        // Load data of the json file
+        #[derive(serde::Deserialize)]
+        struct TrainScheduleInfo {
+            pub train_name: String,
+        }
+        let train_schedule_info: TrainScheduleInfo = serde_json::from_str(include_str!("../tests/train_schedules/simple.json"))
+        .expect("Unable to parse");
+        let train_name = train_schedule_info.train_name;
+
+        // Add a train_schedule in the database
+        let train_schedule: TrainSchedule = create_simple_train_schedule(&mut pool.get_ok(), timetable_id).await;
+        assert_eq!(train_schedule.train_name, train_name);
+
+        // The body 
+        let request = app.post("/search").json(&json!({
+            "object": "trainschedule",
+            "query": [
+                "and",
+                [
+                    "search",
+                    [
+                        "train_name"
+                    ],
+                    train_name
+                ],
+                [
+                    "=",
+                    [
+
+                        "timetable_id"
+                    ],
+                    timetable_id
+                ]
+            ],
+        }));
+
+        let response: Vec<SearchResultItemTrainSchedule> = app.fetch(request).assert_status(StatusCode::OK).json_into();
+
+        assert_eq!(response[0].train_name, train_name);
+    }
+}