14 lines
598 B
Python
14 lines
598 B
Python
from src.database import DatabaseManager
|
|
|
|
def test_insert_chat_data(mocker):
|
|
db_manager = DatabaseManager()
|
|
mocker.patch.object(db_manager, 'insert_chat_data')
|
|
db_manager.insert_chat_data("chatroom1", [{"message": "test"}])
|
|
db_manager.insert_chat_data.assert_called_once()
|
|
|
|
def test_get_messages_by_user(mocker):
|
|
db_manager = DatabaseManager()
|
|
mocker.patch.object(db_manager.db.chatrooms, 'find')
|
|
db_manager.get_messages_by_user("chatroom1", "User1")
|
|
db_manager.db.chatrooms.find.assert_called_with({"_id": "chatroom1", "messages.user": "User1"}, {"messages.$": 1})
|