defmodule Aprsme.SpatialPubSubTest do
  use ExUnit.Case, async: false

  alias Aprsme.SpatialPubSub

  describe "viewport registration" do
    test "date-line-crossing viewport registers without timeout" do
      bounds = %{north: 10.0, south: -10.0, east: -170.0, west: 170.0}
      client_id = "test_dateline_#{:rand.uniform(100_000)}"

      # Before the fix, west=170 east=-170 generated ~341 grid cells.
      # Now it correctly generates ~21 cells.
      assert {:ok, _topic} = SpatialPubSub.register_viewport(client_id, bounds)

      SpatialPubSub.unregister_client(client_id)
    end

    test "normal viewport registers successfully" do
      bounds = %{north: 41.0, south: 40.0, east: -73.0, west: -74.0}
      client_id = "test_normal_#{:rand.uniform(100_000)}"

      assert {:ok, _topic} = SpatialPubSub.register_viewport(client_id, bounds)

      SpatialPubSub.unregister_client(client_id)
    end

    test "update_viewport succeeds after registration" do
      bounds = %{north: 41.0, south: 40.0, east: -73.0, west: -74.0}
      new_bounds = %{north: 42.0, south: 41.0, east: -72.0, west: -73.0}
      client_id = "test_update_#{:rand.uniform(100_000)}"

      {:ok, _topic} = SpatialPubSub.register_viewport(client_id, bounds)
      assert :ok = SpatialPubSub.update_viewport(client_id, new_bounds)

      SpatialPubSub.unregister_client(client_id)
    end
  end

  describe "stats" do
    test "get_stats returns expected keys" do
      stats = SpatialPubSub.get_stats()

      assert is_map(stats)
      assert Map.has_key?(stats, :total_packets)
      assert Map.has_key?(stats, :grid_cells)
    end

    test "grid cells increase after registering viewport" do
      bounds = %{north: 41.0, south: 40.0, east: -73.0, west: -74.0}
      client_id = "test_stats_#{:rand.uniform(100_000)}"

      before_stats = SpatialPubSub.get_stats()
      {:ok, _topic} = SpatialPubSub.register_viewport(client_id, bounds)
      after_stats = SpatialPubSub.get_stats()

      assert after_stats.grid_cells >= before_stats.grid_cells

      SpatialPubSub.unregister_client(client_id)
    end
  end
end