def test_get_id_and_username_all_users(self):
user1 = factories.create_user('vasia')
user2 = factories.create_user('petiya')
initial_array = [(user1, 'vasia'), (user2, 'petiya')]
fetched_array = user_db.get_id_and_username_all_users()
self.assertEqual(initial_array, fetched_array)
self.assertEqual(len(fetched_array), 2)
def test_success_create_message(self):
user_id = factories.create_user()
recipient_id = factories.create_user()
text_message = 'Hello world'
chat_db.create_message(user_id, recipient_id, text_message)
fetched_query = chat_db.get_messages(user_id, recipient_id)
self.assertEqual(text_message, fetched_query[0].text)
self.assertEqual(user_id, fetched_query[0].user_id)
def test_success_get_messages(self):
user_id = factories.create_user()
recipient_id = factories.create_user()
text_message = 'qwe'
chat_db.create_message(user_id, recipient_id, text_message)
messages = chat_db.get_messages(user_id, recipient_id)
messages2 = chat_db.get_messages(recipient_id, user_id)
self.assertEqual((user_id, text_message), (messages[0].user_id, messages[0].text))
self.assertEqual((user_id, text_message), (messages2[0].user_id, messages2[0].text))
def test_get_empty_messages(self):
user_id = factories.create_user()
recipient_id = factories.create_user()
id_without_message = factories.create_user()
text_message = 'Hello world'
chat_db.create_message(user_id, recipient_id, text_message)
fetched_data = chat_db.get_messages(id_without_message, user_id)
fetched_data2 = chat_db.get_messages(recipient_id, id_without_message)
self.assertEqual(fetched_data, [])
self.assertEqual(fetched_data2, [])
def test_get_number_new_messages(self):
user_id = factories.create_user()
user_id2 = factories.create_user()
user_id3 = factories.create_user()
text_message = 'Hello world'
chat_db.create_message(user_id, user_id2, text_message)
chat_db.create_message(user_id, user_id2, text_message)
chat_db.create_message(user_id2, user_id, text_message)
fetched_number_new_messages = chat_db.get_number_new_messages(user_id2)
fetched_number = fetched_number_new_messages[user_id]
self.assertEqual(2, fetched_number)
def test_mark_messages_as_read(self):
user1 = factories.create_user()
user2 = factories.create_user()
user3 = factories.create_user()
chat_db.create_message(user2, user1, 'Hello world')
chat_db.create_message(user3, user1, 'Hello world')
chat_db.mark_messages_as_read(user1, user2)
fetched_data = chat_db.get_number_new_messages(user1)
fetched_number = fetched_data.get(user2)
fetched_number2 = fetched_data.get(user3)
self.assertEqual(None, fetched_number)
self.assertEqual(1, fetched_number2)
def test_del_session(self):
user_id1 = factories.create_user()
user_id2 = factories.create_user()
session_id = str(utils.get_random_string(50))
session_id2 = str(utils.get_random_string(50))
data = {'user_id': user_id1}
data2 = {'user_id': user_id2}
user_db.create_session_data(session_id, data)
user_db.create_session_data(session_id2, data2)
user_db.del_session(session_id)
deleted_row = user_db.get_session(session_id)
session2 = user_db.get_session(session_id2)
self.assertEqual(deleted_row, None)
self.assertEqual(session2.data, data2)
