From 505bc64c0bc7e5757391d59384b5b5952c2aea60 Mon Sep 17 00:00:00 2001
From: Anish Sarkar <104695310+AnishSarkar22@users.noreply.github.com>
Date: Wed, 15 Jul 2026 15:58:07 +0530
Subject: [PATCH] test: update Google Maps, Reddit, and YouTube capabilities to
 reflect billing units

---
 .../tests/unit/capabilities/google_maps/test_registry.py | 9 +++++----
 .../tests/unit/capabilities/reddit/test_registry.py      | 5 +++--
 .../tests/unit/capabilities/youtube/test_registry.py     | 9 +++++----
 3 files changed, 13 insertions(+), 10 deletions(-)

diff --git a/surfsense_backend/tests/unit/capabilities/google_maps/test_registry.py b/surfsense_backend/tests/unit/capabilities/google_maps/test_registry.py
index c6a007645..409d0f78a 100644
--- a/surfsense_backend/tests/unit/capabilities/google_maps/test_registry.py
+++ b/surfsense_backend/tests/unit/capabilities/google_maps/test_registry.py
@@ -7,6 +7,7 @@ import pytest
 from app.capabilities import (
     google_maps,  # noqa: F401  — importing the namespace registers its verbs
 )
+from app.capabilities.core import BillingUnit
 from app.capabilities.core.store import get_capability
 from app.capabilities.google_maps.reviews.schemas import ReviewsInput, ReviewsOutput
 from app.capabilities.google_maps.scrape.schemas import ScrapeInput, ScrapeOutput
@@ -14,19 +15,19 @@ from app.capabilities.google_maps.scrape.schemas import ScrapeInput, ScrapeOutpu
 pytestmark = pytest.mark.unit
 
 
-def test_google_maps_scrape_is_registered_and_free():
+def test_google_maps_scrape_is_registered_and_billable():
     cap = get_capability("google_maps.scrape")
 
     assert cap.name == "google_maps.scrape"
     assert cap.input_schema is ScrapeInput
     assert cap.output_schema is ScrapeOutput
-    assert cap.billing_unit is None
+    assert cap.billing_unit is BillingUnit.GOOGLE_MAPS_PLACE
 
 
-def test_google_maps_reviews_is_registered_and_free():
+def test_google_maps_reviews_is_registered_and_billable():
     cap = get_capability("google_maps.reviews")
 
     assert cap.name == "google_maps.reviews"
     assert cap.input_schema is ReviewsInput
     assert cap.output_schema is ReviewsOutput
-    assert cap.billing_unit is None
+    assert cap.billing_unit is BillingUnit.GOOGLE_MAPS_REVIEW
diff --git a/surfsense_backend/tests/unit/capabilities/reddit/test_registry.py b/surfsense_backend/tests/unit/capabilities/reddit/test_registry.py
index e0bca5f26..a6452470a 100644
--- a/surfsense_backend/tests/unit/capabilities/reddit/test_registry.py
+++ b/surfsense_backend/tests/unit/capabilities/reddit/test_registry.py
@@ -7,16 +7,17 @@ import pytest
 from app.capabilities import (
     reddit,  # noqa: F401  — importing the namespace registers its verbs
 )
+from app.capabilities.core import BillingUnit
 from app.capabilities.core.store import get_capability
 from app.capabilities.reddit.scrape.schemas import ScrapeInput, ScrapeOutput
 
 pytestmark = pytest.mark.unit
 
 
-def test_reddit_scrape_is_registered_and_free():
+def test_reddit_scrape_is_registered_and_billable():
     cap = get_capability("reddit.scrape")
 
     assert cap.name == "reddit.scrape"
     assert cap.input_schema is ScrapeInput
     assert cap.output_schema is ScrapeOutput
-    assert cap.billing_unit is None
+    assert cap.billing_unit is BillingUnit.REDDIT_ITEM
diff --git a/surfsense_backend/tests/unit/capabilities/youtube/test_registry.py b/surfsense_backend/tests/unit/capabilities/youtube/test_registry.py
index 756b65176..7fc98c697 100644
--- a/surfsense_backend/tests/unit/capabilities/youtube/test_registry.py
+++ b/surfsense_backend/tests/unit/capabilities/youtube/test_registry.py
@@ -7,6 +7,7 @@ import pytest
 from app.capabilities import (
     youtube,  # noqa: F401  — importing the namespace registers its verbs
 )
+from app.capabilities.core import BillingUnit
 from app.capabilities.core.store import get_capability
 from app.capabilities.youtube.comments.schemas import CommentsInput, CommentsOutput
 from app.capabilities.youtube.scrape.schemas import ScrapeInput, ScrapeOutput
@@ -14,19 +15,19 @@ from app.capabilities.youtube.scrape.schemas import ScrapeInput, ScrapeOutput
 pytestmark = pytest.mark.unit
 
 
-def test_youtube_scrape_is_registered_and_free():
+def test_youtube_scrape_is_registered_and_billable():
     cap = get_capability("youtube.scrape")
 
     assert cap.name == "youtube.scrape"
     assert cap.input_schema is ScrapeInput
     assert cap.output_schema is ScrapeOutput
-    assert cap.billing_unit is None
+    assert cap.billing_unit is BillingUnit.YOUTUBE_VIDEO
 
 
-def test_youtube_comments_is_registered_and_free():
+def test_youtube_comments_is_registered_and_billable():
     cap = get_capability("youtube.comments")
 
     assert cap.name == "youtube.comments"
     assert cap.input_schema is CommentsInput
     assert cap.output_schema is CommentsOutput
-    assert cap.billing_unit is None
+    assert cap.billing_unit is BillingUnit.YOUTUBE_COMMENT