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

This commit is contained in:
Gabriel Almeida 2023-04-25 21:06:29 -03:00
commit dc72f2e52f

View file

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