notifications-admin/tests/app/main/views/test_add_service.py

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