Conditionally initalise worker state

lib/example/worker.ex

@@ -3,12 +3,6 @@ defmodule Example.Worker do
 
   alias Example.DefaultService
 
-  # When using ElixirLS, defining the service at compile time will result in an
-  # error because ElixirLS always compiles using MIX_ENV=test which mean @service
-  # will always be set to MockService, which does not have `foo/0`
-  # @service Application.get_env(:example, :service, DefaultService)
-  # @service DefaultService
-
   def service() do
     Application.get_env(:example, :service, DefaultService)
   end
@@ -21,28 +15,33 @@ defmodule Example.Worker do
     GenServer.call(__MODULE__, :get_foo)
   end
 
+  def initialise_foo() do
+    pid = Process.whereis(__MODULE__)
+    send(pid, :initialise_foo)
+  end
+
   def init(_init_arg) do
     initial_state = "no foo for you"
-    {:ok, initial_state, {:continue, :get_foo_from_service}}
-  end
-
-  def handle_continue(:get_foo_from_service, _state) do
-    # And here lies the problem. We want to call our service to get
-    # whatever inital state it provides, but in doing so, we break
-    # in the test environment because the MockService doesn't have
-    # a function called `foo/0` until it can be defined in the expects
-    # block within the test - by that time, this code has already
-    # been executed because this GenServer is part of the staticly
-    # defined supervision tree in `application.ex`.
-
-    value_of_foo =
-      if function_exported?(service(), :foo, 0) do
-        service().foo()
-      else
-        "#{inspect(service())} does not support foo"
-      end
-
-    {:noreply, value_of_foo}
+
+    case service() do
+      DefaultService ->
+        {:ok, initial_state, {:continue, :get_foo_from_service}}
+
+      _ ->
+        {:ok, initial_state}
+    end
+  end
+
+  def handle_continue(:get_foo_from_service, state) do
+    send(self(), :initialise_foo)
+
+    {:noreply, state}
+  end
+
+  def handle_info(:initialise_foo, _state) do
+    IO.inspect("initialising foo now")
+    new_state = service().foo()
+    {:noreply, new_state}
   end
 
   def handle_call(:get_foo, _from, state) do

test/test_helper.exs

@@ -1 +1,3 @@
+Mox.defmock(Example.MockService, for: Example.ServiceBehaviour)
+
 ExUnit.start()

test/worker_test.exs

@@ -3,31 +3,30 @@ defmodule Example.WorkerTest do
   import Mox
   alias Example.Worker
 
-  describe "default service" do
-    test "returns default service foo" do
-      assert Worker.get_foo() =~ ~s(default says foo)
-    end
+  setup do
+    Worker.initialise_foo()
+
+    :ok
   end
 
-  describe "mocked service" do
-    setup do
-      # Normally you would add this to `test_helper.ex`, or `support/mocks.ex
-      Mox.defmock(Example.MockService, for: Example.ServiceBehaviour)
+  setup :verify_on_exit!
 
+  describe "mocked service" do
+    @tag :mocked_service
+    test "returns mocked service foo" do
       Example.MockService
-      |> expect(:foo, fn -> "setup all says foo" end)
+      |> expect(:foo, fn -> "mock says foo" end)
+      |> allow(self(), Worker)
 
-      :ok
+      assert Worker.get_foo() =~ ~s(mock says foo)
     end
 
-    setup :verify_on_exit!
-
-    test "returns mocked service foo" do
+    test "returns mocked service moo" do
       Example.MockService
-      |> expect(:foo, fn -> "mock says foo" end)
-      |> allow(self(), Process.whereis(Worker))
+      |> expect(:foo, fn -> "mock says moo" end)
+      |> allow(self(), Worker)
 
-      assert Worker.get_foo() =~ ~s(mock says foo)
+      assert Worker.get_foo() =~ ~s(mock says moo)
     end
   end
 end