From 09731c109e583387303374fd93b5c48be7a9d652 Mon Sep 17 00:00:00 2001 From: Pea Tyczynska Date: Tue, 19 Mar 2019 11:48:37 +0000 Subject: [PATCH] Test update_template_folder can manage multiple users at once --- .../test_template_folder_rest.py | 24 +++++++++++++++---- 1 file changed, 20 insertions(+), 4 deletions(-) diff --git a/tests/app/template_folder/test_template_folder_rest.py b/tests/app/template_folder/test_template_folder_rest.py index cd8d85f91..ab838842e 100644 --- a/tests/app/template_folder/test_template_folder_rest.py +++ b/tests/app/template_folder/test_template_folder_rest.py @@ -173,21 +173,37 @@ def test_update_template_folder_name(admin_request, sample_service): def test_update_template_folder_users(admin_request, sample_service): existing_folder = create_template_folder(sample_service) - user = create_user() + user_1 = create_user(email="notify_1@digital.cabinet-office.gov.uk") + user_2 = create_user(email="notify_2@digital.cabinet-office.gov.uk") + user_3 = create_user(email="notify_3@digital.cabinet-office.gov.uk") + sample_service.users += [user_1, user_2, user_3] assert len(existing_folder.users) == 0 - resp = admin_request.post( + response_1 = admin_request.post( 'template_folder.update_template_folder', service_id=sample_service.id, template_folder_id=existing_folder.id, _data={ 'name': 'foo', - 'users_with_permission': [str(user.id)] + 'users_with_permission': [str(user_1.id)] } ) - assert resp['data']['users_with_permission'] == [str(user.id)] + assert response_1['data']['users_with_permission'] == [str(user_1.id)] assert len(existing_folder.users) == 1 + response_2 = admin_request.post( + 'template_folder.update_template_folder', + service_id=sample_service.id, + template_folder_id=existing_folder.id, + _data={ + 'name': 'foo', + 'users_with_permission': [str(user_2.id), str(user_3.id)] + } + ) + + assert response_2['data']['users_with_permission'] == [str(user_2.id), str(user_3.id)] + assert len(existing_folder.users) == 2 + @pytest.mark.parametrize('data, err', [ ({}, 'name is a required property'),