def test_get_enterprise_customer_sender_alias(self, sender_alias, expected_sender_alias, enterprise_customer): """ Verify get_enterprise_customer_sender_alias returns enterprise sender alias if exists otherwise return default. """ enterprise_customer.return_value = {'sender_alias': sender_alias} sender_alias = get_enterprise_customer_sender_alias('some-site', 'uuid') assert sender_alias == expected_sender_alias
def _get_sender_alias(site, nudge_email): """ Returns the sender alias of an Enterprise Customer. Arguments: nudge_email (CodeAssignmentNudgeEmails): A nudge email sent to the learner. """ enterprise_customer_uuid = get_enterprise_customer_uuid(nudge_email.code) return get_enterprise_customer_sender_alias(site, enterprise_customer_uuid)