diff --git a/tests/test_acl.py b/tests/test_acl.py index 1a81175..4143037 100644 --- a/tests/test_acl.py +++ b/tests/test_acl.py @@ -186,11 +186,14 @@ def test_liege_service_admin(app, base): # Select3 only shows users from Liège service_role = app.get('/manage/roles/%s/' % base.role_service_liege.pk) - select2_url = service_role.pyquery('select#id_user')[0].attrib['data-ajax--url'] - select2_field_id = service_role.pyquery('select#id_user')[0].attrib['data-field_id'] + select2_url = service_role.pyquery('select#id_user_or_role')[0].attrib['data-ajax--url'] + select2_field_id = service_role.pyquery('select#id_user_or_role')[0].attrib['data-field_id'] select2_response = app.get(select2_url, params={'field_id': select2_field_id, 'term': ''}) assert select2_response.json['more'] is False assert len(select2_response.json['results']) == 3 ids = set(result['id'] for result in select2_response.json['results']) - assert User.objects.filter(id__in=ids, ou=base.ou_liege).count() == 3 + user_ids = set(int(_id.split('-')[1]) for _id in ids if _id.startswith('user-')) + role_ids = set(int(_id.split('-')[1]) for _id in ids if _id.startswith('role-')) + assert set(User.objects.filter(id__in=user_ids, ou=base.ou_liege).values_list('id', flat=True)) == user_ids + assert set(Role.objects.filter(id__in=role_ids, ou=base.ou_liege).values_list('id', flat=True)) == role_ids