darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-09-07 06:18:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

Update send letter jobs to return json

Browse Source
This commit is contained in:
Ken Tsang
2017-04-11 15:39:34 +01:00
parent 1e510487e2
commit 155e173363
2 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
app/letters/send_letter_jobs.py
View File

@@ -1,3 +1,5 @@
import json
from flask import Blueprint, jsonify from flask import Blueprint, jsonify
from flask import request from flask import request
@@ -14,10 +16,12 @@ register_errors(letter_job)
@letter_job.route('/send-letter-jobs', methods=['POST']) @letter_job.route('/send-letter-jobs', methods=['POST'])
def send_letter_jobs(): def send_letter_jobs():
job_ids = validate(request.get_json(), letter_job_ids) req_json = request.get_json()
job_ids = validate(req_json, letter_job_ids)
notify_celery.send_task(name="send-files-to-dvla", args=(job_ids['job_ids'],), queue="process-ftp") notify_celery.send_task(name="send-files-to-dvla", args=(job_ids['job_ids'],), queue="process-ftp")
return "Task created to send files to DVLA" return jsonify(data={"response": "Task created to send files to DVLA"}), 200
@letter_job.route('/letter-jobs', methods=['GET']) @letter_job.route('/letter-jobs', methods=['GET'])

2
tests/app/letters/test_send_letter_jobs.py
View File

@@ -17,7 +17,7 @@ def test_send_letter_jobs(client, mocker):
headers=[('Content-Type', 'application/json'), auth_header]) headers=[('Content-Type', 'application/json'), auth_header])
assert response.status_code == 200 assert response.status_code == 200
assert response.get_data(as_text=True) == "Task created to send files to DVLA" assert json.loads(response.get_data())['data'] == {'response': "Task created to send files to DVLA"}
mock_celery.assert_called_once_with(name="send-files-to-dvla", mock_celery.assert_called_once_with(name="send-files-to-dvla",
args=(job_ids['job_ids'],), args=(job_ids['job_ids'],),