diff --git a/app/main/forms.py b/app/main/forms.py index 6f7833184..e2efb9cfa 100644 --- a/app/main/forms.py +++ b/app/main/forms.py @@ -731,6 +731,7 @@ class ServiceUpdateEmailBranding(StripWhitespaceForm): name = StringField('Name of brand') text = StringField('Text') + domain = StringField('Domain') colour = StringField( 'Colour', render_kw={'onchange': 'update_colour(this)'}, @@ -738,6 +739,20 @@ class ServiceUpdateEmailBranding(StripWhitespaceForm): Regexp(regex="^$|^#(?:[0-9a-fA-F]{3}){1,2}$", message='Must be a valid color hex code') ] ) + banner_colour = StringField( + 'Banner colour', + render_kw={'onchange': 'update_colour(this)'}, + validators=[ + Regexp(regex="^$|^#(?:[0-9a-fA-F]{3}){1,2}$", message='Must be a valid color hex code') + ] + ) + single_id_colour = StringField( + 'Single identity colour', + render_kw={'onchange': 'update_colour(this)'}, + validators=[ + Regexp(regex="^$|^#(?:[0-9a-fA-F]{3}){1,2}$", message='Must be a valid color hex code') + ] + ) file = FileField_wtf('Upload a PNG logo', validators=[FileAllowed(['png'], 'PNG Images only!')]) @@ -745,6 +760,7 @@ class ServiceCreateEmailBranding(StripWhitespaceForm): name = StringField('Name of brand') text = StringField('Text') + domain = StringField('Domain') colour = StringField( 'Colour', render_kw={'onchange': 'update_colour(this)'}, @@ -752,6 +768,20 @@ class ServiceCreateEmailBranding(StripWhitespaceForm): Regexp(regex="^$|^#(?:[0-9a-fA-F]{3}){1,2}$", message='Must be a valid color hex code') ] ) + banner_colour = StringField( + 'Banner colour', + render_kw={'onchange': 'update_colour(this)'}, + validators=[ + Regexp(regex="^$|^#(?:[0-9a-fA-F]{3}){1,2}$", message='Must be a valid color hex code') + ] + ) + single_id_colour = StringField( + 'Single identity colour', + render_kw={'onchange': 'update_colour(this)'}, + validators=[ + Regexp(regex="^$|^#(?:[0-9a-fA-F]{3}){1,2}$", message='Must be a valid color hex code') + ] + ) file = FileField_wtf('Upload a PNG logo', validators=[FileAllowed(['png'], 'PNG Images only!')]) diff --git a/app/main/views/email_branding.py b/app/main/views/email_branding.py index 4f55b1e7f..008b94052 100644 --- a/app/main/views/email_branding.py +++ b/app/main/views/email_branding.py @@ -80,7 +80,10 @@ def update_email_branding(branding_id, logo=None): logo=logo, name=form.name.data, text=form.text.data, - colour=form.colour.data + colour=form.colour.data, + banner_colour=form.banner_colour.data, + single_id_colour=form.single_id_colour.data, + domain=form.domain.data, ) return redirect(url_for('.email_branding', branding_id=branding_id)) @@ -88,6 +91,9 @@ def update_email_branding(branding_id, logo=None): form.name.data = email_branding['name'] form.text.data = email_branding['text'] form.colour.data = email_branding['colour'] + form.banner_colour.data = email_branding['banner_colour'] + form.single_id_colour.data = email_branding['single_id_colour'] + form.domain.data = email_branding['domain'] return render_template( 'views/email-branding/manage-branding.html', @@ -128,7 +134,10 @@ def create_email_branding(logo=None): logo=logo, name=form.name.data, text=form.text.data, - colour=form.colour.data + colour=form.colour.data, + banner_colour=form.banner_colour.data, + single_id_colour=form.single_id_colour.data, + domain=form.domain.data ) return redirect(url_for('.email_branding')) diff --git a/app/notify_client/email_branding_client.py b/app/notify_client/email_branding_client.py index ae2375144..0349c3f84 100644 --- a/app/notify_client/email_branding_client.py +++ b/app/notify_client/email_branding_client.py @@ -18,20 +18,26 @@ class EmailBrandingClient(NotifyAdminAPIClient): def get_letter_email_branding(self): return self.get(url='/dvla_organisations') - def create_email_branding(self, logo, name, text, colour): + def create_email_branding(self, logo, name, text, colour, banner_colour, single_id_colour, domain): data = { "logo": logo, "name": name, "text": text, - "colour": colour + "colour": colour, + "banner_colour": banner_colour, + "single_id_colour": single_id_colour, + "domain": domain, } return self.post(url="/email-branding", data=data) - def update_email_branding(self, branding_id, logo, name, text, colour): + def update_email_branding(self, branding_id, logo, name, text, colour, banner_colour, single_id_colour, domain): data = { "logo": logo, "name": name, "text": text, - "colour": colour + "colour": colour, + "banner_colour": banner_colour, + "single_id_colour": single_id_colour, + "domain": domain, } return self.post(url="/email-branding/{}".format(branding_id), data=data) diff --git a/app/templates/views/email-branding/manage-branding.html b/app/templates/views/email-branding/manage-branding.html index d8927ad65..8b628a9a4 100644 --- a/app/templates/views/email-branding/manage-branding.html +++ b/app/templates/views/email-branding/manage-branding.html @@ -12,7 +12,6 @@

