def test_returns_empty_list_if_no_licensee(self):
"""Return empty json list if no licensee provided"""
licensee = mommy.make('kpc.Licensee')
self.user.user_permissions.add(self.p)
self.user.profile.licensees.add(licensee)
response = licensee_contacts(self.request)
self.assertEqual(response.content, b'[]')
def test_only_get_allowed(self):
"""
405 response if request is not a GET
"""
self.user.user_permissions.add(self.p)
for method in ['post', 'put', 'delete', 'head', 'options', 'trace']:
request = getattr(self.factory, method)('')
request.user = self.user
response = licensee_contacts(request)
self.assertEqual(response.status_code, 405)
def test_returns_user_list_if_licensee(self):
"""
Return json list of users associated
to provided licensee
"""
licensee = mommy.make('kpc.Licensee')
self.user.user_permissions.add(self.p)
self.user.profile.licensees.add(licensee)
request = self.factory.get('', {'licensee': licensee.id})
request.user = self.user
response = licensee_contacts(request)
self.assertEqual(json.loads(response.content),
[{'id': self.user.id, 'name': self.user.profile.get_user_display_name()}])
def test_not_accessible_without_perm(self):
"""Cannot access without perm"""
with self.assertRaises(PermissionDenied):
licensee_contacts(self.request)