def __init__(self, user_id): line = line_services.get_by_user_id(user_id) self.context = line.context self.protocol = line.protocol.upper() self.name = line.name self.number = user_line_dao.get_main_exten_by_user_id(user_id)
def delete_line(user): try: line = line_services.get_by_user_id(user.id) except LookupError: return else: line_services.delete(line)
def test_get_by_user_id(self, mock_get_by_user_id): user_id = 1 line = Mock() mock_get_by_user_id.return_value = line result = line_services.get_by_user_id(user_id) mock_get_by_user_id.assert_called_once_with(user_id) self.assertEquals(result, line)
def test_get_by_user_id(self, mock_get_by_user_id): user_id = 1 line = Mock() mock_get_by_user_id.return_value = line result = line_services.get_by_user_id(user_id) mock_get_by_user_id.assert_called_once_with(user_id) self.assertEquals(result, line)
def find_with_user_id(user_id): try: line = line_services.get_by_user_id(user_id) except ElementNotExistsError: raise Exception('expecting line with user ID %r not found' % (user_id)) return line
def find_with_user_id(user_id): try: line = line_services.get_by_user_id(user_id) except NotFoundError: raise Exception('expecting line with user ID %r not found' % (user_id)) return line