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 chat(request, user_id): new_messages_number = db.get_number_new_messages(user_id) users_data = user_db.get_id_and_username_all_users() data = [] for id, name in users_data: number = new_messages_number.get(id) data.append((id, name, number)) return render_to_response("chat/chat.html", { 'data': data, })