Deduplicate setup_with_mocks() calls in WsServerTest

electric-sql · Apr 23, 2024 · 4169713 · 4169713
1 parent 4170143
commit 4169713
Showing 1 changed file with 6 additions and 11 deletions.
diff --git a/components/electric/test/electric/satellite/ws_server_test.exs b/components/electric/test/electric/satellite/ws_server_test.exs
@@ -34,17 +34,6 @@ defmodule Electric.Satellite.WebsocketServerTest do
 
   import Mock
 
-  setup_with_mocks([
-    {Electric.Postgres.Repo, [:passthrough],
-     checkout: fn fun -> fun.() end,
-     transaction: fn fun -> fun.() end,
-     checked_out?: fn -> true end,
-     query: fn _, _ -> {:ok, %Postgrex.Result{columns: nil, rows: []}} end,
-     query!: fn _, _ -> %Postgrex.Result{columns: nil, rows: []} end}
-  ]) do
-    %{}
-  end
-
   setup ctx do
     ctx =
       ctx
@@ -75,6 +64,12 @@ defmodule Electric.Satellite.WebsocketServerTest do
   end
 
   setup_with_mocks([
+    {Electric.Postgres.Repo, [:passthrough],
+     checkout: fn fun -> fun.() end,
+     transaction: fn fun -> fun.() end,
+     checked_out?: fn -> true end,
+     query: fn _, _ -> {:ok, %Postgrex.Result{columns: nil, rows: []}} end,
+     query!: fn _, _ -> %Postgrex.Result{columns: nil, rows: []} end},
     {Electric.Satellite.ClientReconnectionInfo, [:passthrough],
      restore_cache_for_client: fn _, _ -> :ok end},
     {SatelliteConnector, [:passthrough],