def test_update_contact_type_only_success(contact):
test_type = 'whatsapp'
updated_contact = contact_controller.update_contact(contact.id, contact_type=test_type)
assert updated_contact['id'] == contact.id
assert updated_contact['value'] == contact.value
assert updated_contact['type'] == test_type
session = models.get_session()
db_contact = session.query(models.Contact).get(updated_contact['id'])
assert db_contact
assert db_contact.id == updated_contact['id']
assert db_contact.value == updated_contact['value']
assert db_contact.type.type_name == updated_contact['type']
def test_update_contact_value_only_success(contact):
contact_type_type_name = contact.type.type_name
test_value = '5511999888777'
updated_contact = contact_controller.update_contact(contact.id, test_value)
assert updated_contact['id'] == contact.id
assert updated_contact['value'] == test_value
assert updated_contact['type'] == contact_type_type_name
session = models.get_session()
db_contact = session.query(models.Contact).get(updated_contact['id'])
assert db_contact
assert db_contact.id == updated_contact['id']
assert db_contact.value == updated_contact['value']
assert db_contact.type.type_name == updated_contact['type']
def test_update_contact_by_id_both_fields_success(contact):
test_value = '5511999888777'
test_type = 'whatsapp'
updated_contact = contact_controller.update_contact(contact.id, test_value, test_type)
assert updated_contact['id'] == contact.id
assert updated_contact['value'] == test_value
assert updated_contact['type'] == test_type
session = models.get_session()
db_contact = session.query(models.Contact).get(updated_contact['id'])
assert db_contact
assert db_contact.id == updated_contact['id']
assert db_contact.value == updated_contact['value']
assert db_contact.type.type_name == updated_contact['type']
def put(self, contact_id):
try:
input_json = request.get_json()
contact_value = input_json[
'contact_value'] if 'contact_value' in input_json else None
contact_type = input_json[
'contact_type'] if 'contact_type' in input_json else None
except:
return {'error': 'Invalid json input'}, 400
try:
contact = contact_controller.update_contact(
contact_id=contact_id,
contact_value=contact_value,
contact_type=contact_type)
except contact_controller.ContactException as e:
return {'error': e.message}, 502
return contact
def test_update_contact_invalid_contact_type_failure(contact):
with pytest.raises(contact_controller.ContactException):
test_type = 'telegram'
contact_controller.update_contact(contact.id, contact_type=test_type)
def test_update_contact_no_updatable_parameters_failure(contact):
with pytest.raises(contact_controller.ContactException):
contact_controller.update_contact(contact.id)
def test_update_contact_invalid_contact_id_failure():
with pytest.raises(contact_controller.ContactException):
test_value = '5511999888777'
contact_controller.update_contact(10, test_value)