Updated realtime schedules to use prediction repo

lib/dotcom/realtime_schedule.ex

@@ -11,7 +11,6 @@ defmodule Dotcom.RealtimeSchedule do
   alias Dotcom.JsonHelpers
   alias Dotcom.TransitNearMe
   alias Predictions.Prediction
-  alias Predictions.Repo, as: PredictionsRepo
   alias RoutePatterns.RoutePattern
   alias Routes.Repo, as: RoutesRepo
   alias Routes.Route
@@ -25,10 +24,12 @@ defmodule Dotcom.RealtimeSchedule do
 
   @predicted_schedules_per_stop 2
 
+  @predictions_repo Application.compile_env!(:dotcom, :repo_modules)[:predictions]
+
   @default_opts [
     stops_fn: &StopsRepo.get/1,
     routes_fn: &RoutesRepo.by_stop_with_route_pattern/1,
-    predictions_fn: &PredictionsRepo.all_no_cache/1,
+    predictions_fn: Function.capture(@predictions_repo, :all_no_cache, 1),
     schedules_fn: &SchedulesRepo.by_route_ids/2,
     alerts_fn: &Alerts.Repo.by_route_ids/2
   ]

test/dotcom/realtime_schedule_test.exs

@@ -1,6 +1,8 @@
 defmodule Dotcom.RealtimeScheduleTest do
   use ExUnit.Case
 
+  import Mox
+
   alias Alerts.Alert
   alias Alerts.InformedEntity, as: IE
   alias Alerts.InformedEntitySet, as: IESet
@@ -175,10 +177,11 @@ defmodule Dotcom.RealtimeScheduleTest do
   end
 
   test "stop_data/3 returns stop" do
+    expect(Predictions.Repo.Mock, :all_no_cache, 3, fn _ -> @predictions end)
+
     opts = [
       stops_fn: fn _ -> @stop end,
       routes_fn: fn _ -> @route_with_patterns end,
-      predictions_fn: fn _ -> @predictions end,
       schedules_fn: fn _, _ -> @schedules end,
       alerts_fn: fn _, _ -> @alerts end
     ]
@@ -301,10 +304,11 @@ defmodule Dotcom.RealtimeScheduleTest do
   end
 
   test "stop_data/3 returns nil" do
+    expect(Predictions.Repo.Mock, :all_no_cache, fn _ -> [] end)
+
     opts = [
       stops_fn: fn _ -> nil end,
       routes_fn: fn _ -> [] end,
-      predictions_fn: fn _ -> [] end,
       schedules_fn: fn _, _ -> [] end,
       alerts_fn: fn _, _ -> [] end
     ]