darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-08-20 14:29:51 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add create_letter_branding method to letter branding client

Browse Source
This commit is contained in:
Katie Smith
2019-01-31 16:56:00 +00:00
parent b68d0722bb
commit 446b9ccbb2
2 changed files with 22 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
app/notify_client/letter_branding_client.py
View File

@@ -6,5 +6,13 @@ class LetterBrandingClient(NotifyAdminAPIClient):
def get_letter_branding(self): def get_letter_branding(self):
return self.get(url='/dvla_organisations') return self.get(url='/dvla_organisations')
def create_letter_branding(self, filename, name, domain):
data = {
"filename": filename,
"name": name,
"domain": domain,
}
return self.post(url="/letter-branding", data=data)
letter_branding_client = LetterBrandingClient() letter_branding_client = LetterBrandingClient()

14
tests/app/notify_client/test_letter_branding_client.py
View File

@@ -7,3 +7,17 @@ def test_get_letter_branding(mocker):
mock_get.assert_called_once_with( mock_get.assert_called_once_with(
url='/dvla_organisations' url='/dvla_organisations'
) )
def test_create_letter_branding(mocker):
new_branding = {'filename': 'uuid-test', 'name': 'my letters', 'domain': 'example.com'}
mock_post = mocker.patch('app.notify_client.letter_branding_client.LetterBrandingClient.post')
LetterBrandingClient().create_letter_branding(
filename=new_branding['filename'], name=new_branding['name'], domain=new_branding['domain']
)
mock_post.assert_called_once_with(
url='/letter-branding',
data=new_branding
)