darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-05-06 00:48:46 -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

107
tests/app/notify_client/test_template_folder_client.py
View File

@@ -7,41 +7,45 @@ from orderedset import OrderedSet
from app.notify_client.template_folder_api_client import TemplateFolderAPIClient
@pytest.mark.parametrize('parent_id', [uuid.uuid4(), None])
@pytest.mark.parametrize("parent_id", [uuid.uuid4(), None])
def test_create_template_folder_calls_correct_api_endpoint(mocker, parent_id):
mock_redis_delete = mocker.patch('app.extensions.RedisClient.delete')
mock_redis_delete = mocker.patch("app.extensions.RedisClient.delete")
some_service_id = uuid.uuid4()
expected_url = '/service/{}/template-folder'.format(some_service_id)
data = {'name': 'foo', 'parent_id': parent_id}
expected_url = "/service/{}/template-folder".format(some_service_id)
data = {"name": "foo", "parent_id": parent_id}
client = TemplateFolderAPIClient()
mock_post = mocker.patch('app.notify_client.template_folder_api_client.TemplateFolderAPIClient.post')
mock_post = mocker.patch(
"app.notify_client.template_folder_api_client.TemplateFolderAPIClient.post"
)
client.create_template_folder(some_service_id, name='foo', parent_id=parent_id)
client.create_template_folder(some_service_id, name="foo", parent_id=parent_id)
mock_post.assert_called_once_with(expected_url, data)
mock_redis_delete.assert_called_once_with('service-{}-template-folders'.format(some_service_id))
mock_redis_delete.assert_called_once_with(
"service-{}-template-folders".format(some_service_id)
)
def test_get_template_folders_calls_correct_api_endpoint(mocker):
mock_redis_get = mocker.patch('app.extensions.RedisClient.get', return_value=None)
mock_redis_set = mocker.patch('app.extensions.RedisClient.set')
mock_redis_get = mocker.patch("app.extensions.RedisClient.get", return_value=None)
mock_redis_set = mocker.patch("app.extensions.RedisClient.set")
mock_api_get = mocker.patch(
'app.notify_client.NotifyAdminAPIClient.get',
return_value={'template_folders': {'a': 'b'}}
"app.notify_client.NotifyAdminAPIClient.get",
return_value={"template_folders": {"a": "b"}},
)
some_service_id = uuid.uuid4()
expected_url = '/service/{}/template-folder'.format(some_service_id)
redis_key = 'service-{}-template-folders'.format(some_service_id)
expected_url = "/service/{}/template-folder".format(some_service_id)
redis_key = "service-{}-template-folders".format(some_service_id)
client = TemplateFolderAPIClient()
ret = client.get_template_folders(some_service_id)
assert ret == {'a': 'b'}
assert ret == {"a": "b"}
mock_redis_get.assert_called_once_with(redis_key)
mock_api_get.assert_called_once_with(expected_url)
@@ -49,9 +53,8 @@ def test_get_template_folders_calls_correct_api_endpoint(mocker):
def test_move_templates_and_folders(mocker):
mock_redis_delete = mocker.patch('app.extensions.RedisClient.delete')
mock_api_post = mocker.patch('app.notify_client.NotifyAdminAPIClient.post')
mock_redis_delete = mocker.patch("app.extensions.RedisClient.delete")
mock_api_post = mocker.patch("app.notify_client.NotifyAdminAPIClient.post")
some_service_id = uuid.uuid4()
some_folder_id = uuid.uuid4()
@@ -59,82 +62,98 @@ def test_move_templates_and_folders(mocker):
TemplateFolderAPIClient().move_to_folder(
some_service_id,
some_folder_id,
template_ids=OrderedSet(('a', 'b', 'c')),
folder_ids=OrderedSet(('1', '2', '3')),
template_ids=OrderedSet(("a", "b", "c")),
folder_ids=OrderedSet(("1", "2", "3")),
)
mock_api_post.assert_called_once_with(
'/service/{}/template-folder/{}/contents'.format(
"/service/{}/template-folder/{}/contents".format(
some_service_id, some_folder_id
),
{
'folders': ['1', '2', '3'],
'templates': ['a', 'b', 'c'],
"folders": ["1", "2", "3"],
"templates": ["a", "b", "c"],
},
)
assert mock_redis_delete.call_args_list == [
call(
f'service-{some_service_id}-template-a-version-None',
f'service-{some_service_id}-template-b-version-None',
f'service-{some_service_id}-template-c-version-None',
f"service-{some_service_id}-template-a-version-None",
f"service-{some_service_id}-template-b-version-None",
f"service-{some_service_id}-template-c-version-None",
),
call('service-{}-templates'.format(some_service_id)),
call('service-{}-template-folders'.format(some_service_id)),
call("service-{}-templates".format(some_service_id)),
call("service-{}-template-folders".format(some_service_id)),
]
def test_move_templates_and_folders_to_root(mocker):
mock_api_post = mocker.patch('app.notify_client.NotifyAdminAPIClient.post')
mock_api_post = mocker.patch("app.notify_client.NotifyAdminAPIClient.post")
some_service_id = uuid.uuid4()
TemplateFolderAPIClient().move_to_folder(
some_service_id,
None,
template_ids=OrderedSet(('a', 'b', 'c')),
folder_ids=OrderedSet(('1', '2', '3')),
template_ids=OrderedSet(("a", "b", "c")),
folder_ids=OrderedSet(("1", "2", "3")),
)
mock_api_post.assert_called_once_with(
'/service/{}/template-folder/contents'.format(some_service_id),
"/service/{}/template-folder/contents".format(some_service_id),
{
'folders': ['1', '2', '3'],
'templates': ['a', 'b', 'c'],
"folders": ["1", "2", "3"],
"templates": ["a", "b", "c"],
},
)
def test_update_template_folder_calls_correct_api_endpoint(mocker):
mock_redis_delete = mocker.patch('app.extensions.RedisClient.delete')
mock_redis_delete = mocker.patch("app.extensions.RedisClient.delete")
some_service_id = uuid.uuid4()
template_folder_id = uuid.uuid4()
expected_url = '/service/{}/template-folder/{}'.format(some_service_id, template_folder_id)
data = {'name': 'foo', 'users_with_permission': ['some_id']}
expected_url = "/service/{}/template-folder/{}".format(
some_service_id, template_folder_id
)
data = {"name": "foo", "users_with_permission": ["some_id"]}
client = TemplateFolderAPIClient()
mock_post = mocker.patch('app.notify_client.template_folder_api_client.TemplateFolderAPIClient.post')
mock_post = mocker.patch(
"app.notify_client.template_folder_api_client.TemplateFolderAPIClient.post"
)
client.update_template_folder(some_service_id, template_folder_id, name='foo', users_with_permission=['some_id'])
client.update_template_folder(
some_service_id,
template_folder_id,
name="foo",
users_with_permission=["some_id"],
)
mock_post.assert_called_once_with(expected_url, data)
mock_redis_delete.assert_called_once_with('service-{}-template-folders'.format(some_service_id))
mock_redis_delete.assert_called_once_with(
"service-{}-template-folders".format(some_service_id)
)
def test_delete_template_folder_calls_correct_api_endpoint(mocker):
mock_redis_delete = mocker.patch('app.extensions.RedisClient.delete')
mock_redis_delete = mocker.patch("app.extensions.RedisClient.delete")
some_service_id = uuid.uuid4()
template_folder_id = uuid.uuid4()
expected_url = '/service/{}/template-folder/{}'.format(some_service_id, template_folder_id)
expected_url = "/service/{}/template-folder/{}".format(
some_service_id, template_folder_id
)
client = TemplateFolderAPIClient()
mock_delete = mocker.patch('app.notify_client.template_folder_api_client.TemplateFolderAPIClient.delete')
mock_delete = mocker.patch(
"app.notify_client.template_folder_api_client.TemplateFolderAPIClient.delete"
)
client.delete_template_folder(some_service_id, template_folder_id)
mock_delete.assert_called_once_with(expected_url, {})
mock_redis_delete.assert_called_once_with('service-{}-template-folders'.format(some_service_id))
mock_redis_delete.assert_called_once_with(
"service-{}-template-folders".format(some_service_id)
)