Updated form and fixed existing tests.

This commit is contained in:
Nicholas Staples
2016-03-03 13:00:12 +00:00
parent 32b84f4ce2
commit 9e710711cb
7 changed files with 73 additions and 58 deletions

View File

@@ -9,6 +9,10 @@ from app.main.encryption import hashpw
from app import user_api_client
# TODO fix up this, do we really need this class why not just use the clients
# directly??
@login_manager.user_loader
def load_user(user_id):
return get_user_by_id(user_id)

View File

@@ -110,6 +110,10 @@ class RegisterUserFromInviteForm(Form):
class InviteUserForm(Form):
email_address = email_address('Their email address')
send_messages = BooleanField("Send messages")
manage_service = BooleanField("Manage service")
manage_api_keys = BooleanField("Manage API keys")
class TwoFactorForm(Form):
def __init__(self, validate_code_func, *args, **kwargs):

View File

@@ -15,9 +15,10 @@ from notifications_python_client.errors import HTTPError
from app.main import main
from app.main.forms import InviteUserForm
from app.main.dao.services_dao import get_service_by_id_or_404
from app.main.dao.services_dao import get_service_by_id
from app import user_api_client
from app import invite_api_client
from app.utils import user_has_permissions
fake_users = [
{
@@ -32,45 +33,35 @@ fake_users = [
@main.route("/services/<service_id>/users")
@login_required
@user_has_permissions('manage_users', 'manage_templates', 'manage_settings')
def manage_users(service_id):
try:
users = user_api_client.get_users_for_service(service_id=service_id)
invited_users = invite_api_client.get_invites_for_service(service_id=service_id)
return render_template('views/manage-users.html',
service_id=service_id,
users=users,
current_user=current_user,
invited_users=invited_users)
except HTTPError as e:
if e.status_code == 404:
abort(404)
else:
raise e
users = user_api_client.get_users_for_service(service_id=service_id)
invited_users = invite_api_client.get_invites_for_service(service_id=service_id)
return render_template('views/manage-users.html',
service_id=service_id,
users=users,
current_user=current_user,
invited_users=invited_users)
@main.route("/services/<service_id>/users/invite", methods=['GET', 'POST'])
@login_required
@user_has_permissions('manage_users', 'manage_templates', 'manage_settings')
def invite_user(service_id):
service = get_service_by_id(service_id)
form = InviteUserForm()
if form.validate_on_submit():
email_address = form.email_address.data
permissions = _get_permissions(request.form)
try:
invited_user = invite_api_client.create_invite(current_user.id, service_id, email_address, permissions)
flash('Invite sent to {}'.format(invited_user.email_address), 'default_with_tick')
return redirect(url_for('.manage_users', service_id=service_id))
except HTTPError as e:
if e.status_code == 404:
abort(404)
else:
raise e
invited_user = invite_api_client.create_invite(current_user.id, service_id, email_address, permissions)
flash('Invite sent to {}'.format(invited_user.email_address), 'default_with_tick')
return redirect(url_for('.manage_users', service_id=service_id))
return render_template(
'views/invite-user.html',
user={},
service=get_service_by_id_or_404(service_id),
user=None,
service_id=service_id,
form=form
)
@@ -78,29 +69,40 @@ def invite_user(service_id):
@main.route("/services/<service_id>/users/<user_id>", methods=['GET', 'POST'])
@login_required
@user_has_permissions('manage_users', 'manage_templates', 'manage_settings')
def edit_user_permissions(service_id, user_id):
# TODO we should probably using the service id here in the get user
# call as well. eg. /user/<user_id>?&service_id=service_id
user = user_api_client.get_user(user_id)
service = get_service_by_id(service_id)
if request.method == 'POST':
form = InviteUserForm(**{
'email_address': user.email_address,
'send_messages': user.has_permissions(['send_texts', 'send_emails', 'send_letters']),
'manage_service': user.has_permissions(['manage_users', 'manage_templates', 'manage_settings']),
'manage_api_keys': user.has_permissions(['manage_api_keys', 'access_developer_docs'])
})
if form.validate_on_submit():
return redirect(url_for('.manage_users', service_id=service_id))
return render_template(
'views/invite-user.html',
user=fake_users[int(user_id)],
user_id=user_id,
service=get_service_by_id_or_404(service_id),
user=user,
form=form,
service_id=service_id
)
@main.route("/services/<service_id>/users/<user_id>/delete", methods=['GET', 'POST'])
@login_required
@user_has_permissions('manage_users', 'manage_templates', 'manage_settings')
def delete_user(service_id, user_id):
user = user_api_client.get_user(user_id)
service = get_service_by_id(service_id)
if request.method == 'POST':
return redirect(url_for('.manage_users', service_id=service_id))
user = fake_users[int(user_id)]
flash(
'Are you sure you want to delete {}s account?'.format(user.get('name') or user['email_localpart']),
'delete'
@@ -109,13 +111,12 @@ def delete_user(service_id, user_id):
return render_template(
'views/invite-user.html',
user=user,
user_id=user_id,
service=get_service_by_id_or_404(service_id),
service_id=service_id
)
@main.route("/services/<service_id>/cancel-invited-user/<invited_user_id>", methods=['GET'])
@user_has_permissions('manage_users', 'manage_templates', 'manage_settings')
def cancel_invited_user(service_id, invited_user_id):
invite_api_client.cancel_invited_user(service_id=service_id, invited_user_id=invited_user_id)

View File

@@ -130,8 +130,8 @@ class InvitedUser(object):
self.status = status
self.created_at = created_at
def has_permissions(self, permission):
return permission in self.permissions
def has_permissions(self, permissions):
return set(self.permissions) > set(permissions)
def serialize(self):
return {'id': self.id,

View File

@@ -16,22 +16,16 @@ Manage users GOV.UK Notify
<div class="grid-row">
<form method="post" class="column-three-quarters">
{% if user %}
<p class='bottom-gutter'>
{{ current_user.email_address }}
</p>
{% else %}
{{ textbox(form.email_address, hint='Email address must end in .gov.uk', width='1-1') }}
{% endif %}
{{ textbox(form.email_address, hint='Email address must end in .gov.uk', width='1-1') }}
<fieldset class='yes-no-wrapper'>
<legend class='heading-small'>
Permissions
</legend>
<span class="form-hint">All team members can see message history</span>
{{ yes_no('send_messages', 'Send messages', user.permission_send_messages) }}
{{ yes_no('manage_service', 'Manage service', user.permission_manage_service) }}
{{ yes_no('manage_api_keys', 'Manage API keys', user.permission_manage_api_keys) }}
{{ yes_no(form.send_messages.name, form.send_messages.label, form.send_messages.value) }}
{{ yes_no(form.manage_service.name, form.manage_service.label, form.manage_service.value) }}
{{ yes_no(form.manage_api_keys, form.manage_api_keys, form.manage_api_keys.value) }}
</fieldset>
{% if user %}

View File

@@ -28,9 +28,9 @@ Manage users GOV.UK Notify
{% call field() %}
{{ item.name }}
{% endcall %}
{{ boolean_field(item.has_permissions(service_id, 'send_messages')) }}
{{ boolean_field(item.has_permissions(service_id, 'manage_service')) }}
{{ boolean_field(item.has_permissions(service_id, 'manage_api_keys')) }}
{{ boolean_field(item.has_permissions(['send_texts', 'send_emails', 'send_letters'], service_id=service_id)) }}
{{ boolean_field(item.has_permissions(['manage_users', 'manage_templates', 'manage_settings'], service_id=service_id)) }}
{{ boolean_field(item.has_permissions(['manage_api_keys', 'access_developer_docs'], service_id=service_id)) }}
{% endcall %}
{% if invited_users %}
@@ -40,9 +40,9 @@ Manage users GOV.UK Notify
{% call field() %}
{{ item.email_address }}
{% endcall %}
{{ boolean_field(item.has_permissions('send_messages')) }}
{{ boolean_field(item.has_permissions('manage_service')) }}
{{ boolean_field(item.has_permissions('manage_api_keys')) }}
{{ boolean_field(item.has_permissions(['send_texts', 'send_emails', 'send_letters'])) }}
{{ boolean_field(item.has_permissions(['manage_users', 'manage_templates', 'manage_settings'])) }}
{{ boolean_field(item.has_permissions(['manage_api_keys', 'access_developer_docs'])) }}
{% if item.status == 'pending' %}
{% call field(align='right') %}
<a href="{{ url_for('.cancel_invited_user', service_id=service_id, invited_user_id=item.id)}}">Cancel invitation</a>

View File

@@ -9,7 +9,8 @@ def test_should_show_overview_page(
mock_login,
mock_get_service,
mock_get_users_by_service,
mock_get_invites_for_service
mock_get_invites_for_service,
mock_has_permissions
):
with app_.test_request_context():
with app_.test_client() as client:
@@ -25,7 +26,8 @@ def test_should_show_page_for_one_user(
app_,
api_user_active,
mock_login,
mock_get_service
mock_get_service,
mock_has_permissions
):
with app_.test_request_context():
with app_.test_client() as client:
@@ -41,7 +43,8 @@ def test_redirect_after_saving_user(
mock_login,
mock_get_service,
mock_get_users_by_service,
mock_get_invites_for_service
mock_get_invites_for_service,
mock_has_permissions
):
with app_.test_request_context():
with app_.test_client() as client:
@@ -60,7 +63,9 @@ def test_should_show_page_for_inviting_user(
app_,
api_user_active,
mock_login,
mock_get_service
mock_get_user,
mock_get_service,
mock_has_permissions
):
with app_.test_request_context():
with app_.test_client() as client:
@@ -76,9 +81,12 @@ def test_invite_user(
service_one,
api_user_active,
mock_login,
mock_get_user,
mock_get_service,
mock_get_users_by_service,
mock_create_invite,
mock_get_invites_for_service
mock_get_invites_for_service,
mock_has_permissions
):
from_user = api_user_active.id
service_id = service_one['id']
@@ -106,7 +114,11 @@ def test_invite_user(
assert flash_banner == 'Invite sent to test@example.gov.uk'
def test_cancel_invited_user_cancels_user_invitations(app_, api_user_active, mock_login, mocker):
def test_cancel_invited_user_cancels_user_invitations(app_,
api_user_active,
mock_login,
mocker,
mock_has_permissions):
with app_.test_request_context():
with app_.test_client() as client:
mocker.patch('app.invite_api_client.cancel_invited_user')