From 371f66553b6b1d81e64f6066b26d2c85b3397392 Mon Sep 17 00:00:00 2001
From: Erwan Bousse <erwan.bousse@ls2n.fr>
Date: Fri, 16 Sep 2022 10:33:30 +0200
Subject: [PATCH] Add missing "addTrain" operation
---
.../trainreservation/TicketReservationSystem.java | 6 ++++++
.../trainreservation/impl/TicketReservationSystemImpl.java | 6 ++++++
2 files changed, 12 insertions(+)
diff --git a/src/main/java/fr/univnantes/trainreservation/TicketReservationSystem.java b/src/main/java/fr/univnantes/trainreservation/TicketReservationSystem.java
index 48ba128..85f785c 100644
--- a/src/main/java/fr/univnantes/trainreservation/TicketReservationSystem.java
+++ b/src/main/java/fr/univnantes/trainreservation/TicketReservationSystem.java
@@ -81,6 +81,12 @@ public interface TicketReservationSystem {
*/
void addCity(City city);
+ /**
+ * Adds a new train in the system.
+ * @param train The train to add.
+ */
+ void addTrain(Train train);
+
/**
* Creates and registers a new trip in the system.
* This new trip must take place *after* the last trip of the chosen train (based on the real arrival time).
diff --git a/src/main/java/fr/univnantes/trainreservation/impl/TicketReservationSystemImpl.java b/src/main/java/fr/univnantes/trainreservation/impl/TicketReservationSystemImpl.java
index e23cbc2..2517d56 100644
--- a/src/main/java/fr/univnantes/trainreservation/impl/TicketReservationSystemImpl.java
+++ b/src/main/java/fr/univnantes/trainreservation/impl/TicketReservationSystemImpl.java
@@ -164,4 +164,10 @@ public class TicketReservationSystemImpl implements TicketReservationSystem {
public List<Trip> findOrderedTripsOfTrain(Train train) {
return trips.stream().sorted((t1, t2) -> t1.findRealArrivalTime().isBefore(t2.findRealDepartureTime()) ? -1 : 1).collect(Collectors.toList());
}
+
+ @Override
+ public void addTrain(Train train) {
+ this.trains.add(train);
+
+ }
}
--
GitLab