aprs.me/test/aprsme/spatial_pubsub_test.exs

78 lines
2.8 KiB
Elixir

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 "duplicate packet prevention" do
test "does not subscribe to postgres PubSub topic" do
# SpatialPubSub used to subscribe to "postgres:aprsme_packets" causing
# duplicate broadcasts (once from PacketConsumer direct cast, once from PubSub).
# Verify it no longer subscribes.
# The SpatialPubSub process should NOT be in the subscriber list for postgres topic
spatial_pid = Process.whereis(SpatialPubSub)
subscribed? =
Aprsme.PubSub
|> Registry.lookup("postgres:aprsme_packets")
|> Enum.any?(fn {pid, _} -> pid == spatial_pid end)
refute subscribed?, "SpatialPubSub should not subscribe to postgres:aprsme_packets"
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