fix: StopTimes.get is probably supposed to be StopTimes.arrival_depar…

…ture
mbta · Jul 20, 2022 · 6ef6552 · 6ef6552
1 parent 51f419a
commit 6ef6552
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/lib/concentrate/group_filter/scheduled_stop_times.ex b/lib/concentrate/group_filter/scheduled_stop_times.ex
@@ -62,7 +62,7 @@ defmodule Concentrate.GroupFilter.ScheduledStopTimes do
       trip_id = StopTimeUpdate.trip_id(stop_time_update)
       stop_sequence = StopTimeUpdate.stop_sequence(stop_time_update)
 
-      case gtfs_stop_times.get(trip_id, stop_sequence, trip_date) do
+      case gtfs_stop_times.arrival_departure(trip_id, stop_sequence, trip_date) do
         {arrival_time, departure_time} ->
           stop_time_update
           |> StopTimeUpdate.update_arrival_time(arrival_time)

diff --git a/test/concentrate/group_filter/scheduled_stop_times_test.exs b/test/concentrate/group_filter/scheduled_stop_times_test.exs
@@ -5,10 +5,10 @@ defmodule Concentrate.GroupFilter.ScheduledStopTimesTest do
   alias Concentrate.{TripDescriptor, StopTimeUpdate}
 
   defmodule FakeStopTimes do
-    def get("trip1", 10, {2022, 1, 1}), do: {1_610_000_000, 1_610_000_001}
-    def get("trip1", 20, {2022, 1, 1}), do: {1_620_000_000, 1_620_000_001}
-    def get("trip1", 30, {2022, 1, 1}), do: {1_630_000_000, 1_630_000_001}
-    def get(_, _, _), do: :unknown
+    def arrival_departure("trip1", 10, {2022, 1, 1}), do: {1_610_000_000, 1_610_000_001}
+    def arrival_departure("trip1", 20, {2022, 1, 1}), do: {1_620_000_000, 1_620_000_001}
+    def arrival_departure("trip1", 30, {2022, 1, 1}), do: {1_630_000_000, 1_630_000_001}
+    def arrival_departure(_, _, _), do: :unknown
   end
 
   # see config/test.exs