darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-07-30 19:29:43 -04:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #3546 from alphagov/redis-cache-broadcast-message

Browse Source 
Cache broadcast messages in Redis
This commit is contained in:
Chris Hill-Scott
2020-08-04 16:17:21 +01:00
committed by GitHub
parent a74501f1d8 058553b6ee
commit b31fb7de66
2 changed files with 96 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

92
tests/app/notify_client/test_broadcast_message_client.py Normal file
View File

@@ -0,0 +1,92 @@
from app.notify_client.broadcast_message_api_client import (
BroadcastMessageAPIClient,
)
def test_create_broadcast_message(mocker):
client = BroadcastMessageAPIClient()
mocker.patch('app.notify_client.current_user', id='1')
mock_post = mocker.patch(
'app.notify_client.broadcast_message_api_client.BroadcastMessageAPIClient.post'
)
client.create_broadcast_message(
service_id='12345',
template_id='67890',
)
mock_post.assert_called_once_with(
'/service/12345/broadcast-message',
data={
'service_id': '12345',
'template_id': '67890',
'personalisation': {},
'created_by': '1',
},
)
def test_get_broadcast_messages(mocker):
client = BroadcastMessageAPIClient()
mock_get = mocker.patch(
'app.notify_client.broadcast_message_api_client.BroadcastMessageAPIClient.get'
)
client.get_broadcast_messages('12345')
mock_get.assert_called_once_with(
'/service/12345/broadcast-message',
)
def test_get_broadcast_message(mocker):
client = BroadcastMessageAPIClient()
mocker.patch('app.notify_client.current_user', id='1')
mock_get = mocker.patch(
'app.notify_client.broadcast_message_api_client.BroadcastMessageAPIClient.get',
return_value={'abc': 'def'},
)
mock_redis_set = mocker.patch('app.extensions.RedisClient.set')
client.get_broadcast_message(service_id='12345', broadcast_message_id='67890')
mock_get.assert_called_once_with(
'/service/12345/broadcast-message/67890',
)
mock_redis_set.assert_called_once_with(
'broadcast-message-67890',
'{"abc": "def"}',
ex=604_800,
)
def test_update_broadcast_message(mocker):
client = BroadcastMessageAPIClient()
mocker.patch('app.notify_client.current_user', id='1')
mock_post = mocker.patch(
'app.notify_client.broadcast_message_api_client.BroadcastMessageAPIClient.post'
)
mock_redis_delete = mocker.patch('app.extensions.RedisClient.delete')
client.update_broadcast_message(
service_id='12345',
broadcast_message_id='67890',
data={'abc': 'def'},
)
mock_post.assert_called_once_with(
'/service/12345/broadcast-message/67890',
data={'abc': 'def'},
)
mock_redis_delete.assert_called_once_with('broadcast-message-67890')
def test_update_broadcast_message_status(mocker):
client = BroadcastMessageAPIClient()
mocker.patch('app.notify_client.current_user', id='1')
mock_post = mocker.patch(
'app.notify_client.broadcast_message_api_client.BroadcastMessageAPIClient.post'
)
mock_redis_delete = mocker.patch('app.extensions.RedisClient.delete')
client.update_broadcast_message_status(
'cancelled',
service_id='12345',
broadcast_message_id='67890',
)
mock_post.assert_called_once_with(
'/service/12345/broadcast-message/67890/status',
data={'created_by': '1', 'status': 'cancelled'},
)
mock_redis_delete.assert_called_once_with('broadcast-message-67890')