feat: update schedule relationship of Cancelled STUs

mbta · Jan 10, 2025 · 2ca6045 · 2ca6045
1 parent 8548c96
commit 2ca6045
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/lib/concentrate/parser/gtfs_realtime_enhanced.ex b/lib/concentrate/parser/gtfs_realtime_enhanced.ex
@@ -150,12 +150,19 @@ defmodule Concentrate.Parser.GTFSRealtimeEnhanced do
 
             boarding_status = decode_boarding_status(Map.get(stu, "boarding_status"))
 
+            schedule_relationship =
+              if boarding_status == "Cancelled" do
+                :SKIPPED
+              else
+                schedule_relationship(Map.get(stu, "schedule_relationship"))
+              end
+
             StopTimeUpdate.new(
               trip_id:
                 if(descriptor = Map.get(trip_update, "trip"), do: Map.get(descriptor, "trip_id")),
               stop_id: Map.get(stu, "stop_id"),
               stop_sequence: Map.get(stu, "stop_sequence"),
-              schedule_relationship: schedule_relationship(Map.get(stu, "schedule_relationship")),
+              schedule_relationship: schedule_relationship,
               arrival_time: arrival_time,
               departure_time: departure_time,
               passthrough_time: Map.get(stu, "passthrough_time"),

diff --git a/test/concentrate/parser/gtfs_realtime_enhanced_test.exs b/test/concentrate/parser/gtfs_realtime_enhanced_test.exs
@@ -221,6 +221,20 @@ defmodule Concentrate.Parser.GTFSRealtimeEnhancedTest do
       assert StopTimeUpdate.status(stop_update) == "UNIQUE STATUS"
     end
 
+    test "handles cancelled status" do
+      update = %{
+        "trip" => %{},
+        "stop_time_update" => [
+          %{
+            "boarding_status" => "CANCELLED"
+          }
+        ]
+      }
+
+      [_td, stop_update] = decode_trip_update(update, %Options{})
+      assert StopTimeUpdate.schedule_relationship(stop_update) == :SKIPPED
+    end
+
     test "can handle platform id information" do
       update = %{
         "trip" => %{},