Commit cd7e2d84 by Clinton Blackburn Committed by Clinton Blackburn

Loading Organization certificate logo image from Organizations API

ECOM-6860
parent 8021f8ef
...@@ -48,15 +48,19 @@ class OrganizationsApiDataLoader(AbstractDataLoader): ...@@ -48,15 +48,19 @@ class OrganizationsApiDataLoader(AbstractDataLoader):
def update_organization(self, body): def update_organization(self, body):
key = body['short_name'] key = body['short_name']
logo = body['logo']
defaults = { defaults = {
'key': key, 'key': key,
'partner': self.partner 'partner': self.partner,
'certificate_logo_image_url': logo,
} }
if not self.partner.has_marketing_site: if not self.partner.has_marketing_site:
defaults.update({ defaults.update({
'name': body['name'], 'name': body['name'],
'description': body['description'], 'description': body['description'],
'logo_image_url': body['logo'] 'logo_image_url': logo,
}) })
Organization.objects.update_or_create(key__iexact=key, defaults=defaults) Organization.objects.update_or_create(key__iexact=key, defaults=defaults)
......
...@@ -97,6 +97,7 @@ class OrganizationsApiDataLoaderTests(ApiClientTestMixin, DataLoaderTestMixin, T ...@@ -97,6 +97,7 @@ class OrganizationsApiDataLoaderTests(ApiClientTestMixin, DataLoaderTestMixin, T
self.assertEqual(organization.name, AbstractDataLoader.clean_string(body['name'])) self.assertEqual(organization.name, AbstractDataLoader.clean_string(body['name']))
self.assertEqual(organization.description, AbstractDataLoader.clean_string(body['description'])) self.assertEqual(organization.description, AbstractDataLoader.clean_string(body['description']))
self.assertEqual(organization.logo_image_url, AbstractDataLoader.clean_string(body['logo'])) self.assertEqual(organization.logo_image_url, AbstractDataLoader.clean_string(body['logo']))
self.assertEqual(organization.certificate_logo_image_url, AbstractDataLoader.clean_string(body['logo']))
@responses.activate @responses.activate
@ddt.data(True, False) @ddt.data(True, False)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment