mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-07-08 22:22:17 +02:00
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
"""The registry exposes each verb as one Capability entry the doors/agent read from."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from app.capabilities import (
|
|
web, # noqa: F401 — importing the namespace registers its verbs
|
|
)
|
|
from app.capabilities.core.store import get_capability
|
|
from app.capabilities.core.types import BillingUnit
|
|
from app.capabilities.web.discover.schemas import DiscoverInput, DiscoverOutput
|
|
from app.capabilities.web.scrape.schemas import ScrapeInput, ScrapeOutput
|
|
|
|
pytestmark = pytest.mark.unit
|
|
|
|
|
|
def test_web_scrape_is_registered_with_its_schemas_and_billing_unit():
|
|
cap = get_capability("web.scrape")
|
|
|
|
assert cap.name == "web.scrape"
|
|
assert cap.input_schema is ScrapeInput
|
|
assert cap.output_schema is ScrapeOutput
|
|
assert cap.billing_unit is BillingUnit.WEB_CRAWL
|
|
|
|
|
|
def test_web_discover_is_registered_and_free():
|
|
cap = get_capability("web.discover")
|
|
|
|
assert cap.name == "web.discover"
|
|
assert cap.input_schema is DiscoverInput
|
|
assert cap.output_schema is DiscoverOutput
|
|
assert cap.billing_unit is None
|