67 lines
1.9 KiB
Python
67 lines
1.9 KiB
Python
|
import os
|
||
|
|
from typing import Literal
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
import requests
|
||
|
|
from _pytest.monkeypatch import MonkeyPatch
|
||
|
|
|
||
|
|
from core.plugin.entities.plugin_daemon import PluginDaemonBasicResponse
|
||
|
|
from core.tools.entities.common_entities import I18nObject
|
||
|
|
from core.tools.entities.tool_entities import ToolProviderEntity, ToolProviderIdentity
|
||
|
|
|
||
|
|
|
||
|
|
class MockedHttp:
|
||
|
|
@classmethod
|
||
|
|
def list_tools(cls) -> list[ToolProviderEntity]:
|
||
|
|
return [
|
||
|
|
ToolProviderEntity(
|
||
|
|
identity=ToolProviderIdentity(
|
||
|
|
author="Yeuoly",
|
||
|
|
name="Yeuoly",
|
||
|
|
description=I18nObject(en_US="Yeuoly"),
|
||
|
|
icon="ssss.svg",
|
||
|
|
label=I18nObject(en_US="Yeuoly"),
|
||
|
|
)
|
||
|
|
)
|
||
|
|
]
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def requests_request(
|
||
|
|
cls, method: Literal["GET", "POST", "PUT", "DELETE", "PATCH", "HEAD"], url: str, **kwargs
|
||
|
|
) -> requests.Response:
|
||
|
|
"""
|
||
|
|
Mocked requests.request
|
||
|
|
"""
|
||
|
|
request = requests.PreparedRequest()
|
||
|
|
request.method = method
|
||
|
|
request.url = url
|
||
|
|
if url.endswith("/tools"):
|
||
|
|
content = PluginDaemonBasicResponse[list[ToolProviderEntity]](
|
||
|
|
code=0, message="success", data=cls.list_tools()
|
||
|
|
).model_dump_json()
|
||
|
|
else:
|
||
|
|
raise ValueError("")
|
||
|
|
|
||
|
|
response = requests.Response()
|
||
|
|
response.status_code = 200
|
||
|
|
response.request = request
|
||
|
|
response._content = content.encode("utf-8")
|
||
|
|
return response
|
||
|
|
|
||
|
|
|
||
|
|
MOCK_SWITCH = os.getenv("MOCK_SWITCH", "false").lower() == "true"
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def setup_http_mock(request, monkeypatch: MonkeyPatch):
|
||
|
|
if MOCK_SWITCH:
|
||
|
|
monkeypatch.setattr(requests, "request", MockedHttp.requests_request)
|
||
|
|
|
||
|
|
def unpatch():
|
||
|
|
monkeypatch.undo()
|
||
|
|
|
||
|
|
yield
|
||
|
|
|
||
|
|
if MOCK_SWITCH:
|
||
|
|
unpatch()
|