fix the tests

mbta · Jul 5, 2024 · f4dfd3d · f4dfd3d
1 parent a2cef00
commit f4dfd3d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test/dotcom_web/controllers/trip_plan_controller_test.exs b/test/dotcom_web/controllers/trip_plan_controller_test.exs
@@ -812,7 +812,8 @@ defmodule DotcomWeb.TripPlanControllerTest do
     end
 
     test "doesn't set external_agency_name flag for regular routes", %{itineraries: itineraries} do
-      expect(Routes.Repo.Mock, :get, fn id ->
+      # called variable number of times, depending on the generated itineraries
+      stub(Routes.Repo.Mock, :get, fn id ->
         %Routes.Route{id: id}
       end)
 
@@ -840,9 +841,8 @@ defmodule DotcomWeb.TripPlanControllerTest do
           %{i | legs: legs}
         end)
 
-      # set up leg route to not be present in the V3 API, causing the fallback
-      # to populate the external_agency_name value
-      expect(Routes.Repo.Mock, :get, fn _ ->
+      # called variable number of times, depending on the generated itineraries
+      stub(Routes.Repo.Mock, :get, fn id ->
         nil
       end)