From 446b9ccbb242f1667a3b4e9e06a3370f1f5c78a4 Mon Sep 17 00:00:00 2001 From: Katie Smith Date: Thu, 31 Jan 2019 16:56:00 +0000 Subject: [PATCH] Add create_letter_branding method to letter branding client --- app/notify_client/letter_branding_client.py | 8 ++++++++ .../notify_client/test_letter_branding_client.py | 14 ++++++++++++++ 2 files changed, 22 insertions(+) diff --git a/app/notify_client/letter_branding_client.py b/app/notify_client/letter_branding_client.py index 96a0889dc..84ab0b80f 100644 --- a/app/notify_client/letter_branding_client.py +++ b/app/notify_client/letter_branding_client.py @@ -6,5 +6,13 @@ class LetterBrandingClient(NotifyAdminAPIClient): def get_letter_branding(self): 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() diff --git a/tests/app/notify_client/test_letter_branding_client.py b/tests/app/notify_client/test_letter_branding_client.py index eceadf474..23f962971 100644 --- a/tests/app/notify_client/test_letter_branding_client.py +++ b/tests/app/notify_client/test_letter_branding_client.py @@ -7,3 +7,17 @@ def test_get_letter_branding(mocker): mock_get.assert_called_once_with( 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 + )