Add agency to route

opentripplanner · Oct 31, 2023 · 3bc00aa · 3bc00aa
1 parent 9e2035f
commit 3bc00aa
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/main/java/org/opentripplanner/client/model/Route.java b/src/main/java/org/opentripplanner/client/model/Route.java
@@ -2,7 +2,8 @@
 
 import java.util.Optional;
 
-public record Route(Optional<String> shortName, Optional<String> longName, TransitMode mode) {
+public record Route(
+    Optional<String> shortName, Optional<String> longName, TransitMode mode, Agency agency) {
 
   /**
    * Either the short name (if it has one) or the long name.

diff --git a/src/test/java/org/opentripplanner/IntegrationTest.java b/src/test/java/org/opentripplanner/IntegrationTest.java
@@ -101,7 +101,11 @@ public void routes() throws IOException {
     LOG.info("Received {} routes", routes.size());
 
     assertFalse(routes.isEmpty());
-    routes.forEach(r -> assertFalse(r.name().isEmpty(), "Route %s has no name.".formatted(r)));
+    routes.forEach(
+        r -> {
+          assertFalse(r.name().isEmpty(), "Route %s has no name.".formatted(r));
+          assertFalse(r.agency().name().isEmpty());
+        });
   }
 
   @Test