fix tests

mbta · Dec 18, 2024 · bf902d8 · bf902d8
1 parent e293a16
commit bf902d8
Showing 1 changed file with 33 additions and 0 deletions.
diff --git a/test/dotcom_web/live/trip_planner_test.exs b/test/dotcom_web/live/trip_planner_test.exs
@@ -13,6 +13,15 @@ defmodule DotcomWeb.Live.TripPlannerTest do
     expect(OpenTripPlannerClient.Mock, :plan, fn _ ->
       {:ok, %OpenTripPlannerClient.Plan{itineraries: itineraries}}
     end)
+
+    # For certain routes, Dotcom.TripPlan.Alerts.mode_entities/1 is called to help fetch the associated alerts
+    stub(MBTA.Api.Mock, :get_json, fn "/trips/" <> id, [] ->
+      %JsonApi{
+        data: [
+          Test.Support.Factories.MBTA.Api.build(:trip_item, %{id: id})
+        ]
+      }
+    end)
   end
 
   defp stub_populated_otp_results do
@@ -227,6 +236,14 @@ defmodule DotcomWeb.Live.TripPlannerTest do
          }}
       end)
 
+      stub(MBTA.Api.Mock, :get_json, fn "/trips/" <> id, [] ->
+        %JsonApi{
+          data: [
+            Test.Support.Factories.MBTA.Api.build(:trip_item, %{id: id})
+          ]
+        }
+      end)
+
       {:ok, view, _html} = live(conn, ~p"/preview/trip-planner?#{params}")
 
       render_async(view)
@@ -253,6 +270,14 @@ defmodule DotcomWeb.Live.TripPlannerTest do
          }}
       end)
 
+      stub(MBTA.Api.Mock, :get_json, fn "/trips/" <> id, [] ->
+        %JsonApi{
+          data: [
+            Test.Support.Factories.MBTA.Api.build(:trip_item, %{id: id})
+          ]
+        }
+      end)
+
       {:ok, view, _html} = live(conn, ~p"/preview/trip-planner?#{params}")
 
       render_async(view)
@@ -276,6 +301,14 @@ defmodule DotcomWeb.Live.TripPlannerTest do
          }}
       end)
 
+      stub(MBTA.Api.Mock, :get_json, fn "/trips/" <> id, [] ->
+        %JsonApi{
+          data: [
+            Test.Support.Factories.MBTA.Api.build(:trip_item, %{id: id})
+          ]
+        }
+      end)
+
       {:ok, view, _html} = live(conn, ~p"/preview/trip-planner?#{params}")
 
       render_async(view)