{{ '{} email branding'.format('Update' if email_branding else 'Create')}}

- {% if logo %}
@@ -24,6 +23,9 @@
{{textbox(form.name)}}
{{textbox(form.text)}}
{{colour_textbox(form.colour, width='1-4', colour=email_branding.colour if email_branding)}} + {{colour_textbox(form.banner_colour, width='1-4', colour=email_branding.banner_colour if email_branding)}} + {{colour_textbox(form.single_id_colour, width='1-4', colour=email_branding.single_id_colour if email_branding)}} +
{{textbox(form.domain)}}
{{ page_footer( 'Save', back_link=url_for('.email_branding'), diff --git a/tests/app/main/views/test_email_branding.py b/tests/app/main/views/test_email_branding.py index a5b88bc80..9bef00beb 100644 --- a/tests/app/main/views/test_email_branding.py +++ b/tests/app/main/views/test_email_branding.py @@ -51,6 +51,9 @@ def test_edit_email_branding_shows_the_correct_branding_info( assert page.select_one('#name').attrs.get('value') == 'Organisation name' assert page.select_one('#text').attrs.get('value') == 'Organisation text' assert page.select_one('#colour').attrs.get('value') == '#f00' + assert page.select_one('#banner_colour').attrs.get('value') == '#f11' + assert page.select_one('#single_id_colour').attrs.get('value') == '#f22' + assert page.select_one('#domain').attrs.get('value') == 'sample.com' def test_create_email_branding_does_not_show_any_branding_info( @@ -69,6 +72,9 @@ def test_create_email_branding_does_not_show_any_branding_info( assert page.select_one('#name').attrs.get('value') == '' assert page.select_one('#text').attrs.get('value') == '' assert page.select_one('#colour').attrs.get('value') == '' + assert page.select_one('#banner_colour').attrs.get('value') == '' + assert page.select_one('#single_id_colour').attrs.get('value') == '' + assert page.select_one('#domain').attrs.get('value') == '' def test_create_new_email_branding_without_logo( @@ -82,6 +88,9 @@ def test_create_new_email_branding_without_logo( 'colour': '#ff0000', 'text': 'new text', 'name': 'new name', + 'domain': 'sample.com', + 'banner_colour': '#FFFF00', + 'single_id_colour': '#00FF00', } mock_persist = mocker.patch('app.main.views.email_branding.persist_logo') @@ -98,7 +107,10 @@ def test_create_new_email_branding_without_logo( logo=data['logo'], name=data['name'], text=data['text'], - colour=data['colour'] + colour=data['colour'], + banner_colour=data['banner_colour'], + single_id_colour=data['single_id_colour'], + domain=data['domain'] ) assert mock_persist.call_args_list == [] @@ -117,6 +129,9 @@ def test_create_new_email_branding_when_branding_saved( 'colour': '#ff0000', 'text': 'new text', 'name': 'new name', + 'domain': 'sample.com', + 'banner_colour': '#FFFF00', + 'single_id_colour': '#00FF00', } temp_filename = LOGO_LOCATION_STRUCTURE.format( @@ -135,7 +150,10 @@ def test_create_new_email_branding_when_branding_saved( 'colour': data['colour'], 'name': data['name'], 'text': data['text'], - 'cdn_url': 'https://static-logos.cdn.com' + 'cdn_url': 'https://static-logos.cdn.com', + 'domain': data['domain'], + 'banner_colour': data['banner_colour'], + 'single_id_colour': data['single_id_colour'], } ) @@ -144,7 +162,10 @@ def test_create_new_email_branding_when_branding_saved( logo=data['logo'], name=data['name'], text=data['text'], - colour=data['colour'] + colour=data['colour'], + banner_colour=data['banner_colour'], + single_id_colour=data['single_id_colour'], + domain=data['domain'] ) @@ -195,7 +216,7 @@ def test_deletes_previous_temp_logo_after_uploading_logo( assert mocked_delete_temp_file.call_args == call(temp_old_filename) -def test_update_exisiting_branding( +def test_update_existing_branding( logged_in_platform_admin_client, mocker, fake_uuid, @@ -209,7 +230,10 @@ def test_update_exisiting_branding( 'logo': 'test.png', 'colour': '#0000ff', 'text': 'new text', - 'name': 'new name' + 'name': 'new name', + 'banner_colour': '#FFFF00', + 'single_id_colour': '#00FF00', + 'domain': 'sample.com', } temp_filename = LOGO_LOCATION_STRUCTURE.format( @@ -225,7 +249,10 @@ def test_update_exisiting_branding( url_for('.update_email_branding', logo=temp_filename, branding_id=fake_uuid), content_type='multipart/form-data', data={'colour': data['colour'], 'name': data['name'], 'text': data['text'], - 'cdn_url': 'https://static-logos.cdn.com'} + 'cdn_url': 'https://static-logos.cdn.com', + 'banner_colour': data['banner_colour'], 'single_id_colour': data['single_id_colour'], + 'domain': data['domain'] + } ) assert mock_update_email_branding.called @@ -234,7 +261,10 @@ def test_update_exisiting_branding( logo=data['logo'], name=data['name'], text=data['text'], - colour=data['colour'] + colour=data['colour'], + banner_colour=data['banner_colour'], + single_id_colour=data['single_id_colour'], + domain=data['domain'], ) @@ -316,7 +346,10 @@ def test_colour_regex_validation( 'logo': None, 'colour': colour_hex, 'text': 'new text', - 'name': 'new name' + 'name': 'new name', + 'domain': 'sample.com', + 'banner_colour': '#FFFF00', + 'single_id_colour': '#00FF00', } mocker.patch('app.main.views.email_branding.delete_temp_files_created_by') diff --git a/tests/app/notify_client/test_email_branding_client.py b/tests/app/notify_client/test_email_branding_client.py index 7a1fcd8bb..a6dbb29f5 100644 --- a/tests/app/notify_client/test_email_branding_client.py +++ b/tests/app/notify_client/test_email_branding_client.py @@ -26,11 +26,15 @@ def test_get_letter_email_branding(mocker): def test_create_email_branding(mocker): - org_data = {'logo': 'test.png', 'name': 'test name', 'text': 'test name', 'colour': 'red'} + org_data = {'logo': 'test.png', 'name': 'test name', 'text': 'test name', 'colour': 'red', + 'banner_colour': 'blue', 'single_id_colour': 'yellow', 'domain': 'sample.com'} mock_post = mocker.patch('app.notify_client.email_branding_client.EmailBrandingClient.post') EmailBrandingClient().create_email_branding( - logo=org_data['logo'], name=org_data['name'], text=org_data['text'], colour=org_data['colour']) + logo=org_data['logo'], name=org_data['name'], text=org_data['text'], colour=org_data['colour'], + banner_colour=org_data['banner_colour'], single_id_colour=org_data['single_id_colour'], + domain=org_data['domain'] + ) mock_post.assert_called_once_with( url='/email-branding', @@ -39,12 +43,14 @@ def test_create_email_branding(mocker): def test_update_email_branding(mocker, fake_uuid): - org_data = {'logo': 'test.png', 'name': 'test name', 'text': 'test name', 'colour': 'red'} + org_data = {'logo': 'test.png', 'name': 'test name', 'text': 'test name', 'colour': 'red', + 'banner_colour': 'blue', 'single_id_colour': 'yellow', 'domain': 'sample.com'} mock_post = mocker.patch('app.notify_client.email_branding_client.EmailBrandingClient.post') EmailBrandingClient().update_email_branding( branding_id=fake_uuid, logo=org_data['logo'], name=org_data['name'], text=org_data['text'], - colour=org_data['colour']) + colour=org_data['colour'], banner_colour=org_data['banner_colour'], + single_id_colour=org_data['single_id_colour'], domain=org_data['domain']) mock_post.assert_called_once_with( url='/email-branding/{}'.format(fake_uuid), diff --git a/tests/conftest.py b/tests/conftest.py index a056d43b2..c80849a7b 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -2493,7 +2493,10 @@ def mock_get_email_branding(mocker, fake_uuid): 'name': 'Organisation name', 'text': 'Organisation text', 'id': fake_uuid, - 'colour': '#f00' + 'colour': '#f00', + 'banner_colour': '#f11', + 'single_id_colour': '#f22', + 'domain': 'sample.com', } } @@ -2523,7 +2526,7 @@ def mock_get_email_branding_without_brand_text(mocker, fake_uuid): @pytest.fixture(scope='function') def mock_create_email_branding(mocker): - def _create_email_branding(logo, name, text, colour): + def _create_email_branding(logo, name, text, colour, banner_colour, single_id_colour, domain): return return mocker.patch( @@ -2533,7 +2536,7 @@ def mock_create_email_branding(mocker): @pytest.fixture(scope='function') def mock_update_email_branding(mocker): - def _update_email_branding(branding_id, logo, name, text, colour): + def _update_email_branding(branding_id, logo, name, text, colour, banner_colour, single_id_colour, domain): return return mocker.patch(