Fix WsServerTest

components/electric/test/electric/satellite/ws_server_test.exs

@@ -34,6 +34,13 @@ defmodule Electric.Satellite.WebsocketServerTest do
 
   import Mock
 
+  setup_with_mocks([
+    {Electric.Replication.Postgres.Client, [:passthrough],
+     with_pool: fn _, _ -> :ok end, pooled_query!: fn _, _, _ -> :ok end}
+  ]) do
+    %{}
+  end
+
   setup ctx do
     ctx =
       ctx
@@ -44,20 +51,23 @@ defmodule Electric.Satellite.WebsocketServerTest do
       )
       |> Map.put_new(:allowed_unacked_txs, 30)
 
+    connector_config = [origin: "test-origin", connection: []]
     port = 55133
 
     plug =
       {Electric.Plug.SatelliteWebsocketPlug,
        auth_provider: Auth.provider(),
-       connector_config: [origin: "fake_origin"],
+       connector_config: connector_config,
        subscription_data_fun: ctx.subscription_data_fun,
        allowed_unacked_txs: ctx.allowed_unacked_txs}
 
     start_link_supervised!({Bandit, port: port, plug: plug})
 
     server_id = Electric.instance_id()
 
-    {:ok, port: port, server_id: server_id}
+    start_link_supervised!({Electric.Satellite.ClientReconnectionInfo, connector_config})
+
+    %{port: port, server_id: server_id}
   end
 
   setup_with_mocks([
@@ -83,7 +93,7 @@ defmodule Electric.Satellite.WebsocketServerTest do
       stream_transactions: fn _, _, _ -> [] end
     }
   ]) do
-    {:ok, %{}}
+    %{}
   end
 
   # make sure server is cleaning up connections

components/electric/test/support/setup_helpers.ex

@@ -9,7 +9,7 @@ defmodule ElectricTest.SetupHelpers do
   Starts SchemaCache process with a given origin, and
   immediately fills it from given SQL.
   """
-  def start_schema_cache(origin \\ "fake_origin", migrations) do
+  def start_schema_cache(origin \\ "test-origin", migrations) do
     backend = Electric.Postgres.MockSchemaLoader.start_link(migrations: migrations)
 
     start_supervised!(