From 3029bdb3824e2d75b838d21913e9f8dc26ec1941 Mon Sep 17 00:00:00 2001
From: ArnaudOggy <arnaud.ogiela@gmail.com>
Date: Thu, 3 Mar 2022 14:10:08 +0100
Subject: [PATCH] [tech] Fix Equipments sanitizing

---
 Cargo.toml   | 2 +-
 src/model.rs | 6 ++++++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/Cargo.toml b/Cargo.toml
index 0d404d4ad..55a603963 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -1,7 +1,7 @@
 [package]
 authors = ["Kisio Digital <team.coretools@kisio.com>", "Guillaume Pinot <texitoi@texitoi.eu>"]
 name = "transit_model"
-version = "0.47.0"
+version = "0.48.0"
 license = "AGPL-3.0-only"
 description = "Transit data management"
 repository = "https://github.com/CanalTP/transit_model"
diff --git a/src/model.rs b/src/model.rs
index 276c9957e..d95a5e91e 100644
--- a/src/model.rs
+++ b/src/model.rs
@@ -301,6 +301,9 @@ impl Collections {
                 if let Some(level_id) = &sl.level_id {
                     level_id_used.insert(level_id.clone());
                 }
+                if let Some(equipment_id) = &sl.equipment_id {
+                    equipments_used.insert(equipment_id.clone());
+                }
                 comments_used.extend(&mut sl.comment_links.iter().map(|cl| cl.to_string()));
                 true
             })
@@ -407,6 +410,9 @@ impl Collections {
                     if let Some(level_id) = &sa.level_id {
                         level_id_used.insert(level_id.clone());
                     }
+                    if let Some(equipment_id) = &sa.equipment_id {
+                        equipments_used.insert(equipment_id.clone());
+                    }
                     comments_used.extend(&mut sa.comment_links.iter().map(|cl| cl.to_string()));
                     true
                 } else {