apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-07-24 23:41:10 +02:00
Code Issues Projects Releases Packages Wiki Activity

test: assert youtube billing units

Browse source
This commit is contained in:
CREDO23 2026-07-14 21:50:52 +02:00
parent ffc0675ac2
commit 0089bb9001
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
surfsense_backend/tests/unit/capabilities/youtube/test_registry.py
View file

@ -8,25 +8,26 @@ from app.capabilities import (
youtube, # noqa: F401 — importing the namespace registers its verbs youtube, # noqa: F401 — importing the namespace registers its verbs
) )
from app.capabilities.core.store import get_capability from app.capabilities.core.store import get_capability
from app.capabilities.core.types import BillingUnit
from app.capabilities.youtube.comments.schemas import CommentsInput, CommentsOutput from app.capabilities.youtube.comments.schemas import CommentsInput, CommentsOutput
from app.capabilities.youtube.scrape.schemas import ScrapeInput, ScrapeOutput from app.capabilities.youtube.scrape.schemas import ScrapeInput, ScrapeOutput
pytestmark = pytest.mark.unit pytestmark = pytest.mark.unit
def test_youtube_scrape_is_registered_and_free(): def test_youtube_scrape_is_registered_and_billed_per_video():
cap = get_capability("youtube.scrape") cap = get_capability("youtube.scrape")
assert cap.name == "youtube.scrape" assert cap.name == "youtube.scrape"
assert cap.input_schema is ScrapeInput assert cap.input_schema is ScrapeInput
assert cap.output_schema is ScrapeOutput 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_billed_per_comment():
cap = get_capability("youtube.comments") cap = get_capability("youtube.comments")
assert cap.name == "youtube.comments" assert cap.name == "youtube.comments"
assert cap.input_schema is CommentsInput assert cap.input_schema is CommentsInput
assert cap.output_schema is CommentsOutput assert cap.output_schema is CommentsOutput
assert cap.billing_unit is None assert cap.billing_unit is BillingUnit.YOUTUBE_COMMENT