110 lines
3.0 KiB
Python
110 lines
3.0 KiB
Python
import os
|
|
import shutil
|
|
import time
|
|
|
|
import pytest
|
|
|
|
from quixote import cleanup, get_publisher
|
|
from wcs.qommon import errors, sessions
|
|
from wcs.qommon.ident.password_accounts import PasswordAccount
|
|
from wcs.qommon.http_request import HTTPRequest
|
|
from wcs.qommon.template import get_current_theme
|
|
from wcs.categories import Category
|
|
from wcs.workflows import Workflow
|
|
from wcs.formdef import FormDef
|
|
from wcs import fields
|
|
|
|
from utilities import get_app, login, create_temporary_pub, clean_temporary_pub
|
|
|
|
|
|
def setup_module(module):
|
|
cleanup()
|
|
|
|
global pub
|
|
|
|
pub = create_temporary_pub()
|
|
|
|
req = HTTPRequest(None, {'SCRIPT_NAME': '/', 'SERVER_NAME': 'example.net'})
|
|
pub.set_app_dir(req)
|
|
pub.cfg['identification'] = {'methods': ['password']}
|
|
pub.write_cfg()
|
|
|
|
|
|
def teardown_module(module):
|
|
clean_temporary_pub()
|
|
|
|
|
|
def create_agent():
|
|
if pub.user_class.get_users_with_name_identifier('agent'):
|
|
return pub.user_class.get_users_with_name_identifier('agent')[0]
|
|
user1 = pub.user_class(name='agent')
|
|
user1.name_identifiers = ['agent']
|
|
user1.is_admin = False
|
|
user1.store()
|
|
|
|
account1 = PasswordAccount(id='agent')
|
|
account1.set_password('agent')
|
|
account1.user_id = user1.id
|
|
account1.store()
|
|
|
|
pub.cfg['identification'] = {'methods': ['password']}
|
|
pub.write_cfg()
|
|
|
|
return user1
|
|
|
|
|
|
def create_role():
|
|
pub.role_class.wipe()
|
|
role = pub.role_class(name='foobar')
|
|
role.allows_backoffice_access = True
|
|
role.store()
|
|
return role
|
|
|
|
|
|
def teardown_module(module):
|
|
shutil.rmtree(pub.APP_DIR)
|
|
|
|
|
|
def test_with_agent():
|
|
user = create_agent()
|
|
app = login(get_app(pub), username='agent', password='agent')
|
|
resp = app.get('/backoffice/', status=403)
|
|
role = create_role()
|
|
user.roles = [role.id]
|
|
user.store()
|
|
|
|
app = login(get_app(pub), username='agent', password='agent')
|
|
resp = app.get('/backoffice/')
|
|
# check user is automatically redirected to management/
|
|
assert resp.location == 'http://example.net/backoffice/management/'
|
|
|
|
|
|
def test_with_agent_submitter():
|
|
user = create_agent()
|
|
role = create_role()
|
|
user.roles = [role.id]
|
|
user.store()
|
|
FormDef.wipe()
|
|
formdef = FormDef()
|
|
formdef.name = 'form test'
|
|
formdef.fields = []
|
|
formdef.store()
|
|
|
|
app = login(get_app(pub), username='agent', password='agent')
|
|
resp = app.get('/backoffice/')
|
|
# check user is automatically redirected to management/
|
|
assert resp.location == 'http://example.net/backoffice/management/'
|
|
resp = resp.maybe_follow()
|
|
assert resp.pyquery('#sidepage-menu .icon-management')
|
|
assert not resp.pyquery('#sidepage-menu .icon-submission')
|
|
|
|
formdef.backoffice_submission_roles = [role.id]
|
|
formdef.store()
|
|
resp = app.get('/backoffice/') # still redirecting to management
|
|
assert resp.location == 'http://example.net/backoffice/management/'
|
|
resp = resp.maybe_follow()
|
|
|
|
# check the management and submission links are present in side menu
|
|
assert resp.pyquery('#sidepage-menu .icon-management')
|
|
assert resp.pyquery('#sidepage-menu .icon-submission')
|