def test_add_to_group(self):
setattr(request, 'session', 'session')
msg = FallbackStorage(request)
setattr(request, '_messages', msg)
tutor_admin = tutor.TutorAdmin(tutor.Tutor, self.site)
queryset = tutor.Tutor.objects.all()
tutor_admin.add_to_group(request, queryset)
msg = [m.message for m in get_messages(request)]
self.assertIn("{} users added to the group 'tutors'.".format(10), msg)
def test_add_to_group_no_tutor_group(self):
setattr(request, 'session', 'session')
msg = FallbackStorage(request)
setattr(request, '_messages', msg)
for group in Group.objects.all():
group.delete()
tutor_admin = tutor.TutorAdmin(tutor.Tutor, self.site)
queryset = tutor.Tutor.objects.all()
tutor_admin.add_to_group(request, queryset)
msg = [m.message for m in get_messages(request)]
self.assertIn("Group tutors doesn't exist.", msg)