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

Refactor test_get_messages function to use list comprehension

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-26 16:12:22 -03:00
commit 802425c7de
1 changed files with 9 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/unit/test_messages.py
View file

@ -35,16 +35,20 @@ def created_messages(session):
return messages_read return messages_read
def test_get_messages(session): def test_get_messages():
add_messages(Message(text="Test message 1", sender="User", sender_name="User", session_id="session_id2")) add_messages(
add_messages(Message(text="Test message 2", sender="User", sender_name="User", session_id="session_id2")) [
Message(text="Test message 1", sender="User", sender_name="User", session_id="session_id2"),
Message(text="Test message 2", sender="User", sender_name="User", session_id="session_id2"),
]
)
messages = get_messages(sender="User", session_id="session_id2", limit=2) messages = get_messages(sender="User", session_id="session_id2", limit=2)
assert len(messages) == 2 assert len(messages) == 2
assert messages[0].text == "Test message 1" assert messages[0].text == "Test message 1"
assert messages[1].text == "Test message 2" assert messages[1].text == "Test message 2"
def test_add_messages(session): def test_add_messages():
message = Message(text="New Test message", sender="User", sender_name="User", session_id="new_session_id") message = Message(text="New Test message", sender="User", sender_name="User", session_id="new_session_id")
messages = add_messages(message) messages = add_messages(message)
assert len(messages) == 1 assert len(messages) == 1
@ -65,7 +69,7 @@ def test_delete_messages(session):
assert len(messages) == 0 assert len(messages) == 0
def test_store_message(session): def test_store_message():
message = Message(text="Stored message", sender="User", sender_name="User", session_id="stored_session_id") message = Message(text="Stored message", sender="User", sender_name="User", session_id="stored_session_id")
stored_messages = store_message(message) stored_messages = store_message(message)
assert len(stored_messages) == 1 assert len(stored_messages) == 1