darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-08-14 23:07:49 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add job to queue as soon as it in created.

Browse Source 
Added status to job.
This commit is contained in:
Adam Shimali
2016-02-02 14:58:25 +00:00
parent b5c662eca8
commit 0ade39e63f
5 changed files with 65 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tests/app/job/test_job_rest.py
View File

@@ -1,3 +1,5 @@
import boto3
import moto
import json
import uuid
from flask import url_for
@@ -73,7 +75,8 @@ def test_get_job_by_id(notify_api, notify_db, notify_db_session,
assert resp_json['data']['id'] == job_id
def test_post_job(notify_api, notify_db, notify_db_session, sample_template):
@moto.mock_sqs
def test_create_job(notify_api, notify_db, notify_db_session, sample_template):
job_id = uuid.uuid4()
template_id = sample_template.id
service_id = sample_template.service.id
@@ -88,6 +91,7 @@ def test_post_job(notify_api, notify_db, notify_db_session, sample_template):
'bucket_name': bucket_name,
'file_name': file_name,
}
with notify_api.test_request_context():
with notify_api.test_client() as client:
path = url_for('job.create_job', service_id=service_id)
@@ -109,6 +113,15 @@ def test_post_job(notify_api, notify_db, notify_db_session, sample_template):
assert resp_json['data']['template'] == template_id
assert resp_json['data']['original_file_name'] == original_file_name
boto3.setup_default_session(region_name='eu-west-1')
q = boto3.resource('sqs').get_queue_by_name(QueueName='notify-jobs-queue')
messages = q.receive_messages()
assert len(messages) == 1
expected_message = json.loads(messages[0].body)
assert expected_message['job_id'] == str(job_id)
assert expected_message['service_id'] == str(service_id)
def _setup_jobs(notify_db, notify_db_session, template, number_of_jobs=5):
for i in range(number_of_jobs):