chore: test async scheduler

Author: heywhy

lib/elasticlunr/scheduler/async.ex

@@ -13,6 +13,20 @@ defmodule Elasticlunr.Scheduler.Async do
     end
   end
 
+  @spec started?(Index.t() | String.t()) :: boolean()
+  def started?(%Index{name: name}), do: started?(name)
+
+  def started?(name) do
+    started_processes()
+    |> Enum.any?(fn
+      ^name ->
+        true
+
+      _ ->
+        false
+    end)
+  end
+
   @impl true
   def init(state), do: {:ok, state}
 
@@ -53,7 +67,7 @@ defmodule Elasticlunr.Scheduler.Async do
   defp blank_state, do: %{}
 
   defp start_if_not_started(%{name: name} = index) do
-    with false <- loaded?(name),
+    with false <- started?(name),
          {:ok, _} <- start(index) do
       :ok
     else
@@ -66,18 +80,7 @@ defmodule Elasticlunr.Scheduler.Async do
     DynamicSupervisor.start_child(SchedulerSupervisor, {__MODULE__, index})
   end
 
-  defp loaded?(name) do
-    loaded_indices()
-    |> Enum.any?(fn
-      ^name ->
-        true
-
-      _ ->
-        false
-    end)
-  end
-
-  defp loaded_indices do
+  defp started_processes do
     Process.active_processes(SchedulerSupervisor, SchedulerRegistry, __MODULE__)
   end
 end

test/scheduler/async_test.exs

@@ -0,0 +1,61 @@
+defmodule Elasticlunr.Scheduler.AsyncTest do
+  use ExUnit.Case
+
+  alias Elasticlunr.{Index, Scheduler}
+
+  require Logger
+
+  import Mox
+
+  setup :verify_on_exit!
+
+  setup do
+    log_level = Logger.level()
+    current_scheduler = Application.get_env(:elasticlunr, :scheduler)
+
+    Logger.configure(level: :info)
+
+    Mox.stub_with(Scheduler.Mock, Scheduler.Async)
+    Application.put_env(:elasticlunr, :scheduler, Scheduler.Mock)
+
+    on_exit(fn ->
+      Logger.configure(level: log_level)
+      Application.put_env(:elasticlunr, :scheduler, current_scheduler)
+    end)
+  end
+
+  setup context do
+    index = Index.add_field(Index.new(), "message")
+
+    Map.put(context, :index, index)
+  end
+
+  describe "working with async scheduler" do
+    test "pushes action to scheduler", %{index: index} do
+      expect(Scheduler.Mock, :push, fn ^index, :calculate_idf ->
+        :ok
+      end)
+
+      assert Index.add_documents(index, [%{message: "hello world"}])
+    end
+
+    test "starts a new process for index if none exists", %{index: index} do
+      index = Index.add_documents(index, [%{message: "hello world"}])
+
+      assert Scheduler.Async.started?(index)
+      # sleep to allow the async scheduler process the :calculate_idf task
+      Process.sleep(500)
+    end
+
+    test "handles the calculate_idf task", %{index: index} do
+      expect(Scheduler.Mock, :push, fn ^index, :calculate_idf ->
+        {:noreply, %{}} = Scheduler.Async.handle_cast({:calculate_idf, index}, %{})
+        :ok
+      end)
+
+      index = Index.add_documents(index, [%{message: "hello world"}])
+
+      assert [%{matched: 1, positions: %{"message" => [{0, 5}]}}] = Index.search(index, "hello")
+    end
+  end
+end

test/test_helper.exs

@@ -2,4 +2,6 @@ ExUnit.start()
 Faker.start()
 
 Mox.defmock(Elasticlunr.Storage.Mock, for: Elasticlunr.Storage.Provider)
+Mox.defmock(Elasticlunr.Scheduler.Mock, for: Elasticlunr.Scheduler.Provider)
+
 Application.put_env(:elasticlunr, :scheduler, Elasticlunr.Scheduler.Immediate)