From 7cb1dbf226b3f241bab46133fff5c1782143e26b Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Mon, 4 Sep 2023 09:56:27 -0300
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(test=5Fendpoints.py):=20chan?=
 =?UTF-8?q?ge=20header=20key=20from=20'api-key'=20to=20'x-api-key'=20to=20?=
 =?UTF-8?q?align=20with=20API=20key=20header=20convention=20=E2=9C=A8=20fe?=
 =?UTF-8?q?at(test=5Fendpoints.py):=20change=20mock=5Fprocess=5Fgraph=5Fca?=
 =?UTF-8?q?ched=20function=20to=20be=20async=20and=20return=20a=20Result?=
 =?UTF-8?q?=20object=20instead=20of=20a=20tuple=20to=20improve=20code=20re?=
 =?UTF-8?q?adability=20and=20maintainability?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tests/test_endpoints.py | 29 +++++++++++++++--------------
 1 file changed, 15 insertions(+), 14 deletions(-)

diff --git a/tests/test_endpoints.py b/tests/test_endpoints.py
index cbb1eb08c..e48393bed 100644
--- a/tests/test_endpoints.py
+++ b/tests/test_endpoints.py
@@ -1,4 +1,5 @@
 import uuid
+from langflow.processing.process import Result
 from langflow.services.auth.utils import get_password_hash
 from langflow.services.database.models.api_key.api_key import ApiKey
 from langflow.services.utils import get_settings_manager
@@ -105,8 +106,8 @@ def created_api_key(session, active_user):
 
 def test_process_flow_invalid_api_key(client, flow, monkeypatch):
     # Mock de process_graph_cached
-    def mock_process_graph_cached(*args, **kwargs):
-        return {}, "session_id_mock"
+    async def mock_process_graph_cached(*args, **kwargs):
+        return Result(result={}, session_id="session_id_mock")
 
     settings_manager = get_settings_manager()
     settings_manager.auth_settings.AUTO_LOGIN = False
@@ -114,7 +115,7 @@ def test_process_flow_invalid_api_key(client, flow, monkeypatch):
 
     monkeypatch.setattr(endpoints, "process_graph_cached", mock_process_graph_cached)
 
-    headers = {"api-key": "invalid_api_key"}
+    headers = {"x-api-key": "invalid_api_key"}
 
     post_data = {
         "inputs": {"key": "value"},
@@ -130,15 +131,15 @@ def test_process_flow_invalid_api_key(client, flow, monkeypatch):
 
 
 def test_process_flow_invalid_id(client, monkeypatch, created_api_key):
-    def mock_process_graph_cached(*args, **kwargs):
-        return {}, "session_id_mock"
+    async def mock_process_graph_cached(*args, **kwargs):
+        return Result(result={}, session_id="session_id_mock")
 
     from langflow.api.v1 import endpoints
 
     monkeypatch.setattr(endpoints, "process_graph_cached", mock_process_graph_cached)
 
     api_key = created_api_key.api_key
-    headers = {"api-key": api_key}
+    headers = {"x-api-key": api_key}
 
     post_data = {
         "inputs": {"key": "value"},
@@ -163,13 +164,13 @@ def test_process_flow_without_autologin(client, flow, monkeypatch, created_api_k
     settings_manager = get_settings_manager()
     settings_manager.auth_settings.AUTO_LOGIN = False
 
-    def mock_process_graph_cached(*args, **kwargs):
-        return {}, "session_id_mock"
+    async def mock_process_graph_cached(*args, **kwargs):
+        return Result(result={}, session_id="session_id_mock")
 
     monkeypatch.setattr(endpoints, "process_graph_cached", mock_process_graph_cached)
 
     api_key = created_api_key.api_key
-    headers = {"api-key": api_key}
+    headers = {"x-api-key": api_key}
 
     # Dummy POST data
     post_data = {
@@ -185,8 +186,8 @@ def test_process_flow_without_autologin(client, flow, monkeypatch, created_api_k
 
     # Check the response
     assert response.status_code == 200, response.json()
-    assert response.json()["result"] == {}
-    assert response.json()["session_id"] == "session_id_mock"
+    assert response.json()["result"] == {}, response.json()
+    assert response.json()["session_id"] == "session_id_mock", response.json()
 
 
 def test_process_flow_fails_autologin_off(client, flow, monkeypatch):
@@ -196,12 +197,12 @@ def test_process_flow_fails_autologin_off(client, flow, monkeypatch):
     settings_manager = get_settings_manager()
     settings_manager.auth_settings.AUTO_LOGIN = False
 
-    def mock_process_graph_cached(*args, **kwargs):
-        return {}, "session_id_mock"
+    async def mock_process_graph_cached(*args, **kwargs):
+        return Result(result={}, session_id="session_id_mock")
 
     monkeypatch.setattr(endpoints, "process_graph_cached", mock_process_graph_cached)
 
-    headers = {"api-key": "api_key"}
+    headers = {"x-api-key": "api_key"}
 
     # Dummy POST data
     post_data = {