mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-23 00:41:35 -05:00
66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
|
from flask import json
|
|||
|
|
from freezegun import freeze_time
|
|||
|
|
from tests import create_authorization_header
|
|||
|
|
from unittest.mock import ANY
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_broadcast_for_service_without_permission_returns_400(
|
|||
|
|
client,
|
|||
|
|
sample_service,
|
|||
|
|
):
|
|||
|
|
auth_header = create_authorization_header(service_id=sample_service.id)
|
|||
|
|
response = client.post(
|
|||
|
|
path='/v2/broadcast',
|
|||
|
|
data='',
|
|||
|
|
headers=[('Content-Type', 'application/json'), auth_header],
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
assert response.status_code == 400
|
|||
|
|
assert response.get_json()['errors'][0]['message'] == (
|
|||
|
|
'Service is not allowed to send broadcast messages'
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_valid_post_broadcast_returns_201(
|
|||
|
|
client,
|
|||
|
|
sample_broadcast_service,
|
|||
|
|
):
|
|||
|
|
auth_header = create_authorization_header(service_id=sample_broadcast_service.id)
|
|||
|
|
|
|||
|
|
response = client.post(
|
|||
|
|
path='/v2/broadcast',
|
|||
|
|
data=json.dumps({
|
|||
|
|
'content': 'This is a test',
|
|||
|
|
'reference': 'abc123',
|
|||
|
|
'polygons': [[
|
|||
|
|
[1, 2], [3, 4], [5, 6],
|
|||
|
|
]],
|
|||
|
|
}),
|
|||
|
|
headers=[('Content-Type', 'application/json'), auth_header],
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
assert response.status_code == 201
|
|||
|
|
|
|||
|
|
response_json = json.loads(response.get_data(as_text=True))
|
|||
|
|
|
|||
|
|
assert response_json['approved_at'] is None
|
|||
|
|
assert response_json['approved_by_id'] == None
|
|||
|
|
assert response_json['areas'] == []
|
|||
|
|
assert response_json['cancelled_at'] == None
|
|||
|
|
assert response_json['cancelled_by_id'] == None
|
|||
|
|
assert response_json['content'] == 'This is a test'
|
|||
|
|
assert response_json['reference'] == 'abc123'
|
|||
|
|
assert response_json['created_at'] # datetime generated by the DB so can’t freeze it
|
|||
|
|
assert response_json['created_by_id'] == None
|
|||
|
|
assert response_json['finishes_at'] is None
|
|||
|
|
assert response_json['id'] == ANY
|
|||
|
|
assert response_json['personalisation'] is None
|
|||
|
|
assert response_json['service_id'] == str(sample_broadcast_service.id)
|
|||
|
|
assert response_json['simple_polygons'] == [[[1, 2], [3, 4], [5, 6]]]
|
|||
|
|
assert response_json['starts_at'] is None
|
|||
|
|
assert response_json['status'] == 'pending-approval'
|
|||
|
|
assert response_json['template_id'] is None
|
|||
|
|
assert response_json['template_name'] is None
|
|||
|
|
assert response_json['template_version'] is None
|
|||
|
|
assert response_json['updated_at'] is None
|