def test_user_is_not_staff(self):
""" Validate the querysets returned for a user that is not staff.
Regular users should only have access to view sessions in the
organization that they belong to.
"""
# Get the querysets
querysets = get_queryset(self.user)
# Organizations
self.assertEqual(1, len(querysets['organizations']), "There is only one org")
self.assertIn(self.info.organization, querysets['organizations'])
# Sessions
for session in Session.objects.filter(organization=self.info.organization):
self.assertIn(session, querysets['sessions'])
def test_user_is_staff(self):
""" Validate the querysets returned from a user that is staff.
Staff should have access to view all sessions in all
organizations. Verify that that is what is returned.
"""
# Set user as staff
self.user.is_staff = True
self.user.save()
# Generate querysets
querysets = get_queryset(self.user)
# Organizatioins
for organization in Organization.objects.all():
self.assertIn(organization, querysets['organizations'])
# Sessions
for session in Session.objects.all():
self.assertIn(session, querysets['sessions'])