# Stop the RelationTracker - supervisor will restart it
relation_tracker_name = PublicationManager.RelationTracker.name(ctx.stack_id)
GenServer.stop(relation_tracker_name)                 # 653: async restart by supervisor

# After restart, the publication manager should repopulate from ShapeStatus.
# The publication should still have the relation.
assert_pub_tables(ctx, [ctx.relation], 2_000)         # 657: passes IMMEDIATELY

# Verify we can still remove the shape after the restart
PublicationManager.remove_shape(ctx.stack_id, shape_handle)  # 660: (EXIT) no process
assert_pub_tables(ctx, [], 2_000)

test "handles relation tracker restart", ctx do
  shape = generate_shape(ctx.relation_with_oid, @where_clause_1)

  {:ok, shape_handle} = Electric.ShapeCache.ShapeStatus.add_shape(ctx.stack_id, shape)
  assert :ok = PublicationManager.add_shape(ctx.stack_id, shape_handle, shape)
  assert_pub_tables(ctx, [ctx.relation])

  # Stop the RelationTracker - supervisor will restart it
  relation_tracker_name = PublicationManager.RelationTracker.name(ctx.stack_id)
  GenServer.stop(relation_tracker_name)

  # After restart, the publication manager should repopulate from ShapeStatus.
  # The publication should still have the relation.
  assert_pub_tables(ctx, [ctx.relation], 2_000)

  # Wait for the supervisor to actually restart the RelationTracker and finish
  # restoring before issuing further calls. The assert_pub_tables above passes
  # immediately (the relation is never removed), so it does NOT guarantee the
  # new process has registered yet — without this, remove_shape/2 races the
  # restart and hits the via-Registry name before the process exists.
  wait_until_registered(relation_tracker_name, 2_000)
  PublicationManager.RelationTracker.wait_for_restore(ctx.stack_id, timeout: 2_000)

  # Verify we can still remove the shape after the restart
  PublicationManager.remove_shape(ctx.stack_id, shape_handle)
  assert_pub_tables(ctx, [], 2_000)
end

defp wait_until_registered(name, timeout, start_time \\ :erlang.monotonic_time(:millisecond)) do
  case GenServer.whereis(name) do
    pid when is_pid(pid) ->
      pid

    nil ->
      if :erlang.monotonic_time(:millisecond) - start_time < timeout do
        Process.sleep(10)
        wait_until_registered(name, timeout, start_time)
      else
        flunk("RelationTracker was not re-registered within #{timeout}ms after restart")
      end
  end
end