fix team tests

This commit is contained in:
Wayne Witzel III
2016-04-15 15:26:05 -04:00
parent d2a81f46e3
commit 6229e978e9
3 changed files with 10 additions and 5 deletions

View File

@@ -94,6 +94,11 @@ class Role(CommonModelNameNotUnique):
return reverse('api:role_detail', args=(self.pk,))
def __contains__(self, user):
if user.__class__.__name__ == 'Team':
team_type = ContentType.objects.get_for_model(user)
roles = Role.objects.filter(content_type__pk=team_type.id,
object_id=user.id)
return self.ancestors.filter(pk__in=roles).exists()
return self.ancestors.filter(members=user).exists()
def rebuild_role_ancestor_list(self):