88 lines
2.9 KiB
Python
88 lines
2.9 KiB
Python
# authentic2 - versatile identity manager
|
|
# Copyright (C) 2010-2019 Entr'ouvert
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify it
|
|
# under the terms of the GNU Affero General Public License as published
|
|
# by the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import json
|
|
|
|
from webtest import Upload
|
|
|
|
from django_rbac.utils import get_ou_model, get_role_model
|
|
|
|
from .utils import login
|
|
|
|
OU = get_ou_model()
|
|
Role = get_role_model()
|
|
|
|
|
|
def test_manager_ou_export(app, admin, ou1, role_ou1, ou2, role_ou2):
|
|
response = login(app, admin, 'a2-manager-ous')
|
|
|
|
export_response = response.click('Export')
|
|
export = export_response.json
|
|
|
|
assert list(export.keys()) == ['ous']
|
|
assert len(export['ous']) == 3
|
|
assert set([ou['slug'] for ou in export['ous']]) == set(['default', 'ou1', 'ou2'])
|
|
|
|
response.form.set('search-text', 'ou1')
|
|
search_response = response.form.submit()
|
|
|
|
export_response = search_response.click('Export')
|
|
export = export_response.json
|
|
|
|
assert len(export['ous']) == 1
|
|
assert export['ous'][0]['slug'] == 'ou1'
|
|
|
|
|
|
def test_manager_ou_import(app, admin, ou1, role_ou1, ou2, role_ou2):
|
|
response = login(app, admin, 'a2-manager-ous')
|
|
|
|
export_response = response.click('Export')
|
|
export = export_response.json
|
|
|
|
assert len(export['ous']) == 3
|
|
assert not 'roles' in export
|
|
ou1.delete()
|
|
ou2.delete()
|
|
|
|
resp = app.get('/manage/organizational-units/')
|
|
resp = resp.click('Import')
|
|
assert 'Organizational Units Export File' in resp.text
|
|
resp.form['site_json'] = Upload('export.json', json.dumps(export).encode(), 'application/json')
|
|
resp = resp.form.submit().follow()
|
|
|
|
assert OU.objects.filter(name=ou1.name).exists()
|
|
assert OU.objects.filter(name=ou2.name).exists()
|
|
|
|
export_response = response.click('Export')
|
|
new_export = export_response.json
|
|
assert len(export['ous']) == 3
|
|
assert new_export['ous'][1]['uuid'] == export['ous'][1]['uuid']
|
|
assert new_export['ous'][2]['uuid'] == export['ous'][2]['uuid']
|
|
|
|
# in case roles are present in export file, they must not be imported
|
|
export['roles'] = [
|
|
{
|
|
"uuid": "27255f404cb140df9a577da76b59f285",
|
|
"slug": "should_not_exist",
|
|
"name": "should_not_exist",
|
|
}
|
|
]
|
|
resp = resp.click('Import')
|
|
resp.form['site_json'] = Upload('export.json', json.dumps(export).encode(), 'application/json')
|
|
resp = resp.form.submit().follow()
|
|
|
|
assert not Role.objects.filter(slug="should_not_exist").exists()
|