Spaces:
Paused
Paused
| import time | |
| import uuid | |
| from urllib.parse import urlencode | |
| import pytest | |
| from core.app.entities.app_invoke_entities import InvokeFrom | |
| from core.workflow.entities.variable_pool import VariablePool | |
| from core.workflow.enums import SystemVariableKey | |
| from core.workflow.graph_engine.entities.graph import Graph | |
| from core.workflow.graph_engine.entities.graph_init_params import GraphInitParams | |
| from core.workflow.graph_engine.entities.graph_runtime_state import GraphRuntimeState | |
| from core.workflow.nodes.http_request.node import HttpRequestNode | |
| from models.enums import UserFrom | |
| from models.workflow import WorkflowType | |
| from tests.integration_tests.workflow.nodes.__mock.http import setup_http_mock | |
| def init_http_node(config: dict): | |
| graph_config = { | |
| "edges": [ | |
| { | |
| "id": "start-source-next-target", | |
| "source": "start", | |
| "target": "1", | |
| }, | |
| ], | |
| "nodes": [{"data": {"type": "start"}, "id": "start"}, config], | |
| } | |
| graph = Graph.init(graph_config=graph_config) | |
| init_params = GraphInitParams( | |
| tenant_id="1", | |
| app_id="1", | |
| workflow_type=WorkflowType.WORKFLOW, | |
| workflow_id="1", | |
| graph_config=graph_config, | |
| user_id="1", | |
| user_from=UserFrom.ACCOUNT, | |
| invoke_from=InvokeFrom.DEBUGGER, | |
| call_depth=0, | |
| ) | |
| # construct variable pool | |
| variable_pool = VariablePool( | |
| system_variables={SystemVariableKey.FILES: [], SystemVariableKey.USER_ID: "aaa"}, | |
| user_inputs={}, | |
| environment_variables=[], | |
| conversation_variables=[], | |
| ) | |
| variable_pool.add(["a", "b123", "args1"], 1) | |
| variable_pool.add(["a", "b123", "args2"], 2) | |
| return HttpRequestNode( | |
| id=str(uuid.uuid4()), | |
| graph_init_params=init_params, | |
| graph=graph, | |
| graph_runtime_state=GraphRuntimeState(variable_pool=variable_pool, start_at=time.perf_counter()), | |
| config=config, | |
| ) | |
| def test_get(setup_http_mock): | |
| node = init_http_node( | |
| config={ | |
| "id": "1", | |
| "data": { | |
| "title": "http", | |
| "desc": "", | |
| "method": "get", | |
| "url": "http://example.com", | |
| "authorization": { | |
| "type": "api-key", | |
| "config": { | |
| "type": "basic", | |
| "api_key": "ak-xxx", | |
| "header": "api-key", | |
| }, | |
| }, | |
| "headers": "X-Header:123", | |
| "params": "A:b", | |
| "body": None, | |
| }, | |
| } | |
| ) | |
| result = node._run() | |
| assert result.process_data is not None | |
| data = result.process_data.get("request", "") | |
| assert "?A=b" in data | |
| assert "X-Header: 123" in data | |
| def test_no_auth(setup_http_mock): | |
| node = init_http_node( | |
| config={ | |
| "id": "1", | |
| "data": { | |
| "title": "http", | |
| "desc": "", | |
| "method": "get", | |
| "url": "http://example.com", | |
| "authorization": { | |
| "type": "no-auth", | |
| "config": None, | |
| }, | |
| "headers": "X-Header:123", | |
| "params": "A:b", | |
| "body": None, | |
| }, | |
| } | |
| ) | |
| result = node._run() | |
| assert result.process_data is not None | |
| data = result.process_data.get("request", "") | |
| assert "?A=b" in data | |
| assert "X-Header: 123" in data | |
| def test_custom_authorization_header(setup_http_mock): | |
| node = init_http_node( | |
| config={ | |
| "id": "1", | |
| "data": { | |
| "title": "http", | |
| "desc": "", | |
| "method": "get", | |
| "url": "http://example.com", | |
| "authorization": { | |
| "type": "api-key", | |
| "config": { | |
| "type": "custom", | |
| "api_key": "Auth", | |
| "header": "X-Auth", | |
| }, | |
| }, | |
| "headers": "X-Header:123", | |
| "params": "A:b", | |
| "body": None, | |
| }, | |
| } | |
| ) | |
| result = node._run() | |
| assert result.process_data is not None | |
| data = result.process_data.get("request", "") | |
| assert "?A=b" in data | |
| assert "X-Header: 123" in data | |
| def test_template(setup_http_mock): | |
| node = init_http_node( | |
| config={ | |
| "id": "1", | |
| "data": { | |
| "title": "http", | |
| "desc": "", | |
| "method": "get", | |
| "url": "http://example.com/{{#a.b123.args2#}}", | |
| "authorization": { | |
| "type": "api-key", | |
| "config": { | |
| "type": "basic", | |
| "api_key": "ak-xxx", | |
| "header": "api-key", | |
| }, | |
| }, | |
| "headers": "X-Header:123\nX-Header2:{{#a.b123.args2#}}", | |
| "params": "A:b\nTemplate:{{#a.b123.args2#}}", | |
| "body": None, | |
| }, | |
| } | |
| ) | |
| result = node._run() | |
| assert result.process_data is not None | |
| data = result.process_data.get("request", "") | |
| assert "?A=b" in data | |
| assert "Template=2" in data | |
| assert "X-Header: 123" in data | |
| assert "X-Header2: 2" in data | |
| def test_json(setup_http_mock): | |
| node = init_http_node( | |
| config={ | |
| "id": "1", | |
| "data": { | |
| "title": "http", | |
| "desc": "", | |
| "method": "post", | |
| "url": "http://example.com", | |
| "authorization": { | |
| "type": "api-key", | |
| "config": { | |
| "type": "basic", | |
| "api_key": "ak-xxx", | |
| "header": "api-key", | |
| }, | |
| }, | |
| "headers": "X-Header:123", | |
| "params": "A:b", | |
| "body": { | |
| "type": "json", | |
| "data": [ | |
| { | |
| "key": "", | |
| "type": "text", | |
| "value": '{"a": "{{#a.b123.args1#}}"}', | |
| }, | |
| ], | |
| }, | |
| }, | |
| } | |
| ) | |
| result = node._run() | |
| assert result.process_data is not None | |
| data = result.process_data.get("request", "") | |
| assert '{"a": "1"}' in data | |
| assert "X-Header: 123" in data | |
| def test_x_www_form_urlencoded(setup_http_mock): | |
| node = init_http_node( | |
| config={ | |
| "id": "1", | |
| "data": { | |
| "title": "http", | |
| "desc": "", | |
| "method": "post", | |
| "url": "http://example.com", | |
| "authorization": { | |
| "type": "api-key", | |
| "config": { | |
| "type": "basic", | |
| "api_key": "ak-xxx", | |
| "header": "api-key", | |
| }, | |
| }, | |
| "headers": "X-Header:123", | |
| "params": "A:b", | |
| "body": { | |
| "type": "x-www-form-urlencoded", | |
| "data": [ | |
| { | |
| "key": "a", | |
| "type": "text", | |
| "value": "{{#a.b123.args1#}}", | |
| }, | |
| { | |
| "key": "b", | |
| "type": "text", | |
| "value": "{{#a.b123.args2#}}", | |
| }, | |
| ], | |
| }, | |
| }, | |
| } | |
| ) | |
| result = node._run() | |
| assert result.process_data is not None | |
| data = result.process_data.get("request", "") | |
| assert "a=1&b=2" in data | |
| assert "X-Header: 123" in data | |
| def test_form_data(setup_http_mock): | |
| node = init_http_node( | |
| config={ | |
| "id": "1", | |
| "data": { | |
| "title": "http", | |
| "desc": "", | |
| "method": "post", | |
| "url": "http://example.com", | |
| "authorization": { | |
| "type": "api-key", | |
| "config": { | |
| "type": "basic", | |
| "api_key": "ak-xxx", | |
| "header": "api-key", | |
| }, | |
| }, | |
| "headers": "X-Header:123", | |
| "params": "A:b", | |
| "body": { | |
| "type": "form-data", | |
| "data": [ | |
| { | |
| "key": "a", | |
| "type": "text", | |
| "value": "{{#a.b123.args1#}}", | |
| }, | |
| { | |
| "key": "b", | |
| "type": "text", | |
| "value": "{{#a.b123.args2#}}", | |
| }, | |
| ], | |
| }, | |
| }, | |
| } | |
| ) | |
| result = node._run() | |
| assert result.process_data is not None | |
| data = result.process_data.get("request", "") | |
| assert 'form-data; name="a"' in data | |
| assert "1" in data | |
| assert 'form-data; name="b"' in data | |
| assert "2" in data | |
| assert "X-Header: 123" in data | |
| def test_none_data(setup_http_mock): | |
| node = init_http_node( | |
| config={ | |
| "id": "1", | |
| "data": { | |
| "title": "http", | |
| "desc": "", | |
| "method": "post", | |
| "url": "http://example.com", | |
| "authorization": { | |
| "type": "api-key", | |
| "config": { | |
| "type": "basic", | |
| "api_key": "ak-xxx", | |
| "header": "api-key", | |
| }, | |
| }, | |
| "headers": "X-Header:123", | |
| "params": "A:b", | |
| "body": {"type": "none", "data": []}, | |
| }, | |
| } | |
| ) | |
| result = node._run() | |
| assert result.process_data is not None | |
| data = result.process_data.get("request", "") | |
| assert "X-Header: 123" in data | |
| assert "123123123" not in data | |
| def test_mock_404(setup_http_mock): | |
| node = init_http_node( | |
| config={ | |
| "id": "1", | |
| "data": { | |
| "title": "http", | |
| "desc": "", | |
| "method": "get", | |
| "url": "http://404.com", | |
| "authorization": { | |
| "type": "no-auth", | |
| "config": None, | |
| }, | |
| "body": None, | |
| "params": "", | |
| "headers": "X-Header:123", | |
| }, | |
| } | |
| ) | |
| result = node._run() | |
| assert result.outputs is not None | |
| resp = result.outputs | |
| assert 404 == resp.get("status_code") | |
| assert "Not Found" in resp.get("body", "") | |
| def test_multi_colons_parse(setup_http_mock): | |
| node = init_http_node( | |
| config={ | |
| "id": "1", | |
| "data": { | |
| "title": "http", | |
| "desc": "", | |
| "method": "get", | |
| "url": "http://example.com", | |
| "authorization": { | |
| "type": "no-auth", | |
| "config": None, | |
| }, | |
| "params": "Referer:http://example1.com\nRedirect:http://example2.com", | |
| "headers": "Referer:http://example3.com\nRedirect:http://example4.com", | |
| "body": { | |
| "type": "form-data", | |
| "data": [ | |
| { | |
| "key": "Referer", | |
| "type": "text", | |
| "value": "http://example5.com", | |
| }, | |
| { | |
| "key": "Redirect", | |
| "type": "text", | |
| "value": "http://example6.com", | |
| }, | |
| ], | |
| }, | |
| }, | |
| } | |
| ) | |
| result = node._run() | |
| assert result.process_data is not None | |
| assert result.outputs is not None | |
| resp = result.outputs | |
| assert urlencode({"Redirect": "http://example2.com"}) in result.process_data.get("request", "") | |
| assert 'form-data; name="Redirect"\r\n\r\nhttp://example6.com' in result.process_data.get("request", "") | |
| # assert "http://example3.com" == resp.get("headers", {}).get("referer") | |
| def test_image_file(monkeypatch): | |
| from types import SimpleNamespace | |
| monkeypatch.setattr( | |
| "core.tools.tool_file_manager.ToolFileManager.create_file_by_raw", | |
| lambda *args, **kwargs: SimpleNamespace(id="1"), | |
| ) | |
| node = init_http_node( | |
| config={ | |
| "id": "1", | |
| "data": { | |
| "title": "http", | |
| "desc": "", | |
| "method": "get", | |
| "url": "https://cloud.dify.ai/logo/logo-site.png", | |
| "authorization": { | |
| "type": "no-auth", | |
| "config": None, | |
| }, | |
| "params": "", | |
| "headers": "", | |
| "body": None, | |
| }, | |
| } | |
| ) | |
| result = node._run() | |
| assert result.process_data is not None | |
| assert result.outputs is not None | |
| resp = result.outputs | |
| assert len(resp.get("files", [])) == 1 | |