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

refactor(test_websocket.py): change sender field to is_bot field in chat history tests

Browse source
This commit is contained in:
Gabriel Almeida 2023-04-25 21:06:29 -03:00
commit dc72f2e52f
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/test_websocket.py
View file

@ -28,7 +28,7 @@ def test_chat_history(client: TestClient):
# Receive the response from the server # Receive the response from the server
response = websocket.receive_json() response = websocket.receive_json()
assert json.loads(response) == { assert json.loads(response) == {
"sender": "bot", "is_bot": True,
"message": None, "message": None,
"intermediate_steps": "", "intermediate_steps": "",
"type": "start", "type": "start",
@ -40,7 +40,7 @@ def test_chat_history(client: TestClient):
# Receive the response from the server # Receive the response from the server
response = websocket.receive_json() response = websocket.receive_json()
assert json.loads(response) == { assert json.loads(response) == {
"sender": "bot", "is_bot": True,
"message": "Hello, I'm a mock response!", "message": "Hello, I'm a mock response!",
"intermediate_steps": "", "intermediate_steps": "",
"type": "end", "type": "end",