def test_can_get_same_team_users(self):
new_team2 = factory.TeamFactory()
expected_users = [factory.UserFactory() for _ in range(10)]
new_team2.users.add(expected_users[0])
for user in expected_users[0:6]:
self.new_team.users.add(user)
for user in expected_users[4:]:
new_team2.users.add(user)
self.new_team.save()
resulted_users = Team.users_at_same_team(expected_users[0])
self.assertEqual(resulted_users, expected_users)
def test_can_get_same_team_users(self):
new_team2 = factory.TeamFactory()
expected_users = [factory.UserFactory() for _ in range(10)]
new_team2.users.add(expected_users[0])
for user in expected_users[0:6]:
self.new_team.users.add(user)
for user in expected_users[4:]:
new_team2.users.add(user)
self.new_team.save()
resulted_users = Team.users_at_same_team(expected_users[0])
self.assertEqual(resulted_users, expected_users)
def queryset(self, request):
qs = None
if request.user.has_perm(self.perm_add_database_infra):
qs = super(TaskHistoryAdmin, self).queryset(request)
return qs
if request.GET.get('user'):
query_dict_copy = request.GET.copy()
del query_dict_copy['user']
request.GET = query_dict_copy
qs = super(TaskHistoryAdmin, self).queryset(request)
same_team_users = Team.users_at_same_team(request.user)
return qs.filter(user__in=[user.username for user in same_team_users])
def queryset(self, request):
qs = None
if request.user.has_perm(self.perm_add_database_infra):
qs = super(TaskHistoryAdmin, self).queryset(request)
return qs
if request.GET.get('user'):
query_dict_copy = request.GET.copy()
del query_dict_copy['user']
request.GET = query_dict_copy
qs = super(TaskHistoryAdmin, self).queryset(request)
same_team_users = Team.users_at_same_team(request.user)
return qs.filter(user__in=[user.username for user in same_team_users])
def test_can_get_same_team_users(self):
new_team2 = mommy.make_recipe('account.team')
expected_users = sorted(mommy.make_recipe('account.user',
_quantity=10),
key=lambda u: u.id)
new_team2.users.add(expected_users[0])
for user in expected_users[0:6]:
self.new_team.users.add(user)
self.new_team.save()
for user in expected_users[4:]:
new_team2.users.add(user)
new_team2.save()
resulted_users = sorted(Team.users_at_same_team(expected_users[0]),
key=lambda u: u.id)
self.assertEqual(resulted_users, expected_users)
def test_can_get_same_team_users_empty_user(self):
resulted_users = set(Team.users_at_same_team(None))
self.assertFalse(resulted_users)
def test_can_get_same_team_users_empty_user(self):
resulted_users = set(Team.users_at_same_team(None))
self.assertFalse(resulted_users)
