def _find_extension_id_for_line(channel, line_id): from xivo_dao.data_handler.line_extension import services as line_extension_services line_extension = line_extension_services.find_by_line_id(line_id) if line_extension: channel.send(line_extension.extension_id) else: channel.send(None)
def test_find_by_line_id(self, dao_find_by_line_id): line_extension = Mock(LineExtension, line_id=1) dao_find_by_line_id.return_value = line_extension result = line_extension_service.find_by_line_id(1) assert_that(result, equal_to(line_extension)) dao_find_by_line_id.assert_called_once_with(1)
def _delete_line_associations(channel, line_id): from xivo_dao.data_handler.line import services as line_services from xivo_dao.data_handler.line_extension import services as line_extension_services from xivo_dao.data_handler.user_line import services as user_line_services line = line_services.get(line_id) line.device_id = None line.device_slot = 1 line_services.edit(line) line_extension = line_extension_services.find_by_line_id(line_id) if line_extension: line_extension_services.dissociate(line_extension) user_lines = user_line_services.find_all_by_line_id(line_id) secondary_associations = [ul for ul in user_lines if not ul.main_user] main_associations = [ul for ul in user_lines if ul.main_user] for user_line in secondary_associations + main_associations: user_line_services.dissociate(user_line)