Merge pull request #98 from opentripplanner/alerts

Add query for fetching alerts

Author: leonardehrenfried

src/main/java/org/opentripplanner/client/OtpApiClient.java

@@ -17,6 +17,7 @@
 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.Alert;
 import org.opentripplanner.client.model.Pattern;
 import org.opentripplanner.client.model.Route;
 import org.opentripplanner.client.model.Stop;
@@ -152,6 +153,13 @@ public List<Stop> stopSearch(String nameMask) throws IOException {
     return deserializeList(jsonNode, listType(Stop.class), "/data/stops");
   }
 
+  /** Get all alerts. */
+  public List<Alert> alerts() throws IOException {
+    var query = GraphQLQueries.alerts();
+    final var jsonNode = sendRequest(query);
+    return deserializeList(jsonNode, listType(Alert.class), "/data/alerts");
+  }
+
   private <T> T deserialize(JsonNode jsonNode, String path, Class<T> clazz) throws IOException {
     try {
       var plan = jsonNode.at(path);

src/main/java/org/opentripplanner/client/model/Alert.java

@@ -0,0 +1,13 @@
+package org.opentripplanner.client.model;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import java.util.Objects;
+
+public record Alert(
+    @JsonProperty("alertHeaderText") String header,
+    @JsonProperty("alertDescriptionText") String description) {
+  public Alert {
+    Objects.requireNonNull(header);
+    Objects.requireNonNull(description);
+  }
+}

src/main/java/org/opentripplanner/client/query/GraphQLQueries.java

@@ -33,6 +33,10 @@ public static String stops() {
     return loadQuery("stops");
   }
 
+  public static String alerts() {
+    return loadQuery("alerts");
+  }
+
   private static String loadQuery(String name) {
     var is =
         GraphQLQueries.class

src/main/resources/queries/alerts.graphql

@@ -0,0 +1,9 @@
+query {
+    alerts {
+        alertHeaderText
+        alertDescriptionText
+        alertSeverityLevel
+        alertCause
+        alertSeverityLevel
+     }
+}

src/test/java/org/opentripplanner/IntegrationTest.java

@@ -64,7 +64,13 @@ public void plan() throws IOException {
 
     var leg = result.itineraries().get(0).legs().get(0);
 
-    var transitLeg = result.transitItineraries().get(0).transitLegs().get(0);
+    var transitLeg =
+        result.transitItineraries().stream()
+            .filter(i -> i.legs().stream().anyMatch(l -> !l.intermediatePlaces().isEmpty()))
+            .findFirst()
+            .get()
+            .transitLegs()
+            .get(0);
     assertFalse(transitLeg.from().stop().isEmpty());
     assertNotNull(transitLeg.from().coordinate());
     assertNotNull(transitLeg.from().point());
@@ -365,6 +371,16 @@ public void stops() throws IOException {
     assertNotNull(stop.id());
   }
 
+  @Test
+  public void alerts() throws IOException {
+    var result = client.alerts();
+
+    LOG.info("Received {} alerts", result.size());
+
+    assertNotNull(result);
+    assertFalse(result.isEmpty());
+  }
+
   @Disabled
   @Test
   public void seattleFares() throws IOException {