def test_add_contact(app, session): # pylint:disable=unused-argument
"""Assert that a contact can be added to an Entity."""
entity_info = {'businessIdentifier': 'CP1234567'}
with app.app_context():
entity = EntityService.create_entity(entity_info)
contact_info = {'emailAddress': '*****@*****.**'}
updated_entity = EntityService.add_contact(entity.business_identifier,
contact_info)
assert updated_entity is not None
contact = EntityService.get_contact_for_business(
updated_entity.business_identifier)
assert contact is not None
assert contact.email == '*****@*****.**'
def test_update_contact(app, session): # pylint:disable=unused-argument
"""Assert that a contact for an entity can be updated."""
entity_info = {'businessIdentifier': 'CP1234567'}
with app.app_context():
entity = EntityService.create_entity(entity_info)
contact_info = {'emailAddress': '*****@*****.**'}
EntityService.add_contact(entity.business_identifier, contact_info)
contact_info['phoneNumber'] = '(555)-555-5555'
updated_entity = EntityService.update_contact(
entity.business_identifier, contact_info)
assert updated_entity
updated_contact = EntityService.get_contact_for_business(
updated_entity.business_identifier)
assert updated_contact is not None
assert updated_contact.phone == '(555)-555-5555'
