import uuid from datetime import datetime import pytest from freezegun import freeze_time from app import db from app.models import Service from app.dao.services_dao import dao_archive_service, dao_fetch_service_by_id from app.dao.api_key_dao import expire_api_key from app.dao.templates_dao import dao_update_template from tests import create_authorization_header, unwrap_function from tests.app.db import create_template, create_api_key def test_archive_only_allows_post(client, notify_db_session): auth_header = create_authorization_header() response = client.get('/service/{}/archive'.format(uuid.uuid4()), headers=[auth_header]) assert response.status_code == 405 def test_archive_service_errors_with_bad_service_id(client, notify_db_session): auth_header = create_authorization_header() response = client.post('/service/{}/archive'.format(uuid.uuid4()), headers=[auth_header]) assert response.status_code == 404 def test_deactivating_inactive_service_does_nothing(client, sample_service): auth_header = create_authorization_header() sample_service.active = False response = client.post('/service/{}/archive'.format(sample_service.id), headers=[auth_header]) assert response.status_code == 204 assert sample_service.name == 'Sample service' @pytest.fixture def archived_service(client, notify_db, sample_service): create_template(sample_service, template_name='a') create_template(sample_service, template_name='b') create_api_key(sample_service) create_api_key(sample_service) notify_db.session.commit() auth_header = create_authorization_header() response = client.post('/service/{}/archive'.format(sample_service.id), headers=[auth_header]) assert response.status_code == 204 assert response.data == b'' return sample_service @freeze_time('2018-07-07 12:00:00') def test_deactivating_service_changes_name_and_email(client, sample_service): auth_header = create_authorization_header() client.post('/service/{}/archive'.format(sample_service.id), headers=[auth_header]) archived_service = dao_fetch_service_by_id(sample_service.id) assert archived_service.name == '_archived_2018-07-07_Sample service' assert archived_service.email_from == '_archived_2018-07-07_sample.service' def test_deactivating_service_revokes_api_keys(archived_service): assert len(archived_service.api_keys) == 2 for key in archived_service.api_keys: assert key.expiry_date is not None assert key.version == 2 def test_deactivating_service_archives_templates(archived_service): assert len(archived_service.templates) == 2 for template in archived_service.templates: assert template.archived is True assert template.version == 2 def test_deactivating_service_creates_history(archived_service): ServiceHistory = Service.get_history_model() history = ServiceHistory.query.filter_by( id=archived_service.id ).order_by( ServiceHistory.version.desc() ).first() assert history.version == 2 assert history.active is False @pytest.fixture def archived_service_with_deleted_stuff(client, sample_service): with freeze_time('2001-01-01'): template = create_template(sample_service, template_name='a') api_key = create_api_key(sample_service) expire_api_key(sample_service.id, api_key.id) template.archived = True dao_update_template(template) with freeze_time('2002-02-02'): auth_header = create_authorization_header() response = client.post('/service/{}/archive'.format(sample_service.id), headers=[auth_header]) assert response.status_code == 204 assert response.data == b'' return sample_service def test_deactivating_service_doesnt_affect_existing_archived_templates(archived_service_with_deleted_stuff): assert archived_service_with_deleted_stuff.templates[0].archived is True assert archived_service_with_deleted_stuff.templates[0].updated_at == datetime(2001, 1, 1, 0, 0, 0) assert archived_service_with_deleted_stuff.templates[0].version == 2 def test_deactivating_service_doesnt_affect_existing_revoked_api_keys(archived_service_with_deleted_stuff): assert archived_service_with_deleted_stuff.api_keys[0].expiry_date == datetime(2001, 1, 1, 0, 0, 0) assert archived_service_with_deleted_stuff.api_keys[0].version == 2 def test_deactivating_service_rolls_back_everything_on_error(sample_service, sample_api_key, sample_template): unwrapped_deactive_service = unwrap_function(dao_archive_service) unwrapped_deactive_service(sample_service.id) assert sample_service in db.session.dirty assert sample_api_key in db.session.dirty assert sample_template in db.session.dirty