mirror of
https://github.com/GSA/notifications-api.git
synced 2026-01-21 10:01:59 -05:00
88 lines
3.1 KiB
Python
88 lines
3.1 KiB
Python
import uuid
|
|
from datetime import datetime
|
|
|
|
import pytest
|
|
from freezegun import freeze_time
|
|
|
|
from app.models import Service
|
|
from tests import create_admin_authorization_header
|
|
|
|
|
|
@pytest.mark.parametrize("endpoint", ["suspend", "resume"])
|
|
def test_only_allows_post(client, endpoint):
|
|
auth_header = create_admin_authorization_header()
|
|
response = client.get(
|
|
"/service/{}/{}".format(uuid.uuid4(), endpoint), headers=[auth_header]
|
|
)
|
|
assert response.status_code == 405
|
|
|
|
|
|
@pytest.mark.parametrize("endpoint", ["suspend", "resume"])
|
|
def test_returns_404_when_service_does_not_exist(client, endpoint):
|
|
auth_header = create_admin_authorization_header()
|
|
response = client.post(
|
|
"/service/{}/{}".format(uuid.uuid4(), endpoint), headers=[auth_header]
|
|
)
|
|
assert response.status_code == 404
|
|
|
|
|
|
@pytest.mark.parametrize("action, active", [("suspend", False), ("resume", True)])
|
|
def test_has_not_effect_when_service_is_already_that_state(
|
|
client, sample_service, action, active, mocker
|
|
):
|
|
mocked = mocker.patch("app.service.rest.dao_{}_service".format(action))
|
|
sample_service.active = active
|
|
auth_header = create_admin_authorization_header()
|
|
response = client.post(
|
|
"/service/{}/{}".format(sample_service.id, action), headers=[auth_header]
|
|
)
|
|
assert response.status_code == 204
|
|
mocked.assert_not_called()
|
|
assert sample_service.active == active
|
|
|
|
|
|
@freeze_time("2001-01-01T23:59:00")
|
|
def test_suspending_service_revokes_api_keys(client, sample_service, sample_api_key):
|
|
auth_header = create_admin_authorization_header()
|
|
response = client.post(
|
|
"/service/{}/suspend".format(sample_service.id), headers=[auth_header]
|
|
)
|
|
assert response.status_code == 204
|
|
assert sample_api_key.expiry_date == datetime(2001, 1, 1, 23, 59, 00)
|
|
|
|
|
|
def test_resume_service_leaves_api_keys_revokes(client, sample_service, sample_api_key):
|
|
with freeze_time("2001-10-22T11:59:00"):
|
|
auth_header = create_admin_authorization_header()
|
|
client.post(
|
|
"/service/{}/suspend".format(sample_service.id), headers=[auth_header]
|
|
)
|
|
with freeze_time("2001-10-22T13:59:00"):
|
|
auth_header = create_admin_authorization_header()
|
|
response = client.post(
|
|
"/service/{}/resume".format(sample_service.id), headers=[auth_header]
|
|
)
|
|
assert response.status_code == 204
|
|
assert sample_api_key.expiry_date == datetime(2001, 10, 22, 11, 59, 00)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"action, original_state", [("suspend", True), ("resume", False)]
|
|
)
|
|
def test_service_history_is_created(client, sample_service, action, original_state):
|
|
sample_service.active = original_state
|
|
auth_header = create_admin_authorization_header()
|
|
response = client.post(
|
|
"/service/{}/{}".format(sample_service.id, action), headers=[auth_header]
|
|
)
|
|
ServiceHistory = Service.get_history_model()
|
|
history = (
|
|
ServiceHistory.query.filter_by(id=sample_service.id)
|
|
.order_by(ServiceHistory.version.desc())
|
|
.first()
|
|
)
|
|
|
|
assert response.status_code == 204
|
|
assert history.version == 2
|
|
assert history.active != original_state
|