admin/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

refactor: split changes for api/tests/test_containers_integration_tes… (#29897)

Browse Source
This commit is contained in:
Asuka Minato
2025-12-19 18:50:58 +09:00
committed by GitHub
parent 39ad9d1569
commit 054d3f0da5
1 changed files with 9 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
api/tests/test_containers_integration_tests/services/tools/test_api_tools_manage_service.py
View File

@@ -2,7 +2,9 @@ from unittest.mock import patch
import pytest import pytest
from faker import Faker from faker import Faker
from pydantic import TypeAdapter, ValidationError
from core.tools.entities.tool_entities import ApiProviderSchemaType
from models import Account, Tenant from models import Account, Tenant
from models.tools import ApiToolProvider from models.tools import ApiToolProvider
from services.tools.api_tools_manage_service import ApiToolManageService from services.tools.api_tools_manage_service import ApiToolManageService
@@ -298,7 +300,7 @@ class TestApiToolManageService:
provider_name = fake.company() provider_name = fake.company()
icon = {"type": "emoji", "value": "🔧"} icon = {"type": "emoji", "value": "🔧"}
credentials = {"auth_type": "none", "api_key_header": "X-API-Key", "api_key_value": ""} credentials = {"auth_type": "none", "api_key_header": "X-API-Key", "api_key_value": ""}
schema_type = "openapi" schema_type = ApiProviderSchemaType.OPENAPI
schema = self._create_test_openapi_schema() schema = self._create_test_openapi_schema()
privacy_policy = "https://example.com/privacy" privacy_policy = "https://example.com/privacy"
custom_disclaimer = "Custom disclaimer text" custom_disclaimer = "Custom disclaimer text"
@@ -364,7 +366,7 @@ class TestApiToolManageService:
provider_name = fake.company() provider_name = fake.company()
icon = {"type": "emoji", "value": "🔧"} icon = {"type": "emoji", "value": "🔧"}
credentials = {"auth_type": "none"} credentials = {"auth_type": "none"}
schema_type = "openapi" schema_type = ApiProviderSchemaType.OPENAPI
schema = self._create_test_openapi_schema() schema = self._create_test_openapi_schema()
privacy_policy = "https://example.com/privacy" privacy_policy = "https://example.com/privacy"
custom_disclaimer = "Custom disclaimer text" custom_disclaimer = "Custom disclaimer text"
@@ -428,21 +430,10 @@ class TestApiToolManageService:
labels = ["test"] labels = ["test"]
# Act & Assert: Try to create provider with invalid schema type # Act & Assert: Try to create provider with invalid schema type
with pytest.raises(ValueError) as exc_info: with pytest.raises(ValidationError) as exc_info:
ApiToolManageService.create_api_tool_provider( TypeAdapter(ApiProviderSchemaType).validate_python(schema_type)
user_id=account.id,
tenant_id=tenant.id,
provider_name=provider_name,
icon=icon,
credentials=credentials,
schema_type=schema_type,
schema=schema,
privacy_policy=privacy_policy,
custom_disclaimer=custom_disclaimer,
labels=labels,
)
assert "invalid schema type" in str(exc_info.value) assert "validation error" in str(exc_info.value)
def test_create_api_tool_provider_missing_auth_type( def test_create_api_tool_provider_missing_auth_type(
self, flask_req_ctx_with_containers, db_session_with_containers, mock_external_service_dependencies self, flask_req_ctx_with_containers, db_session_with_containers, mock_external_service_dependencies
@@ -464,7 +455,7 @@ class TestApiToolManageService:
provider_name = fake.company() provider_name = fake.company()
icon = {"type": "emoji", "value": "🔧"} icon = {"type": "emoji", "value": "🔧"}
credentials = {} # Missing auth_type credentials = {} # Missing auth_type
schema_type = "openapi" schema_type = ApiProviderSchemaType.OPENAPI
schema = self._create_test_openapi_schema() schema = self._create_test_openapi_schema()
privacy_policy = "https://example.com/privacy" privacy_policy = "https://example.com/privacy"
custom_disclaimer = "Custom disclaimer text" custom_disclaimer = "Custom disclaimer text"
@@ -507,7 +498,7 @@ class TestApiToolManageService:
provider_name = fake.company() provider_name = fake.company()
icon = {"type": "emoji", "value": "🔑"} icon = {"type": "emoji", "value": "🔑"}
credentials = {"auth_type": "api_key", "api_key_header": "X-API-Key", "api_key_value": fake.uuid4()} credentials = {"auth_type": "api_key", "api_key_header": "X-API-Key", "api_key_value": fake.uuid4()}
schema_type = "openapi" schema_type = ApiProviderSchemaType.OPENAPI
schema = self._create_test_openapi_schema() schema = self._create_test_openapi_schema()
privacy_policy = "https://example.com/privacy" privacy_policy = "https://example.com/privacy"
custom_disclaimer = "Custom disclaimer text" custom_disclaimer = "Custom disclaimer text"