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 })