Adding new data models for letter branding.

This commit is contained in:
Rebecca Law
2019-01-22 17:27:00 +00:00
parent 52a1b534ee
commit f11aa55e0b
10 changed files with 204 additions and 21 deletions

View File

@@ -50,6 +50,7 @@ from app.models import (
Complaint,
InvitedUser,
TemplateFolder,
LetterBranding
)
@@ -724,3 +725,14 @@ def create_template_folder(service, name='foo', parent=None):
db.session.add(tf)
db.session.commit()
return tf
def create_letter_branding(name='HM Government', filename='hm-government', domain=None, platform_default=True):
test_domain_branding = LetterBranding(name=name,
filename=filename,
domain=domain,
platform_default=platform_default
)
db.session.add(test_domain_branding)
db.session.commit()
return test_domain_branding