mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-22 10:25:23 -04:00
Yeuoly
e4b0d5a181
- Introduced a class method `_load_yaml_file` to encapsulate YAML file loading logic. - Enhanced the `validate_endpoints` method to support both direct endpoint configurations and YAML file paths. - Added unit tests for loading endpoint configurations from YAML strings and files.
54 lines
1.3 KiB
Python
54 lines
1.3 KiB
Python
import os
|
|
|
|
import yaml
|
|
|
|
os.environ["GEVENT_SUPPORT"] = "true"
|
|
|
|
from dify_plugin.entities.endpoint import EndpointProviderConfiguration
|
|
|
|
|
|
def test_load_endpoint_group():
|
|
endpoint_group = EndpointProviderConfiguration(
|
|
**yaml.safe_load(
|
|
"""endpoints:
|
|
- path: /test
|
|
method: GET
|
|
hidden: false
|
|
extra:
|
|
python:
|
|
source: test.py
|
|
"""
|
|
)
|
|
)
|
|
|
|
assert endpoint_group.endpoints[0].path == "/test"
|
|
assert endpoint_group.endpoints[0].method == "GET"
|
|
assert endpoint_group.endpoints[0].hidden is False
|
|
assert endpoint_group.endpoints[0].extra.python.source == "test.py"
|
|
|
|
|
|
def test_load_endpoint_from_file(monkeypatch):
|
|
monkeypatch.setattr(
|
|
EndpointProviderConfiguration,
|
|
"_load_yaml_file",
|
|
lambda x: {
|
|
"path": "/test",
|
|
"method": "GET",
|
|
"hidden": False,
|
|
"extra": {"python": {"source": "test.py"}},
|
|
},
|
|
)
|
|
|
|
endpoint_group = EndpointProviderConfiguration(
|
|
**yaml.safe_load(
|
|
"""endpoints:
|
|
- "endpoints/aaa.yaml"
|
|
"""
|
|
)
|
|
)
|
|
|
|
assert endpoint_group.endpoints[0].path == "/test"
|
|
assert endpoint_group.endpoints[0].method == "GET"
|
|
assert endpoint_group.endpoints[0].hidden is False
|
|
assert endpoint_group.endpoints[0].extra.python.source == "test.py"
|