darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-08-19 14:09:20 -04:00
Code Issues Packages Projects Releases Wiki Activity

notify-api-412 use black to enforce python coding style

Browse Source
This commit is contained in:
Kenneth Kehl
2023-08-25 09:12:23 -07:00
parent c6eb007386
commit 8c9721d8e2
201 changed files with 31660 additions and 28105 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
tests/app/notify_client/test_status_api_client.py
View File

@@ -2,38 +2,35 @@ from app.notify_client.status_api_client import StatusApiClient
def test_get_count_of_live_services_and_organizations(mocker):
mocker.patch('app.extensions.RedisClient.get', return_value=None)
mocker.patch("app.extensions.RedisClient.get", return_value=None)
client = StatusApiClient()
mock = mocker.patch.object(client, 'get', return_value={})
mock = mocker.patch.object(client, "get", return_value={})
client.get_count_of_live_services_and_organizations()
mock.assert_called_once_with(url='/_status/live-service-and-organization-counts')
mock.assert_called_once_with(url="/_status/live-service-and-organization-counts")
def test_sets_value_in_cache(mocker):
client = StatusApiClient()
mock_redis_get = mocker.patch(
'app.extensions.RedisClient.get',
return_value=None
)
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'},
"app.notify_client.NotifyAdminAPIClient.get",
return_value={"data_from": "api"},
)
mock_redis_set = mocker.patch(
'app.extensions.RedisClient.set',
"app.extensions.RedisClient.set",
)
assert client.get_count_of_live_services_and_organizations() == {'data_from': 'api'}
assert client.get_count_of_live_services_and_organizations() == {"data_from": "api"}
mock_redis_get.assert_called_once_with('live-service-and-organization-counts')
mock_api_get.assert_called_once_with(url='/_status/live-service-and-organization-counts')
mock_redis_get.assert_called_once_with("live-service-and-organization-counts")
mock_api_get.assert_called_once_with(
url="/_status/live-service-and-organization-counts"
)
mock_redis_set.assert_called_once_with(
'live-service-and-organization-counts',
'{"data_from": "api"}',
ex=3600
"live-service-and-organization-counts", '{"data_from": "api"}', ex=3600
)
@@ -41,19 +38,21 @@ def test_returns_value_from_cache(mocker):
client = StatusApiClient()
mock_redis_get = mocker.patch(
'app.extensions.RedisClient.get',
"app.extensions.RedisClient.get",
return_value=b'{"data_from": "cache"}',
)
mock_api_get = mocker.patch(
'app.notify_client.NotifyAdminAPIClient.get',
"app.notify_client.NotifyAdminAPIClient.get",
)
mock_redis_set = mocker.patch(
'app.extensions.RedisClient.set',
"app.extensions.RedisClient.set",
)
assert client.get_count_of_live_services_and_organizations() == {'data_from': 'cache'}
assert client.get_count_of_live_services_and_organizations() == {
"data_from": "cache"
}
mock_redis_get.assert_called_once_with('live-service-and-organization-counts')
mock_redis_get.assert_called_once_with("live-service-and-organization-counts")
assert mock_api_get.called is False
assert mock_redis_set.called is False