mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-14 00:00:57 +08:00
50 lines
1.8 KiB
Python
50 lines
1.8 KiB
Python
import pytest
|
|
from aiohttp import web
|
|
|
|
from comfy.app.custom_node_manager import CustomNodeManager
|
|
from comfy.component_model.folder_path_types import FolderNames, ModelPaths
|
|
from comfy.execution_context import context_folder_names_and_paths
|
|
|
|
pytestmark = (
|
|
pytest.mark.asyncio
|
|
) # This applies the asyncio mark to all test functions in the module
|
|
|
|
fn = FolderNames()
|
|
|
|
|
|
@pytest.fixture
|
|
def custom_node_manager():
|
|
with context_folder_names_and_paths(fn):
|
|
return CustomNodeManager()
|
|
|
|
|
|
@pytest.fixture
|
|
def app(custom_node_manager):
|
|
with context_folder_names_and_paths(fn):
|
|
app = web.Application()
|
|
routes = web.RouteTableDef()
|
|
custom_node_manager.add_routes(routes, app, [("ComfyUI-TestExtension1", "ComfyUI-TestExtension1")])
|
|
app.add_routes(routes)
|
|
return app
|
|
|
|
|
|
async def test_get_workflow_templates(aiohttp_client, app, tmp_path):
|
|
client = await aiohttp_client(app)
|
|
# Setup temporary custom nodes file structure with 1 workflow file
|
|
custom_nodes_dir = tmp_path / "custom_nodes"
|
|
example_workflows_dir = custom_nodes_dir / "ComfyUI-TestExtension1" / "example_workflows"
|
|
example_workflows_dir.mkdir(parents=True)
|
|
template_file = example_workflows_dir / "workflow1.json"
|
|
template_file.write_text('')
|
|
|
|
fn.add(ModelPaths(folder_names=["custom_nodes"], additional_absolute_directory_paths=[custom_nodes_dir]))
|
|
|
|
with context_folder_names_and_paths(fn):
|
|
response = await client.get('/workflow_templates')
|
|
assert response.status == 200
|
|
workflows_dict = await response.json()
|
|
assert isinstance(workflows_dict, dict)
|
|
assert "ComfyUI-TestExtension1" in workflows_dict
|
|
assert isinstance(workflows_dict["ComfyUI-TestExtension1"], list)
|
|
assert workflows_dict["ComfyUI-TestExtension1"][0] == "workflow1"
|