mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-12-18 02:23:06 +08:00
27 lines
1010 B
Python
27 lines
1010 B
Python
import aiohttp
|
|
import pytest
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_tags_listing_endpoint(http: aiohttp.ClientSession, api_base: str):
|
|
# Include zero-usage tags by default
|
|
async with http.get(api_base + "/api/tags", params={"limit": "50"}) as r1:
|
|
body1 = await r1.json()
|
|
assert r1.status == 200
|
|
names = [t["name"] for t in body1["tags"]]
|
|
# A few system tags from migration should exist:
|
|
assert "models" in names
|
|
assert "checkpoints" in names
|
|
|
|
# Only used tags
|
|
async with http.get(api_base + "/api/tags", params={"include_zero": "false"}) as r2:
|
|
body2 = await r2.json()
|
|
assert r2.status == 200
|
|
# Should contain no tags
|
|
assert not [t["name"] for t in body2["tags"]]
|
|
|
|
# TODO-1: add some asset
|
|
# TODO-2: check that "used" tags are now non zero amount
|
|
|
|
# TODO-3: do a global teardown, so the state of ComfyUI is clear after each test, and all test can be run solo or one-by-one without any problems.
|