Merge branch 'master' into celery-jobs

This commit is contained in:
Martyn Inglis
2016-02-25 12:06:53 +00:00
15 changed files with 328 additions and 2 deletions

View File

@@ -0,0 +1,23 @@
from app.models import InvitedUser
from app.dao.invited_user_dao import save_invited_user
def test_create_invited_user(notify_db, notify_db_session, sample_service):
assert InvitedUser.query.count() == 0
email_address = 'invited_user@service.gov.uk'
invite_from = sample_service.users[0]
data = {
'service': sample_service,
'email_address': email_address,
'from_user': invite_from
}
invited_user = InvitedUser(**data)
save_invited_user(invited_user)
assert InvitedUser.query.count() == 1
assert invited_user.email_address == email_address
assert invited_user.from_user == invite_from

View File

View File

@@ -0,0 +1,70 @@
import json
from tests import create_authorization_header
def test_create_invited_user(notify_api, sample_service):
with notify_api.test_request_context():
with notify_api.test_client() as client:
email_address = 'invited_user@service.gov.uk'
invite_from = sample_service.users[0]
data = {
'service': str(sample_service.id),
'email_address': email_address,
'from_user': invite_from.id
}
data = json.dumps(data)
auth_header = create_authorization_header(
path='/service/{}/invite'.format(sample_service.id),
method='POST',
request_body=data
)
response = client.post(
'/service/{}/invite'.format(sample_service.id),
headers=[('Content-Type', 'application/json'), auth_header],
data=data
)
assert response.status_code == 201
json_resp = json.loads(response.get_data(as_text=True))
assert json_resp['data']['service'] == str(sample_service.id)
assert json_resp['data']['email_address'] == email_address
assert json_resp['data']['from_user'] == invite_from.id
assert json_resp['data']['id']
def test_create_invited_user_invalid_email(notify_api, sample_service):
with notify_api.test_request_context():
with notify_api.test_client() as client:
email_address = 'notanemail'
invite_from = sample_service.users[0]
data = {
'service': str(sample_service.id),
'email_address': email_address,
'from_user': invite_from.id
}
data = json.dumps(data)
auth_header = create_authorization_header(
path='/service/{}/invite'.format(sample_service.id),
method='POST',
request_body=data
)
response = client.post(
'/service/{}/invite'.format(sample_service.id),
headers=[('Content-Type', 'application/json'), auth_header],
data=data
)
assert response.status_code == 400
json_resp = json.loads(response.get_data(as_text=True))
assert json_resp['result'] == 'error'
assert json_resp['message'] == {'email_address': ['Invalid email']}

View File

@@ -4,6 +4,7 @@ import uuid
from app.dao.users_dao import save_model_user
from app.models import User
from tests import create_authorization_header
from tests.app.conftest import sample_service as create_service
def test_get_service_list(notify_api, service_factory):
@@ -341,3 +342,49 @@ def test_update_service_should_404_if_id_is_invalid(notify_api, notify_db):
headers=[('Content-Type', 'application/json'), auth_header]
)
assert resp.status_code == 404
def test_get_users_by_service(notify_api, notify_db, notify_db_session, sample_user):
with notify_api.test_request_context():
with notify_api.test_client() as client:
sample_service = create_service(notify_db=notify_db, notify_db_session=notify_db_session,
service_name='Sample service', user=sample_user)
auth_header = create_authorization_header(
path='/service/{}/users'.format(sample_service.id),
method='GET'
)
resp = client.get(
'/service/{}/users'.format(sample_service.id),
headers=[('Content-Type', 'application/json'), auth_header]
)
assert resp.status_code == 200
result = json.loads(resp.get_data(as_text=True))
assert len(result['data']) == 1
assert result['data'][0]['name'] == sample_user.name
assert result['data'][0]['email_address'] == sample_user.email_address
assert result['data'][0]['mobile_number'] == sample_user.mobile_number
def test_get_users_for_service_returns_empty_list_if_no_users_associated_with_service(notify_api,
notify_db,
notify_db_session,
sample_service,
sample_user):
with notify_api.test_request_context():
with notify_api.test_client() as client:
sample_service.users.remove(sample_user)
auth_header = create_authorization_header(
path='/service/{}/users'.format(sample_service.id),
method='GET'
)
response = client.get(
'/service/{}/users'.format(sample_service.id),
headers=[('Content-Type', 'application/json'), auth_header]
)
assert response.status_code == 200
result = json.loads(response.get_data(as_text=True))
assert result['data'] == []

View File

@@ -0,0 +1,40 @@
import uuid
from flask import url_for
service_id = str(uuid.uuid4())
def test_url_for_get_services(notify_api):
with notify_api.test_request_context():
url = url_for('service.get_services')
assert str(url) == '/service'
url_with_user_id = url_for('service.get_services', user_id=1)
assert str(url_with_user_id) == '/service?user_id=1'
def test_url_for_get_service_by_id(notify_api):
with notify_api.test_request_context():
url = url_for('service.get_service_by_id', service_id=service_id)
assert str(url) == '/service/{}'.format(service_id)
url_with_user_id = url_for('service.get_service_by_id', service_id=service_id, user_id=1)
assert str(url_with_user_id) == '/service/{0}?user_id={1}'.format(service_id, 1)
def test_url_for_create_service(notify_api):
with notify_api.test_request_context():
url = url_for('service.create_service')
assert str(url) == '/service'.format(service_id)
def test_url_for_update_service(notify_api):
with notify_api.test_request_context():
url = url_for('service.update_service', service_id=service_id)
assert str(url) == '/service/{}'.format(service_id)
def test_url_for_renew_api_key(notify_api):
with notify_api.test_request_context():
url = url_for('service.renew_api_key', service_id=service_id)
assert str(url) == '/service/{}/api-key'.format(service_id)