ForcePilot/backend/package/yuxi/channel/sdk/test_utils.py

27 lines
783 B
Python
Raw Normal View History

from typing import Any
from yuxi.channel.sdk.dependencies import PluginResourceProvider, inject_resources
def create_test_resources(
http_client: Any | None = None,
db_session_factory: Any | None = None,
redis_client: Any | None = None,
config_provider: Any | None = None,
metadata: dict | None = None,
**kwargs,
) -> PluginResourceProvider:
provider = PluginResourceProvider(
http_client=http_client,
db_session_factory=db_session_factory,
redis_client=redis_client,
config_provider=config_provider,
metadata=metadata or {},
)
for key, value in kwargs.items():
setattr(provider, key, value)
return provider
def reset_test_resources() -> None:
inject_resources(PluginResourceProvider())