mirror of
https://github.com/langgenius/dify.git
synced 2026-06-23 04:11:09 +08:00
45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
from inspect import unwrap
|
|
from unittest.mock import patch
|
|
|
|
import controllers.console.spec as spec_module
|
|
|
|
|
|
class TestSpecSchemaDefinitionsApi:
|
|
def test_get_success(self):
|
|
api = spec_module.SpecSchemaDefinitionsApi()
|
|
method = unwrap(api.get)
|
|
|
|
schema_definitions = [{"type": "string"}]
|
|
|
|
with patch.object(
|
|
spec_module,
|
|
"SchemaManager",
|
|
) as schema_manager_cls:
|
|
schema_manager_cls.return_value.get_all_schema_definitions.return_value = schema_definitions
|
|
|
|
resp, status = method(api)
|
|
|
|
assert status == 200
|
|
assert resp == schema_definitions
|
|
|
|
def test_get_exception_returns_empty_list(self):
|
|
api = spec_module.SpecSchemaDefinitionsApi()
|
|
method = unwrap(api.get)
|
|
|
|
with (
|
|
patch.object(
|
|
spec_module,
|
|
"SchemaManager",
|
|
side_effect=Exception("boom"),
|
|
),
|
|
patch.object(
|
|
spec_module.logger,
|
|
"exception",
|
|
) as log_exception,
|
|
):
|
|
resp, status = method(api)
|
|
|
|
assert status == 200
|
|
assert resp == []
|
|
log_exception.assert_called_once()
|