Added updated the tests for transit near me

lib/dotcom/transit_near_me.ex

@@ -85,6 +85,8 @@ defmodule Dotcom.TransitNearMe do
     "place-hsmnl"
   ]
 
+  @predictions_repo Application.compile_env!(:dotcom, :repo_modules)[:predictions]
+
   @spec build(Address.t(), Keyword.t()) :: stops_with_distances
   def build(%Address{} = location, opts) do
     opts = Keyword.merge(@default_opts, opts)
@@ -382,7 +384,9 @@ defmodule Dotcom.TransitNearMe do
           PredictedSchedule.t()
         ]
   defp get_predicted_schedules(schedules, params, opts) do
-    predictions_fn = Keyword.get(opts, :predictions_fn, &Predictions.Repo.all/1)
+    predictions_fn =
+      Keyword.get(opts, :predictions_fn, Function.capture(@predictions_repo, :all, 1))
+
     now = Keyword.fetch!(opts, :now)
 
     predictions = predictions_fn.(params)

test/dotcom/transit_near_me_test.exs

@@ -1,6 +1,8 @@
 defmodule Dotcom.TransitNearMeTest do
   use ExUnit.Case
 
+  import Mox
+
   alias LocationService.Address
   alias Predictions.Prediction
   alias Routes.Route
@@ -725,7 +727,9 @@ defmodule Dotcom.TransitNearMeTest do
     }
 
     test "returns time data for the next 2 predictions" do
-      predictions_fn = fn _ -> [@prediction1, @prediction2, @prediction3] end
+      expect(Predictions.Repo.Mock, :all, fn _ ->
+        [@prediction1, @prediction2, @prediction3]
+      end)
 
       schedules_fn = fn _, _ ->
         [@schedule1, @schedule2, @schedule3]
@@ -787,15 +791,14 @@ defmodule Dotcom.TransitNearMeTest do
       actual =
         TransitNearMe.time_data_for_route_by_stop(@route.id, 1,
           schedules_fn: schedules_fn,
-          predictions_fn: predictions_fn,
           now: @now
         )
 
       assert actual == expected
     end
 
     test "returns no  data when schedules is empty" do
-      predictions_fn = fn _ -> [@prediction1] end
+      expect(Predictions.Repo.Mock, :all, fn _ -> [@prediction1] end)
 
       schedules_fn = fn _, _ ->
         []
@@ -804,15 +807,14 @@ defmodule Dotcom.TransitNearMeTest do
       actual =
         TransitNearMe.time_data_for_route_by_stop(@route.id, 1,
           schedules_fn: schedules_fn,
-          predictions_fn: predictions_fn,
           now: @now
         )
 
       assert %{"95" => [%{}]} = actual
     end
 
     test "return neither schedules nor predictions if date is outside rating" do
-      predictions_fn = fn _ -> [@prediction1] end
+      expect(Predictions.Repo.Mock, :all, fn _ -> [@prediction1] end)
 
       schedules_fn = fn _, _ ->
         {:error,
@@ -833,7 +835,6 @@ defmodule Dotcom.TransitNearMeTest do
       actual =
         TransitNearMe.time_data_for_route_by_stop(@route.id, 1,
           schedules_fn: schedules_fn,
-          predictions_fn: predictions_fn,
           now: @now
         )