mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-09-09 17:45:32 -04:00
Add tests for existing status api client
Essentially copies the tests found in the performance_platform api client.
This commit is contained in:
59
tests/app/notify_client/test_status_api_client.py
Normal file
59
tests/app/notify_client/test_status_api_client.py
Normal file
@@ -0,0 +1,59 @@
|
|||||||
|
from app.notify_client.status_api_client import StatusApiClient
|
||||||
|
|
||||||
|
|
||||||
|
def test_get_count_of_live_services_and_organisations(mocker):
|
||||||
|
mocker.patch('app.extensions.RedisClient.get', return_value=None)
|
||||||
|
client = StatusApiClient()
|
||||||
|
mock = mocker.patch.object(client, 'get', return_value={})
|
||||||
|
|
||||||
|
client.get_count_of_live_services_and_organisations()
|
||||||
|
|
||||||
|
mock.assert_called_once_with(url='/_status/live-service-and-organisation-counts')
|
||||||
|
|
||||||
|
|
||||||
|
def test_sets_value_in_cache(mocker):
|
||||||
|
client = StatusApiClient()
|
||||||
|
|
||||||
|
mock_redis_get = mocker.patch(
|
||||||
|
'app.extensions.RedisClient.get',
|
||||||
|
return_value=None
|
||||||
|
)
|
||||||
|
mock_api_get = mocker.patch(
|
||||||
|
'app.notify_client.NotifyAdminAPIClient.get',
|
||||||
|
return_value={'data_from': 'api'},
|
||||||
|
)
|
||||||
|
mock_redis_set = mocker.patch(
|
||||||
|
'app.extensions.RedisClient.set',
|
||||||
|
)
|
||||||
|
|
||||||
|
assert client.get_count_of_live_services_and_organisations() == {'data_from': 'api'}
|
||||||
|
|
||||||
|
mock_redis_get.assert_called_once_with('live-service-and-organisation-counts')
|
||||||
|
mock_api_get.assert_called_once_with(url='/_status/live-service-and-organisation-counts')
|
||||||
|
mock_redis_set.assert_called_once_with(
|
||||||
|
'live-service-and-organisation-counts',
|
||||||
|
'{"data_from": "api"}',
|
||||||
|
ex=604800
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def test_returns_value_from_cache(mocker):
|
||||||
|
client = StatusApiClient()
|
||||||
|
|
||||||
|
mock_redis_get = mocker.patch(
|
||||||
|
'app.extensions.RedisClient.get',
|
||||||
|
return_value=b'{"data_from": "cache"}',
|
||||||
|
)
|
||||||
|
mock_api_get = mocker.patch(
|
||||||
|
'app.notify_client.NotifyAdminAPIClient.get',
|
||||||
|
)
|
||||||
|
mock_redis_set = mocker.patch(
|
||||||
|
'app.extensions.RedisClient.set',
|
||||||
|
)
|
||||||
|
|
||||||
|
assert client.get_count_of_live_services_and_organisations() == {'data_from': 'cache'}
|
||||||
|
|
||||||
|
mock_redis_get.assert_called_once_with('live-service-and-organisation-counts')
|
||||||
|
|
||||||
|
assert mock_api_get.called is False
|
||||||
|
assert mock_redis_set.called is False
|
||||||
Reference in New Issue
Block a user