Remove link for adding service if nonwhitelist user

This commit is contained in:
Imdad Ahad
2016-10-25 18:12:46 +01:00
parent 84762edef4
commit bb85782230
3 changed files with 24 additions and 2 deletions

View File

@@ -1,8 +1,11 @@
from flask import (render_template, redirect, url_for, session)
import re
from flask import (current_app, render_template, redirect, url_for, session)
from flask_login import login_required, current_user
from app.main import main
from app import service_api_client
from app.notify_client.service_api_client import ServicesBrowsableItem
from app.utils import user_in_whitelist
@main.route("/services")
@@ -11,7 +14,8 @@ def choose_service():
return render_template(
'views/choose-service.html',
services=[ServicesBrowsableItem(x) for x in
service_api_client.get_services({'user_id': current_user.id})['data']]
service_api_client.get_services({'user_id': current_user.id})['data']],
can_add_service=user_in_whitelist(current_user.email_address)
)

View File

@@ -12,12 +12,14 @@
</h1>
{{ browse_list(services) }}
{% if can_add_service %}
{{ browse_list([
{
'title': 'Add a new service…',
'link': url_for('.add_service')
},
]) }}
{% endif %}
{% endblock %}

View File

@@ -487,6 +487,22 @@ def platform_admin_user(fake_uuid):
return user
@pytest.fixture(scope='function')
def nonwhitelist_user(fake_uuid):
from app.notify_client.user_api_client import User
user_data = {'id': fake_uuid,
'name': 'Platform admin user',
'password': 'somepassword',
'email_address': 'someuser@notonwhitelist.com',
'mobile_number': '07700 900762',
'state': 'active',
'failed_login_count': 0,
'permissions': {}
}
user = User(user_data)
return user
@pytest.fixture(scope='function')
def api_user_active(fake_uuid):
from app.notify_client.user_api_client import User