Function track_api_usage

tests/conftest.py:420–470

Track API usage.

(snapshots: SnapshotFixture)

@pytest.fixture(autouse=True)
def track_api_usage(snapshots: SnapshotFixture):
    """Track API usage."""
    yield
    if (request := REQUEST_CONTEXT.get()) is None:
        return
    response_mocker = ResponseMocker()
    calls = {}

    for call in response_mocker.calls:
        if (_test_caller := call.pop("_test_caller", None)) is None:
            continue
        if _test_caller not in calls:
            if call.get("_uses_setup_integration"):
                calls[_test_caller] = {
                    c: -1
                    for c in [
                        "https://data-v2.hacs.xyz/appdaemon/data.json",
                        "https://data-v2.hacs.xyz/critical/data.json",
                        "https://data-v2.hacs.xyz/integration/data.json",
                        "https://data-v2.hacs.xyz/plugin/data.json",
                        "https://data-v2.hacs.xyz/python_script/data.json",
                        "https://data-v2.hacs.xyz/removed/data.json",
                        "https://data-v2.hacs.xyz/template/data.json",
                        "https://data-v2.hacs.xyz/theme/data.json",
                    ]
                }
            else:
                calls[_test_caller] = {}
        if (url := call.get("url")) not in calls[_test_caller]:
            calls[_test_caller][url] = 0
        calls[_test_caller][url] += 1

    filtered_calls = {
        k: v
        for k, v in {t: {k: v for k, v in c.items() if v != 0} for t, c in calls.items()}.items()
        if v
    }

    snapshotfile = f"api-usage/{request.node.location[0].replace(".py", "")}{
        slugify(f"::{request.node.name}")}.json"

    if not filtered_calls:
        assert not os.path.exists(f"tests/snapshots/{snapshotfile}")
        return

    response_mocker.calls = []

    snapshots.assert_match(
        safe_json_dumps(filtered_calls),
        snapshotfile
    )