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

chore(api/tests): apply ruff reformat #7590 (#7591)

Browse Source 
Co-authored-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
Bowen Liang
2024-08-23 23:52:25 +08:00
committed by GitHub
parent 2da63654e5
commit b035c02f78
155 changed files with 4279 additions and 5925 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
api/tests/integration_tests/workflow/nodes/__mock/code_executor.py
View File

@@ -7,25 +7,22 @@ from jinja2 import Template
from core.helper.code_executor.code_executor import CodeExecutor, CodeLanguage
MOCK = os.getenv('MOCK_SWITCH', 'false') == 'true'
MOCK = os.getenv("MOCK_SWITCH", "false") == "true"
class MockedCodeExecutor:
@classmethod
def invoke(cls, language: Literal['python3', 'javascript', 'jinja2'],
code: str, inputs: dict) -> dict:
def invoke(cls, language: Literal["python3", "javascript", "jinja2"], code: str, inputs: dict) -> dict:
# invoke directly
match language:
case CodeLanguage.PYTHON3:
return {
"result": 3
}
return {"result": 3}
case CodeLanguage.JINJA2:
return {
"result": Template(code).render(inputs)
}
return {"result": Template(code).render(inputs)}
case _:
raise Exception("Language not supported")
@pytest.fixture
def setup_code_executor_mock(request, monkeypatch: MonkeyPatch):
if not MOCK:

30
api/tests/integration_tests/workflow/nodes/__mock/http.py
View File

@@ -6,38 +6,32 @@ import httpx
import pytest
from _pytest.monkeypatch import MonkeyPatch
MOCK = os.getenv('MOCK_SWITCH', 'false') == 'true'
MOCK = os.getenv("MOCK_SWITCH", "false") == "true"
class MockedHttp:
def httpx_request(method: Literal['GET', 'POST', 'PUT', 'DELETE', 'PATCH', 'HEAD'],
url: str, **kwargs) -> httpx.Response:
def httpx_request(
method: Literal["GET", "POST", "PUT", "DELETE", "PATCH", "HEAD"], url: str, **kwargs
) -> httpx.Response:
"""
Mocked httpx.request
"""
if url == 'http://404.com':
response = httpx.Response(
status_code=404,
request=httpx.Request(method, url),
content=b'Not Found'
)
if url == "http://404.com":
response = httpx.Response(status_code=404, request=httpx.Request(method, url), content=b"Not Found")
return response
# get data, files
data = kwargs.get('data', None)
files = kwargs.get('files', None)
data = kwargs.get("data", None)
files = kwargs.get("files", None)
if data is not None:
resp = dumps(data).encode('utf-8')
resp = dumps(data).encode("utf-8")
elif files is not None:
resp = dumps(files).encode('utf-8')
resp = dumps(files).encode("utf-8")
else:
resp = b'OK'
resp = b"OK"
response = httpx.Response(
status_code=200,
request=httpx.Request(method, url),
headers=kwargs.get('headers', {}),
content=resp
status_code=200, request=httpx.Request(method, url), headers=kwargs.get("headers", {}), content=resp
)
return response