def test_deletes_previous_temp_logo_after_uploading_logo( logged_in_platform_admin_client, mocker, fake_uuid): with logged_in_platform_admin_client.session_transaction() as session: user_id = session["user_id"] temp_old_filename = LOGO_LOCATION_STRUCTURE.format( temp=TEMP_TAG.format(user_id=user_id), unique_id=fake_uuid, filename='old_test.png') temp_filename = LOGO_LOCATION_STRUCTURE.format( temp=TEMP_TAG.format(user_id=user_id), unique_id=fake_uuid, filename='test.png') mocked_upload_logo = mocker.patch( 'app.main.views.organisations.upload_logo', return_value=temp_filename) mocked_delete_temp_file = mocker.patch( 'app.main.views.organisations.delete_temp_file') logged_in_platform_admin_client.post( url_for('.manage_org', logo=temp_old_filename), data={'file': (BytesIO(''.encode('utf-8')), 'test.png')}, content_type='multipart/form-data') assert mocked_upload_logo.called assert mocked_delete_temp_file.called assert mocked_delete_temp_file.call_args == call(temp_old_filename)
def test_create_new_email_branding_when_branding_saved( logged_in_platform_admin_client, mocker, mock_create_email_branding, fake_uuid): with logged_in_platform_admin_client.session_transaction() as session: user_id = session["user_id"] data = {'logo': 'test.png', 'colour': '#ff0000', 'name': 'new name'} temp_filename = LOGO_LOCATION_STRUCTURE.format( temp=TEMP_TAG.format(user_id=user_id), unique_id=fake_uuid, filename=data['logo']) mocker.patch('app.main.views.email_branding.persist_logo', return_value=data['logo']) mocker.patch('app.main.views.email_branding.delete_temp_files_created_by') logged_in_platform_admin_client.post(url_for('.create_email_branding', logo=temp_filename), content_type='multipart/form-data', data={ 'colour': data['colour'], 'name': data['name'], 'cdn_url': 'https://static-logos.cdn.com' }) assert mock_create_email_branding.called assert mock_create_email_branding.call_args == call(logo=data['logo'], name=data['name'], colour=data['colour'])
def test_logo_persisted_when_organisation_saved( logged_in_platform_admin_client, mocker, fake_uuid): with logged_in_platform_admin_client.session_transaction() as session: user_id = session["user_id"] temp_filename = LOGO_LOCATION_STRUCTURE.format( temp=TEMP_TAG.format(user_id=user_id), unique_id=fake_uuid, filename='test.png') mocked_upload_logo = mocker.patch( 'app.main.views.email_branding.upload_logo') mocked_persist_logo = mocker.patch( 'app.main.views.email_branding.persist_logo', return_value='test.png') mocked_delete_temp_files_by = mocker.patch( 'app.main.views.email_branding.delete_temp_files_created_by') logged_in_platform_admin_client.post(url_for('.create_email_branding', logo=temp_filename), content_type='multipart/form-data') assert not mocked_upload_logo.called assert mocked_persist_logo.called assert mocked_delete_temp_files_by.called assert mocked_delete_temp_files_by.call_args == call(user_id)
def test_temp_logo_is_shown_after_uploading_logo( logged_in_platform_admin_client, mocker, fake_uuid, ): with logged_in_platform_admin_client.session_transaction() as session: user_id = session["user_id"] temp_filename = LOGO_LOCATION_STRUCTURE.format( temp=TEMP_TAG.format(user_id=user_id), unique_id=fake_uuid, filename='test.png') mocker.patch('app.main.views.email_branding.upload_logo', return_value=temp_filename) mocker.patch('app.main.views.email_branding.delete_temp_file') response = logged_in_platform_admin_client.post( url_for('main.create_email_branding'), data={'file': (BytesIO(''.encode('utf-8')), 'test.png')}, content_type='multipart/form-data', follow_redirects=True) assert response.status_code == 200 page = BeautifulSoup(response.data.decode('utf-8'), 'html.parser') assert page.select_one('#logo-img > img').attrs['src'].endswith( temp_filename)
def create_email_branding(logo=None): form = ServiceCreateEmailBranding() if form.validate_on_submit(): if form.file.data: upload_filename = upload_logo(form.file.data.filename, form.file.data, current_app.config['AWS_REGION'], user_id=session["user_id"]) if logo and logo.startswith( TEMP_TAG.format(user_id=session['user_id'])): delete_temp_file(logo) return redirect( url_for('.create_email_branding', logo=upload_filename)) if logo: logo = persist_logo(logo, session["user_id"]) delete_temp_files_created_by(session["user_id"]) email_branding_client.create_email_branding(logo=logo, name=form.name.data, colour=form.colour.data) return redirect(url_for('.email_branding')) return render_template('views/email-branding/manage-branding.html', form=form, cdn_url=get_cdn_domain(), logo=logo)
def manage_org(logo=None): form = ServiceManageOrg() org = session.get("organisation") logo = logo if logo else org.get('logo') if org else None if form.validate_on_submit(): if form.file.data: upload_filename = upload_logo( form.file.data.filename, form.file.data, current_app.config['AWS_REGION'], user_id=session["user_id"] ) if logo and logo.startswith(TEMP_TAG.format(user_id=session['user_id'])): delete_temp_file(logo) return redirect( url_for('.manage_org', logo=upload_filename)) if logo: logo = persist_logo(logo, session["user_id"]) delete_temp_files_created_by(session["user_id"]) if org: organisations_client.update_organisation( org_id=org['id'], logo=logo, name=form.name.data, colour=form.colour.data) org_id = org['id'] else: resp = organisations_client.create_organisation( logo=logo, name=form.name.data, colour=form.colour.data) org_id = resp['data']['id'] return redirect(url_for('.organisations', organisation_id=org_id)) if org: form.name.data = org['name'] form.colour.data = org['colour'] return render_template( 'views/organisations/manage-org.html', form=form, organisation=org, cdn_url=get_cdn_domain(), logo=logo )
def update_email_branding(branding_id, logo=None): email_branding = email_branding_client.get_email_branding(branding_id)['email_branding'] form = ServiceUpdateEmailBranding() logo = logo if logo else email_branding.get('logo') if email_branding else None if form.validate_on_submit(): if form.file.data: upload_filename = upload_logo( form.file.data.filename, form.file.data, current_app.config['AWS_REGION'], user_id=session["user_id"] ) if logo and logo.startswith(TEMP_TAG.format(user_id=session['user_id'])): delete_temp_file(logo) return redirect(url_for('.update_email_branding', branding_id=branding_id, logo=upload_filename)) if logo: logo = persist_logo(logo, session["user_id"]) delete_temp_files_created_by(session["user_id"]) email_branding_client.update_email_branding( branding_id=branding_id, logo=logo, name=form.name.data, text=form.text.data, colour=form.colour.data ) return redirect(url_for('.email_branding', branding_id=branding_id)) form.name.data = email_branding['name'] form.text.data = email_branding['text'] form.colour.data = email_branding['colour'] return render_template( 'views/email-branding/manage-branding.html', form=form, email_branding=email_branding, cdn_url=get_cdn_domain(), logo=logo )
def test_existing_organisation_updated_when_organisation_saved( logged_in_platform_admin_client, mocker, fake_uuid): with logged_in_platform_admin_client.session_transaction() as session: session["organisation"] = sample_orgs[0] user_id = session["user_id"] update_org = {'logo': 'test.png', 'colour': 'blue', 'name': 'new name'} temp_filename = LOGO_LOCATION_STRUCTURE.format( temp=TEMP_TAG.format(user_id=user_id), unique_id=fake_uuid, filename=update_org['logo']) mocked_update_org = mocker.patch( 'app.organisations_client.update_organisation') mocker.patch('app.main.views.organisations.persist_logo', return_value=update_org['logo']) mocker.patch('app.main.views.organisations.delete_temp_files_created_by') logged_in_platform_admin_client.post(url_for('.manage_org', logo=temp_filename), content_type='multipart/form-data', data={ 'colour': update_org['colour'], 'name': update_org['name'], 'cdn_url': 'https://static-logos.cdn.com' }) assert mocked_update_org.called assert mocked_update_org.call_args == call(org_id=sample_orgs[0]['id'], logo=update_org['logo'], name=update_org['name'], colour=update_org['colour'])
def request_post_manage_org_redirect(logged_in_platform_admin_client, mocker, fake_uuid): with logged_in_platform_admin_client.session_transaction() as session: user_id = session["user_id"] temp_filename = LOGO_LOCATION_STRUCTURE.format( temp=TEMP_TAG.format(user_id=user_id), unique_id=fake_uuid, filename='test.png') mocker.patch('app.main.views.organisations.upload_logo', return_value=temp_filename) mocker.patch('app.main.views.organisations.delete_temp_file') mocker.patch('app.main.views.organisations.delete_temp_files_created_by') response = logged_in_platform_admin_client.post( url_for('.manage_org'), data={'file': (BytesIO(''.encode('utf-8')), 'test.png')}, content_type='multipart/form-data', follow_redirects=True) assert response.status_code == 200 return BeautifulSoup(response.data.decode('utf-8'), 'html.parser'), temp_filename
def upload_filename(fake_uuid): return LOGO_LOCATION_STRUCTURE.format( temp=TEMP_TAG.format(user_id=fake_uuid), unique_id=upload_id, filename=filename)