2017-04-06 12:14:17 +01:00
|
|
|
import uuid
|
|
|
|
|
|
2017-04-06 12:18:34 +01:00
|
|
|
from flask import current_app
|
2017-04-06 12:14:17 +01:00
|
|
|
from flask import json
|
|
|
|
|
|
|
|
|
|
from tests import create_authorization_header
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_send_letter_jobs(client, mocker):
|
|
|
|
|
mock_celery = mocker.patch("app.letters.send_letter_jobs.notify_celery.send_task")
|
|
|
|
|
job_ids = {"job_ids": [str(uuid.uuid4()), str(uuid.uuid4()), str(uuid.uuid4())]}
|
|
|
|
|
|
|
|
|
|
auth_header = create_authorization_header()
|
|
|
|
|
|
|
|
|
|
response = client.post(
|
|
|
|
|
path='/send-letter-jobs',
|
|
|
|
|
data=json.dumps(job_ids),
|
|
|
|
|
headers=[('Content-Type', 'application/json'), auth_header])
|
|
|
|
|
|
|
|
|
|
assert response.status_code == 200
|
|
|
|
|
assert response.get_data(as_text=True) == "Task created to send files to DVLA"
|
|
|
|
|
|
2017-04-06 15:26:50 +01:00
|
|
|
mock_celery.assert_called_once_with(name="send-files-to-dvla",
|
2017-04-06 15:03:06 +01:00
|
|
|
args=(job_ids['job_ids'],),
|
2017-04-06 12:18:34 +01:00
|
|
|
queue="process-ftp")
|
2017-04-06 12:14:17 +01:00
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_send_letter_jobs_throws_validation_error(client, mocker):
|
|
|
|
|
mock_celery = mocker.patch("app.letters.send_letter_jobs.notify_celery.send_task")
|
|
|
|
|
|
|
|
|
|
job_ids = {"job_ids": ["1", "2"]}
|
|
|
|
|
|
|
|
|
|
auth_header = create_authorization_header()
|
|
|
|
|
|
|
|
|
|
response = client.post(
|
|
|
|
|
path='/send-letter-jobs',
|
|
|
|
|
data=json.dumps(job_ids),
|
|
|
|
|
headers=[('Content-Type', 'application/json'), auth_header])
|
|
|
|
|
|
|
|
|
|
assert response.status_code == 400
|
|
|
|
|
|
|
|
|
|
assert not mock_celery.called
|
