Merge pull request #48 from raumobil/feat/agencies-query

Add agencies query
opentripplanner · Jun 12, 2024 · 32e3fa9 · 32e3fa9
2 parents c985415 + d4fbe54
commit 32e3fa9
Show file tree

Hide file tree

Showing 4 changed files with 39 additions and 0 deletions.
diff --git a/src/main/java/org/opentripplanner/client/OtpApiClient.java b/src/main/java/org/opentripplanner/client/OtpApiClient.java
@@ -16,6 +16,7 @@
 import org.apache.hc.client5.http.impl.classic.HttpClientBuilder;
 import org.apache.hc.core5.http.ContentType;
 import org.apache.hc.core5.http.io.entity.StringEntity;
+import org.opentripplanner.client.model.Agency;
 import org.opentripplanner.client.model.Pattern;
 import org.opentripplanner.client.model.Route;
 import org.opentripplanner.client.model.Stop;
@@ -109,6 +110,18 @@ public List<Pattern> patterns() throws IOException {
     return deserializeList(json, type, "/data/patterns");
   }
 
+  /**
+   * Return the list of agencies.
+   *
+   * @link <a href="https://docs.opentripplanner.org/api/dev-2.x/graphql-gtfs/queries/agencies">OTP
+   *     API docs</a>
+   */
+  public List<Agency> agencies() throws IOException {
+    var json = sendRequest(GraphQLQueries.agencies());
+    var type = listType(Agency.class);
+    return deserializeList(json, type, "/data/agencies");
+  }
+
   /**
    * Returns a Stop with limited information.
    *

diff --git a/src/main/java/org/opentripplanner/client/query/GraphQLQueries.java b/src/main/java/org/opentripplanner/client/query/GraphQLQueries.java
@@ -21,6 +21,10 @@ public static String patterns() {
     return loadQuery("patterns");
   }
 
+  public static String agencies() {
+    return loadQuery("agencies");
+  }
+
   public static String stop() {
     return loadQuery("stop");
   }

diff --git a/src/main/resources/queries/agencies.graphql b/src/main/resources/queries/agencies.graphql
@@ -0,0 +1,6 @@
+    query {
+        agencies {
+            gtfsId
+            name
+        }
+    }
diff --git a/src/test/java/org/opentripplanner/IntegrationTest.java b/src/test/java/org/opentripplanner/IntegrationTest.java
@@ -196,6 +196,22 @@ public void patterns() throws IOException {
         });
   }
 
+  @Test
+  public void agencies() throws IOException {
+
+    var result = client.agencies();
+
+    LOG.info("Received {} agencies", result.size());
+
+    assertFalse(result.isEmpty());
+
+    result.forEach(
+        agency -> {
+          assertNotNull(agency.name());
+          assertNotNull(agency.id());
+        });
+  }
+
   @Test
   public void stop() throws IOException {