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)