def get_by_user_id(session, user_id): row = _fetch_by_user_id(session, user_id) if not row: raise ElementNotExistsError('User', id=user_id) if row.voicemail_id is None or row.voicemail_id == 0: raise UserVoicemailNotExistsError.from_user_id(user_id) return db_converter.to_model(row)
def test_get_voicemail_associated_to_a_user_with_no_voicemail(self, user_voicemail_get_by_user_id): user_id = 1 expected_status_code = 404 expected_result = ['User with id=%s does not have a voicemail' % user_id] user_voicemail_get_by_user_id.side_effect = UserVoicemailNotExistsError.from_user_id(user_id) result = self.app.get(BASE_URL % user_id) assert_that(result.status_code, equal_to(expected_status_code)) assert_that(self._serialize_decode(result.data), equal_to(expected_result))