dify

dify/api/tests/integration_tests/tools/__mock/http.py

@@ -0,0 +1,35 @@
+import json
+from typing import Literal
+
+import httpx
+import pytest
+
+from core.helper import ssrf_proxy
+
+
+class MockedHttp:
+    @staticmethod
+    def httpx_request(
+        method: Literal["GET", "POST", "PUT", "DELETE", "PATCH", "HEAD"], url: str, **kwargs
+    ) -> httpx.Response:
+        """
+        Mocked httpx.request
+        """
+        request = httpx.Request(
+            method, url, params=kwargs.get("params"), headers=kwargs.get("headers"), cookies=kwargs.get("cookies")
+        )
+        data = kwargs.get("data")
+        resp = json.dumps(data).encode("utf-8") if data else b"OK"
+        response = httpx.Response(
+            status_code=200,
+            request=request,
+            content=resp,
+        )
+        return response
+
+
+@pytest.fixture
+def setup_http_mock(request, monkeypatch: pytest.MonkeyPatch):
+    monkeypatch.setattr(ssrf_proxy, "make_request", MockedHttp.httpx_request)
+    yield
+    monkeypatch.undo()