Add module tests for SIRI extra journeys

Author: vpaturet

application/src/test/java/org/opentripplanner/updater/trip/siri/SiriEtBuilder.java

@@ -25,6 +25,7 @@
 import uk.org.siri.siri21.StopAssignmentStructure;
 import uk.org.siri.siri21.StopPointRefStructure;
 import uk.org.siri.siri21.VehicleJourneyRef;
+import uk.org.siri.siri21.VehicleModesEnumeration;
 
 /**
  * This is a helper class for constucting Siri ET messages to use in tests.
@@ -97,6 +98,25 @@ public SiriEtBuilder withLineRef(String lineRef) {
     return this;
   }
 
+  public SiriEtBuilder withExternalLineRef(String externalLineRef) {
+    var ref = new LineRef();
+    ref.setValue(externalLineRef);
+    evj.setExternalLineRef(ref);
+    return this;
+  }
+
+  public SiriEtBuilder withPublishedLineName(String lineName) {
+    var name = new NaturalLanguageStringStructure();
+    name.setValue(lineName);
+    evj.getPublishedLineNames().add(name);
+    return this;
+  }
+
+  public SiriEtBuilder withVehicleMode(VehicleModesEnumeration mode) {
+    evj.getVehicleModes().add(mode);
+    return this;
+  }
+
   public SiriEtBuilder withRecordedCalls(
     Function<RecordedCallsBuilder, RecordedCallsBuilder> producer
   ) {

application/src/test/java/org/opentripplanner/updater/trip/siri/moduletests/extrajourney/ExtraJourneyTest.java

@@ -11,6 +11,8 @@
 import org.opentripplanner.transit.model._data.TransitTestEnvironment;
 import org.opentripplanner.transit.model._data.TransitTestEnvironmentBuilder;
 import org.opentripplanner.transit.model._data.TripInput;
+import org.opentripplanner.transit.model.basic.SubMode;
+import org.opentripplanner.transit.model.basic.TransitMode;
 import org.opentripplanner.transit.model.network.Route;
 import org.opentripplanner.transit.model.organization.Operator;
 import org.opentripplanner.transit.model.site.RegularStop;
@@ -22,12 +24,14 @@
 import org.opentripplanner.updater.trip.RealtimeTestConstants;
 import org.opentripplanner.updater.trip.SiriTestHelper;
 import org.opentripplanner.updater.trip.siri.SiriEtBuilder;
+import uk.org.siri.siri21.VehicleModesEnumeration;
 
 class ExtraJourneyTest implements RealtimeTestConstants {
 
   private static final String ADDED_TRIP_ID = "newJourney";
   private static final String OPERATOR_ID = "operatorId";
   private static final String ROUTE_ID = "routeId";
+  private static final String RAIL_ROUTE_ID = "railRouteId";
 
   private final TransitTestEnvironmentBuilder ENV_BUILDER = TransitTestEnvironment.of();
   private final RegularStop STOP_A = ENV_BUILDER.stop(STOP_A_ID);
@@ -43,6 +47,15 @@ class ExtraJourneyTest implements RealtimeTestConstants {
     .addStop(STOP_A, "0:00:10", "0:00:11")
     .addStop(STOP_B, "0:00:20", "0:00:21");
 
+  private final Route RAIL_ROUTE = ENV_BUILDER.route(RAIL_ROUTE_ID, r ->
+    r.withMode(TransitMode.RAIL)
+  );
+
+  private final TripInput RAIL_TRIP_INPUT = TripInput.of("railTrip1")
+    .withRoute(RAIL_ROUTE)
+    .addStop(STOP_A, "0:00:10", "0:00:11")
+    .addStop(STOP_B, "0:00:20", "0:00:21");
+
   @Test
   void testAddJourneyWithExistingRoute() {
     var env = ENV_BUILDER.addTrip(TRIP_1_INPUT).build();
@@ -242,6 +255,123 @@ void testReplaceJourneyWithoutEstimatedVehicleJourneyCode() {
     assertFailure(UpdateError.UpdateErrorType.UNKNOWN, result);
   }
 
+  @Test
+  void testAddJourneyWithNewRouteAndShortName() {
+    var env = ENV_BUILDER.addTrip(TRIP_1_INPUT).build();
+    var siri = SiriTestHelper.of(env);
+
+    String newRouteRef = "newRouteForShortName";
+    var updates = createValidAddedJourney(siri)
+      .withLineRef(newRouteRef)
+      .withPublishedLineName("L1")
+      .buildEstimatedTimetableDeliveries();
+
+    var result = siri.applyEstimatedTimetable(updates);
+    assertEquals(1, result.successful());
+
+    Route newRoute = env.transitService().getRoute(id(newRouteRef));
+    assertNotNull(newRoute);
+    assertEquals(
+      "L1",
+      newRoute.getShortName(),
+      "PublishedLineName should be mapped to the new route short name"
+    );
+  }
+
+  @Test
+  void testAddJourneyWithNewRouteAndOperator() {
+    var env = ENV_BUILDER.addTrip(TRIP_1_INPUT).build();
+    var siri = SiriTestHelper.of(env);
+
+    String newRouteRef = "newRouteForOperator";
+    var updates = createValidAddedJourney(siri)
+      .withLineRef(newRouteRef)
+      .buildEstimatedTimetableDeliveries();
+
+    var result = siri.applyEstimatedTimetable(updates);
+    assertEquals(1, result.successful());
+
+    Trip trip = env.transitService().getTrip(id(ADDED_TRIP_ID));
+    assertNotNull(trip);
+    assertEquals(
+      OPERATOR,
+      trip.getOperator(),
+      "The new trip is assigned to the operator specified in OperatorRef"
+    );
+  }
+
+  @Test
+  void testAddJourneyWithNewRouteResolvesAgencyFromOperator() {
+    var env = ENV_BUILDER.addTrip(TRIP_1_INPUT).build();
+    var siri = SiriTestHelper.of(env);
+
+    String newRouteRef = "newRouteForAgency";
+    var updates = createValidAddedJourney(siri)
+      .withLineRef(newRouteRef)
+      .buildEstimatedTimetableDeliveries();
+
+    var result = siri.applyEstimatedTimetable(updates);
+    assertEquals(1, result.successful());
+
+    Route newRoute = env.transitService().getRoute(id(newRouteRef));
+    assertNotNull(newRoute);
+    assertEquals(
+      ROUTE.getAgency(),
+      newRoute.getAgency(),
+      "The new route agency should be mapped to the agency of any existing route that has the same operator"
+    );
+  }
+
+  @Test
+  void testAddJourneyBusReplacingRailHasRailReplacementSubmode() {
+    var env = ENV_BUILDER.addTrip(TRIP_1_INPUT).addTrip(RAIL_TRIP_INPUT).build();
+    var siri = SiriTestHelper.of(env);
+
+    String newRouteRef = "busReplacementRoute";
+    var updates = createValidAddedJourney(siri)
+      .withLineRef(newRouteRef)
+      .withExternalLineRef(RAIL_ROUTE_ID)
+      .withVehicleMode(VehicleModesEnumeration.BUS)
+      .buildEstimatedTimetableDeliveries();
+
+    var result = siri.applyEstimatedTimetable(updates);
+    assertEquals(1, result.successful());
+
+    Route newRoute = env.transitService().getRoute(id(newRouteRef));
+    assertNotNull(newRoute);
+    assertEquals(TransitMode.BUS, newRoute.getMode());
+    assertEquals(
+      SubMode.of("railReplacementBus"),
+      newRoute.getNetexSubmode(),
+      "When an added bus trip is assigned to an existing rail route, the submode should be 'railReplacementBus'"
+    );
+  }
+
+  @Test
+  void testAddJourneyRailReplacingRailHasReplacementRailSubmode() {
+    var env = ENV_BUILDER.addTrip(TRIP_1_INPUT).addTrip(RAIL_TRIP_INPUT).build();
+    var siri = SiriTestHelper.of(env);
+
+    String newRouteRef = "railReplacementRoute";
+    var updates = createValidAddedJourney(siri)
+      .withLineRef(newRouteRef)
+      .withExternalLineRef(RAIL_ROUTE_ID)
+      .withVehicleMode(VehicleModesEnumeration.RAIL)
+      .buildEstimatedTimetableDeliveries();
+
+    var result = siri.applyEstimatedTimetable(updates);
+    assertEquals(1, result.successful());
+
+    Route newRoute = env.transitService().getRoute(id(newRouteRef));
+    assertNotNull(newRoute);
+    assertEquals(TransitMode.RAIL, newRoute.getMode());
+    assertEquals(
+      SubMode.of("replacementRailService"),
+      newRoute.getNetexSubmode(),
+      "When an added rail trip is assigned to an existing rail route, the submode should be 'replacementRailService'"
+    );
+  }
+
   private SiriEtBuilder createValidAddedJourney(SiriTestHelper siri) {
     return siri
       .etBuilder()