mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-05 10:53:28 -05:00
74 lines
3.8 KiB
Python
74 lines
3.8 KiB
Python
from flask import url_for
|
|
from app.main.dao import verify_codes_dao, services_dao
|
|
from tests import create_test_user
|
|
from app.models import User
|
|
|
|
|
|
def test_get_should_render_add_service_template(app_,
|
|
db_,
|
|
db_session,
|
|
mock_api_user,
|
|
mock_get_service,
|
|
mock_get_services,
|
|
mock_user_loader,
|
|
mock_user_dao_get_by_email):
|
|
with app_.test_request_context():
|
|
with app_.test_client() as client:
|
|
client.login(mock_api_user)
|
|
response = client.get(url_for('main.add_service'))
|
|
assert response.status_code == 200
|
|
assert 'Set up notifications for your service' in response.get_data(as_text=True)
|
|
|
|
|
|
def test_should_add_service_and_redirect_to_next_page(app_,
|
|
db_,
|
|
db_session,
|
|
mock_create_service,
|
|
mock_get_services,
|
|
mock_api_user,
|
|
mock_user_loader,
|
|
mock_user_dao_get_by_email):
|
|
with app_.test_request_context():
|
|
with app_.test_client() as client:
|
|
client.login(mock_api_user)
|
|
response = client.post(
|
|
url_for('main.add_service'),
|
|
data={'name': 'testing the post'})
|
|
assert response.status_code == 302
|
|
assert response.location == url_for('main.service_dashboard', service_id=101, _external=True)
|
|
assert mock_create_service.called
|
|
assert mock_get_services.called
|
|
|
|
|
|
def test_should_return_form_errors_when_service_name_is_empty(app_,
|
|
db_,
|
|
db_session,
|
|
mock_api_user,
|
|
mock_get_service,
|
|
mock_get_services,
|
|
mock_user_loader,
|
|
mock_user_dao_get_by_email):
|
|
with app_.test_request_context():
|
|
with app_.test_client() as client:
|
|
client.login(mock_api_user)
|
|
response = client.post(url_for('main.add_service'), data={})
|
|
assert response.status_code == 200
|
|
assert 'Service name can not be empty' in response.get_data(as_text=True)
|
|
|
|
|
|
def test_should_return_form_errors_with_duplicate_service_name(app_,
|
|
db_,
|
|
db_session,
|
|
mock_get_services,
|
|
mock_user_loader,
|
|
mock_api_user,
|
|
mock_user_dao_get_by_email):
|
|
with app_.test_request_context():
|
|
with app_.test_client() as client:
|
|
client.login(mock_api_user)
|
|
response = client.post(
|
|
url_for('main.add_service'), data={'name': 'service_one'})
|
|
assert response.status_code == 200
|
|
assert 'Service name already exists' in response.get_data(as_text=True)
|
|
assert mock_get_services.called
|