"""Tests for advertisement API routes.""" from datetime import datetime, timedelta, timezone from meshcore_hub.common.models import Advertisement, EventObserver class TestListAdvertisements: """Tests for GET /advertisements endpoint.""" def test_list_advertisements_empty(self, client_no_auth): """Test listing advertisements when database is empty.""" response = client_no_auth.get("/api/v1/advertisements") assert response.status_code == 200 data = response.json() assert data["items"] == [] assert data["total"] == 0 def test_list_advertisements_with_data(self, client_no_auth, sample_advertisement): """Test listing advertisements with data in database.""" response = client_no_auth.get("/api/v1/advertisements") assert response.status_code == 200 data = response.json() assert len(data["items"]) == 1 assert data["total"] == 1 assert data["items"][0]["public_key"] == sample_advertisement.public_key assert data["items"][0]["adv_type"] == sample_advertisement.adv_type def test_list_advertisements_with_observers( self, client_no_auth, api_db_session, receiver_node, ): """Test that observers list is included in advertisement response.""" from hashlib import md5 event_hash = md5(b"test-ad-observers").hexdigest() advert = Advertisement( public_key="obs123obs123obs123obs123obs123ob", name="ObservedAd", adv_type="REPEATER", received_at=datetime.now(timezone.utc), observer_node_id=receiver_node.id, event_hash=event_hash, ) api_db_session.add(advert) api_db_session.commit() observer = EventObserver( event_type="advertisement", event_hash=event_hash, observer_node_id=receiver_node.id, observed_at=datetime.now(timezone.utc), ) api_db_session.add(observer) api_db_session.commit() response = client_no_auth.get("/api/v1/advertisements") assert response.status_code == 200 data = response.json() assert len(data["items"]) == 1 item = data["items"][0] assert "observers" in item assert len(item["observers"]) == 1 def test_list_advertisements_with_node_tag_name( self, client_no_auth, api_db_session, sample_node_with_name_tag ): """Test that node_tag_name is resolved from name tags.""" advert = Advertisement( public_key=sample_node_with_name_tag.public_key, name="AdName", adv_type="CLIENT", received_at=datetime.now(timezone.utc), node_id=sample_node_with_name_tag.id, ) api_db_session.add(advert) api_db_session.commit() response = client_no_auth.get("/api/v1/advertisements") assert response.status_code == 200 data = response.json() assert len(data["items"]) == 1 assert data["items"][0]["node_tag_name"] == "Friendly Search Name" class TestGetAdvertisement: """Tests for GET /advertisements/{id} endpoint.""" def test_get_advertisement_success(self, client_no_auth, sample_advertisement): """Test getting a specific advertisement.""" response = client_no_auth.get( f"/api/v1/advertisements/{sample_advertisement.id}" ) assert response.status_code == 200 data = response.json() assert data["public_key"] == sample_advertisement.public_key def test_get_advertisement_not_found(self, client_no_auth): """Test getting a non-existent advertisement.""" response = client_no_auth.get("/api/v1/advertisements/nonexistent-id") assert response.status_code == 404 def test_get_advertisement_with_observers( self, client_no_auth, api_db_session, receiver_node, ): """Test that get includes observers list.""" from hashlib import md5 event_hash = md5(b"test-get-ad-observers").hexdigest() advert = Advertisement( public_key="getobs123getobs123getobs123getob", name="GetObservedAd", adv_type="REPEATER", received_at=datetime.now(timezone.utc), observer_node_id=receiver_node.id, event_hash=event_hash, ) api_db_session.add(advert) api_db_session.commit() observer = EventObserver( event_type="advertisement", event_hash=event_hash, observer_node_id=receiver_node.id, observed_at=datetime.now(timezone.utc), ) api_db_session.add(observer) api_db_session.commit() response = client_no_auth.get(f"/api/v1/advertisements/{advert.id}") assert response.status_code == 200 data = response.json() assert "observers" in data assert len(data["observers"]) == 1 assert data["observers"][0]["public_key"] == receiver_node.public_key def test_get_advertisement_with_tag_names( self, client_no_auth, api_db_session, sample_node_with_name_tag ): """Test that get includes node_tag_name and observer_tag_name.""" advert = Advertisement( public_key=sample_node_with_name_tag.public_key, name="AdName", adv_type="CLIENT", received_at=datetime.now(timezone.utc), node_id=sample_node_with_name_tag.id, ) api_db_session.add(advert) api_db_session.commit() response = client_no_auth.get(f"/api/v1/advertisements/{advert.id}") assert response.status_code == 200 data = response.json() assert data["node_tag_name"] == "Friendly Search Name" class TestListAdvertisementsFilters: """Tests for advertisement list query filters.""" def test_filter_by_search_public_key(self, client_no_auth, sample_advertisement): """Test filtering advertisements by public key search.""" # Partial public key match response = client_no_auth.get("/api/v1/advertisements?search=abc123") assert response.status_code == 200 data = response.json() assert len(data["items"]) == 1 # No match response = client_no_auth.get("/api/v1/advertisements?search=zzz999") assert response.status_code == 200 data = response.json() assert len(data["items"]) == 0 def test_filter_by_search_name(self, client_no_auth, sample_advertisement): """Test filtering advertisements by name search.""" response = client_no_auth.get("/api/v1/advertisements?search=TestNode") assert response.status_code == 200 data = response.json() assert len(data["items"]) == 1 def test_list_advertisements_filter_by_observed_by_single( self, client_no_auth, sample_advertisement, sample_advertisement_with_receiver, receiver_node, ): """Test filtering advertisements by a single receiver node.""" response = client_no_auth.get( f"/api/v1/advertisements?observed_by={receiver_node.public_key}" ) assert response.status_code == 200 data = response.json() assert len(data["items"]) == 1 def test_list_advertisements_filter_by_observed_by_multiple( self, client_no_auth, api_db_session, receiver_node, ): """Test filtering advertisements by multiple receiver nodes.""" # Create second receiver node second_receiver = receiver_node.__class__( public_key="2nd1232nd1232nd1232nd1232nd1232n", name="SecondObserver", first_seen=datetime.now(timezone.utc), ) api_db_session.add(second_receiver) api_db_session.commit() # Create two advertisements, each observed by a different receiver ad1 = Advertisement( public_key="ad1pubad1pubad1pubad1pubad1pubad", name="AD1", adv_type="CLIENT", received_at=datetime.now(timezone.utc), observer_node_id=receiver_node.id, ) ad2 = Advertisement( public_key="ad2pubad2pubad2pubad2pubad2pubad", name="AD2", adv_type="CLIENT", received_at=datetime.now(timezone.utc), observer_node_id=second_receiver.id, ) api_db_session.add_all([ad1, ad2]) api_db_session.commit() # Filter by both receivers response = client_no_auth.get( f"/api/v1/advertisements?observed_by={receiver_node.public_key}&observed_by={second_receiver.public_key}" ) assert response.status_code == 200 data = response.json() assert len(data["items"]) == 2 # Filter by just the first receiver response = client_no_auth.get( f"/api/v1/advertisements?observed_by={receiver_node.public_key}" ) assert response.status_code == 200 data = response.json() assert len(data["items"]) == 1 assert data["items"][0]["name"] == "AD1" def test_filter_by_since(self, client_no_auth, api_db_session): """Test filtering advertisements by since timestamp.""" now = datetime.now(timezone.utc) old_time = now - timedelta(days=7) # Create an old advertisement old_advert = Advertisement( public_key="old123old123old123old123old123ol", name="Old Advertisement", adv_type="CLIENT", received_at=old_time, ) api_db_session.add(old_advert) api_db_session.commit() # Filter since yesterday - should not include old advertisement since = (now - timedelta(days=1)).strftime("%Y-%m-%dT%H:%M:%S") response = client_no_auth.get(f"/api/v1/advertisements?since={since}") assert response.status_code == 200 data = response.json() assert len(data["items"]) == 0 def test_filter_by_until(self, client_no_auth, api_db_session): """Test filtering advertisements by until timestamp.""" now = datetime.now(timezone.utc) old_time = now - timedelta(days=7) # Create an old advertisement old_advert = Advertisement( public_key="until123until123until123until12", name="Old Advertisement Until", adv_type="CLIENT", received_at=old_time, ) api_db_session.add(old_advert) api_db_session.commit() # Filter until 5 days ago - should include old advertisement until = (now - timedelta(days=5)).strftime("%Y-%m-%dT%H:%M:%S") response = client_no_auth.get(f"/api/v1/advertisements?until={until}") assert response.status_code == 200 data = response.json() assert len(data["items"]) == 1 class TestAdvertisementSort: """Tests for advertisement list sort parameters.""" def test_sort_by_time_default(self, client_no_auth, api_db_session): """Default sort is received_at DESC.""" now = datetime.now(timezone.utc) ad_old = Advertisement( public_key="aa" * 16, name="Old", adv_type="CLIENT", received_at=now - timedelta(hours=1), ) ad_new = Advertisement( public_key="bb" * 16, name="New", adv_type="CLIENT", received_at=now, ) api_db_session.add_all([ad_old, ad_new]) api_db_session.commit() response = client_no_auth.get("/api/v1/advertisements") assert response.status_code == 200 items = response.json()["items"] assert items[0]["name"] == "New" assert items[1]["name"] == "Old" def test_sort_by_time_asc(self, client_no_auth, api_db_session): """sort=time&order=asc returns oldest first.""" now = datetime.now(timezone.utc) ad_old = Advertisement( public_key="aa" * 16, name="Old", adv_type="CLIENT", received_at=now - timedelta(hours=1), ) ad_new = Advertisement( public_key="bb" * 16, name="New", adv_type="CLIENT", received_at=now, ) api_db_session.add_all([ad_old, ad_new]) api_db_session.commit() response = client_no_auth.get("/api/v1/advertisements?sort=time&order=asc") assert response.status_code == 200 items = response.json()["items"] assert items[0]["name"] == "Old" assert items[1]["name"] == "New" def test_sort_by_node_name(self, client_no_auth, api_db_session): """sort=node_name sorts by display name (COALESCE).""" from meshcore_hub.common.models import Node now = datetime.now(timezone.utc) node_b = Node( public_key="aa" * 16, name="Bravo", first_seen=now, ) node_a = Node( public_key="bb" * 16, name="Alpha", first_seen=now, ) api_db_session.add_all([node_b, node_a]) api_db_session.commit() ad_b = Advertisement( public_key="aa" * 16, name="AdBravo", adv_type="CLIENT", received_at=now, node_id=node_b.id, ) ad_a = Advertisement( public_key="bb" * 16, name="AdAlpha", adv_type="CLIENT", received_at=now, node_id=node_a.id, ) api_db_session.add_all([ad_b, ad_a]) api_db_session.commit() response = client_no_auth.get("/api/v1/advertisements?sort=node_name&order=asc") assert response.status_code == 200 items = response.json()["items"] assert items[0]["node_name"] == "Alpha" assert items[1]["node_name"] == "Bravo" def test_sort_by_node_name_tag_priority(self, client_no_auth, api_db_session): """Name tag takes priority over SourceNode.name in sort.""" from meshcore_hub.common.models import Node, NodeTag now = datetime.now(timezone.utc) node_b = Node( public_key="aa" * 16, name="Alpha", first_seen=now, ) node_a = Node( public_key="bb" * 16, name="Bravo", first_seen=now, ) api_db_session.add_all([node_b, node_a]) api_db_session.commit() tag_b = NodeTag(node_id=node_b.id, key="name", value="Zebra") tag_a = NodeTag(node_id=node_a.id, key="name", value="Aardvark") api_db_session.add_all([tag_b, tag_a]) api_db_session.commit() ad_b = Advertisement( public_key="aa" * 16, name="AdB", adv_type="CLIENT", received_at=now, node_id=node_b.id, ) ad_a = Advertisement( public_key="bb" * 16, name="AdA", adv_type="CLIENT", received_at=now, node_id=node_a.id, ) api_db_session.add_all([ad_b, ad_a]) api_db_session.commit() response = client_no_auth.get("/api/v1/advertisements?sort=node_name&order=asc") assert response.status_code == 200 items = response.json()["items"] assert items[0]["node_name"] == "Bravo" assert items[1]["node_name"] == "Alpha" def test_sort_by_public_key(self, client_no_auth, api_db_session): """sort=public_key orders by public_key.""" now = datetime.now(timezone.utc) ad_b = Advertisement( public_key="bb" * 16, name="Bravo", adv_type="CLIENT", received_at=now, ) ad_a = Advertisement( public_key="aa" * 16, name="Alpha", adv_type="CLIENT", received_at=now, ) api_db_session.add_all([ad_b, ad_a]) api_db_session.commit() response = client_no_auth.get( "/api/v1/advertisements?sort=public_key&order=asc" ) assert response.status_code == 200 items = response.json()["items"] assert items[0]["public_key"] == "aa" * 16 def test_sort_invalid_ignored(self, client_no_auth, api_db_session): """Invalid sort value falls back to default (time desc).""" now = datetime.now(timezone.utc) ad_old = Advertisement( public_key="aa" * 16, name="Old", adv_type="CLIENT", received_at=now - timedelta(hours=1), ) ad_new = Advertisement( public_key="bb" * 16, name="New", adv_type="CLIENT", received_at=now, ) api_db_session.add_all([ad_old, ad_new]) api_db_session.commit() response = client_no_auth.get("/api/v1/advertisements?sort=invalid_column") assert response.status_code == 200 items = response.json()["items"] assert items[0]["name"] == "New"