Fix alert filtering

HSLdevcom · Feb 28, 2023 · 71db701 · 71db701
1 parent 095ffe6
commit 71db701
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/...ava/org/opentripplanner/ext/legacygraphqlapi/datafetchers/LegacyGraphQLQueryTypeImpl.java b/...ava/org/opentripplanner/ext/legacygraphqlapi/datafetchers/LegacyGraphQLQueryTypeImpl.java
@@ -118,26 +118,26 @@ public DataFetcher<Iterable<TransitAlert>> alerts() {
       var args = new LegacyGraphQLTypes.LegacyGraphQLQueryTypeAlertsArgs(
         environment.getArguments()
       );
-      List<String> severities = args.getLegacyGraphQLSeverityLevel() == null
+      List<LegacyGraphQLTypes.LegacyGraphQLAlertSeverityLevelType> severities = args.getLegacyGraphQLSeverityLevel() ==
+        null
         ? null
         : (
           (List<LegacyGraphQLTypes.LegacyGraphQLAlertSeverityLevelType>) args.getLegacyGraphQLSeverityLevel()
         ).stream()
-          .map(Enum::name)
           .toList();
-      List<String> effects = args.getLegacyGraphQLEffect() == null
+      List<LegacyGraphQLTypes.LegacyGraphQLAlertEffectType> effects = args.getLegacyGraphQLEffect() ==
+        null
         ? null
         : (
           (List<LegacyGraphQLTypes.LegacyGraphQLAlertEffectType>) args.getLegacyGraphQLEffect()
         ).stream()
-          .map(Enum::name)
           .toList();
-      List<String> causes = args.getLegacyGraphQLCause() == null
+      List<LegacyGraphQLTypes.LegacyGraphQLAlertCauseType> causes = args.getLegacyGraphQLCause() ==
+        null
         ? null
         : (
           (List<LegacyGraphQLTypes.LegacyGraphQLAlertCauseType>) args.getLegacyGraphQLCause()
         ).stream()
-          .map(Enum::name)
           .toList();
       return alerts
         .stream()