Setup celery config

app/notifications/rest.py

@@ -49,7 +49,8 @@ def create_sms_notification():
     send_sms.apply_async((
         api_user['client'],
         notification_id,
-        encryption.encrypt(notification)))
+        encryption.encrypt(notification)),
+        queue='sms')
     return jsonify({'notification_id': notification_id}), 201
 
 

config.py

@@ -1,5 +1,4 @@
 import os
-from datetime import timedelta
 
 
 class Config(object):
@@ -26,7 +25,7 @@ class Config(object):
         'region': 'eu-west-1',
         'polling_interval': 10,  # 1 second
         'visibility_timeout': 3600,  # 1 hour
-        'queue_name_prefix': 'NOTIFY-CELERY-TEST-'
+        'queue_name_prefix': 'preview'
     }
     CELERY_ENABLE_UTC = True,
     CELERY_TIMEZONE = 'Europe/London'

run_celery.py

@@ -1,6 +1,4 @@
 #!/usr/bin/env python
-import os
 from app import notify_celery, create_app
-
-application = create_app(os.getenv('NOTIFY_API_ENVIRONMENT') or 'development')
+application = create_app()
 application.app_context().push()

scripts/run_celery.sh

@@ -0,0 +1,6 @@
+#!/bin/bash
+
+set -e
+
+source environment.sh
+celery -A run_celery.notify_celery worker --loglevel=INFO --concurrency=1 --autoreload -Q sms

tests/app/notifications/test_rest.py

@@ -30,7 +30,6 @@ def test_get_notification_by_id(notify_api, sample_notification):
 def test_get_notifications_empty_result(notify_api, sample_api_key):
     with notify_api.test_request_context():
         with notify_api.test_client() as client:
-
             missing_notification_id = uuid.uuid4()
             auth_header = create_authorization_header(
                 service_id=sample_api_key.service_id,
@@ -222,7 +221,8 @@ def test_should_allow_valid_sms_notification(notify_api, sample_template, mocker
             app.celery.tasks.send_sms.apply_async.assert_called_once_with(
                 (str(sample_template.service_id),
                  notification_id,
-                 "something_encrypted")
+                 "something_encrypted"),
+                queue="sms"
             )
             assert response.status_code == 201
             assert notification_id