jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

🔥 chore(test_websocket.py): remove outdated test_chat_history test

Browse source 
The test_chat_history test is outdated and no longer works with the current implementation of the application. It has been removed to avoid confusion and to keep the test suite clean.
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-06-12 09:51:32 -03:00
commit 4485f55d53
1 changed files with 34 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

66
tests/test_websocket.py
View file

@ -10,38 +10,40 @@ def test_websocket_connection(client: TestClient):
assert websocket.scope["path"] == "/api/v1/chat/test_client"
def test_chat_history(client: TestClient):
# Mock the process_graph function to return a specific value
with patch("langflow.chat.manager.process_graph") as mock_process_graph:
mock_process_graph.return_value = ("Hello, I'm a mock response!", "")
# This does not work anymore because now we require
# the flow to be built before sending messages
# def test_chat_history(client: TestClient):
# # Mock the process_graph function to return a specific value
# with patch("langflow.chat.manager.process_graph") as mock_process_graph:
# mock_process_graph.return_value = ("Hello, I'm a mock response!", "")
with client.websocket_connect("api/v1/chat/test_client") as websocket:
# First message should be the history
history = websocket.receive_json()
assert history == [] # Empty history
# Send a message
payload = {"message": "Hello"}
websocket.send_json(json.dumps(payload))
# with client.websocket_connect("api/v1/chat/test_client") as websocket:
# # First message should be the history
# history = websocket.receive_json()
# assert history == [] # Empty history
# # Send a message
# payload = {"message": "Hello"}
# websocket.send_json(json.dumps(payload))
# Receive the response from the server
response = websocket.receive_json()
assert response == {
"is_bot": True,
"message": None,
"type": "start",
"intermediate_steps": "",
"files": [],
}
# Send another message
payload = {"message": "How are you?"}
websocket.send_json(json.dumps(payload))
# # Receive the response from the server
# response = websocket.receive_json()
# assert response == {
# "is_bot": True,
# "message": None,
# "type": "start",
# "intermediate_steps": "",
# "files": [],
# }
# # Send another message
# payload = {"message": "How are you?"}
# websocket.send_json(json.dumps(payload))
# Receive the response from the server
response = websocket.receive_json()
assert response == {
"is_bot": True,
"message": "Hello, I'm a mock response!",
"type": "end",
"intermediate_steps": "",
"files": [],
}
# # Receive the response from the server
# response = websocket.receive_json()
# assert response == {
# "is_bot": True,
# "message": "Hello, I'm a mock response!",
# "type": "end",
# "intermediate_steps": "",
# "files": [],
# }