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 user_chat(request, recipient_id, user_id):
user_id = int(user_id)
recipient_id = int(recipient_id)
message_history = db.get_messages(user_id, recipient_id)
db.mark_messages_as_read(user_id, recipient_id)
user = user_db.get_user(user_id=user_id)
recipient = user_db.get_user(user_id=recipient_id)
if request.method == 'POST':
text_message = request.POST['message']
db.create_message(user_id, recipient_id, text_message)
return redirect('/chat/user/{}/'.format(recipient_id))
return render_to_response('chat/user_chat.html', {
'recipient_id': recipient.id,
'user_id': user.id,
'message_history': message_history,
'user_name': user.username,
'recipient_name': recipient.username
})
