def test_function_get_all_roles(self):
first_user = UserCreationForm(
data = dict(
username = '******',
password1 = '123456',
password2 = '123456',
)
).save()
second_user = UserCreationForm(
data = dict(
username = '******',
password1 = '654321',
password2 = '654321',
)
).save()
first_role = SGRSRole(
name = 'first_role',
description = 'first role for test',
can_download = True,
)
first_role.save()
second_role = SGRSRole(
name = 'second_role',
description = 'second role for test',
can_download = True,
)
second_role.save()
first_user_assignment = SGRSUserAssignment(user = first_user)
first_user_assignment.save()
first_user_assignment.roles = [second_role, first_role]
first_user_assignment.save()
second_user_assignment = SGRSUserAssignment(user = second_user)
second_user_assignment.save()
second_user_assignment.roles = [first_role, ]
second_user_assignment.save()
sort_key_function = lambda x:x.id
self.assertEqual(
sorted(first_user_assignment.roles.all(),key = sort_key_function),
sorted(first_user.get_all_roles(),key = sort_key_function)
)
self.assertEqual(
sorted(second_user_assignment.roles.all(),key = sort_key_function),
sorted(second_user.get_all_roles(),key = sort_key_function)
)
