def share_edit(request, object_type, object_id, user_id):
model = Network if object_type == 'network' else Host
obj, edit = get_object_or_forbidden(model, object_id, request.user)
user = User.objects.get(pk=user_id)
if edit:
revoke_edit(obj, user)
else:
grant_edit(obj, user)
return share_list(request, object_type, object_id)
def share_edit(request, object_type, object_id, user_id):
model = Network if object_type == 'network' else Host
obj, edit = get_object_or_forbidden(model, object_id, request.user)
user = User.objects.get(pk=user_id)
if edit:
revoke_edit(obj, user)
else:
grant_edit(obj, user)
return share_list(request, object_type, object_id)
def test_user_network_share(self):
grant_access(self.net, self.other_user)
access = user_has_access(self.net, self.other_user)
self.assertEqual(access, True)
access = user_can_edit(self.net, self.other_user)
self.assertEqual(access, True)
revoke_edit(self.net, self.other_user)
access = user_can_edit(self.net, self.other_user)
self.assertEqual(access, False)
revoke_access(self.net, self.other_user)
access = user_has_access(self.net, self.other_user)
self.assertEqual(access, False)
