Add test for mode-specific transfers.

HSLdevcom · Nov 29, 2024 · 44e4d31 · 44e4d31
1 parent 4c882f4
commit 44e4d31
Showing 1 changed file with 57 additions and 2 deletions.
diff --git a/...n/src/test/java/org/opentripplanner/graph_builder/module/DirectTransferGeneratorTest.java b/...n/src/test/java/org/opentripplanner/graph_builder/module/DirectTransferGeneratorTest.java
@@ -196,7 +196,7 @@ public void testMultipleRequestsWithoutPatterns() {
     reqWalk.journey().transfer().setMode(StreetMode.WALK);
 
     var reqBike = new RouteRequest();
-    reqWalk.journey().transfer().setMode(StreetMode.BIKE);
+    reqBike.journey().transfer().setMode(StreetMode.BIKE);
 
     var transferRequests = List.of(reqWalk, reqBike);
 
@@ -223,7 +223,7 @@ public void testMultipleRequestsWithPatterns() {
     reqWalk.journey().transfer().setMode(StreetMode.WALK);
 
     var reqBike = new RouteRequest();
-    reqWalk.journey().transfer().setMode(StreetMode.BIKE);
+    reqBike.journey().transfer().setMode(StreetMode.BIKE);
 
     var transferRequests = List.of(reqWalk, reqBike);
 
@@ -250,6 +250,61 @@ public void testMultipleRequestsWithPatterns() {
     );
   }
 
+  @Test
+  public void testPathTransfersWithModesForMultipleRequestsWithPatterns() {
+    var reqWalk = new RouteRequest();
+    reqWalk.journey().transfer().setMode(StreetMode.WALK);
+
+    var reqBike = new RouteRequest();
+    reqBike.journey().transfer().setMode(StreetMode.BIKE);
+
+    var transferRequests = List.of(reqWalk, reqBike);
+
+    TestOtpModel model = model(true);
+    var graph = model.graph();
+    graph.hasStreets = true;
+    var timetableRepository = model.timetableRepository();
+
+    new DirectTransferGenerator(
+      graph,
+      timetableRepository,
+      DataImportIssueStore.NOOP,
+      MAX_TRANSFER_DURATION,
+      transferRequests
+    )
+      .buildGraph();
+
+    var walkTransfers = timetableRepository
+      .getAllPathTransfers()
+      .stream()
+      .filter(pathTransfer -> pathTransfer.getModes().contains(StreetMode.WALK))
+      .toList();
+    var bikeTransfers = timetableRepository
+      .getAllPathTransfers()
+      .stream()
+      .filter(pathTransfer -> pathTransfer.getModes().contains(StreetMode.BIKE))
+      .toList();
+    var carTransfers = timetableRepository
+      .getAllPathTransfers()
+      .stream()
+      .filter(pathTransfer -> pathTransfer.getModes().contains(StreetMode.CAR))
+      .toList();
+
+    assertTransfers(
+      walkTransfers,
+      tr(S0, 100, List.of(V0, V11), S11),
+      tr(S0, 100, List.of(V0, V21), S21),
+      tr(S11, 100, List.of(V11, V21), S21)
+    );
+    assertTransfers(
+      bikeTransfers,
+      tr(S0, 100, List.of(V0, V11), S11),
+      tr(S0, 100, List.of(V0, V21), S21),
+      tr(S11, 110, List.of(V11, V22), S22)
+    );
+    assertTransfers(carTransfers);
+  }
+
   @Test
   public void testTransferOnIsolatedStations() {
     var otpModel = model(true, false, true);