org.specrunner.sql.meta.impl.UtilSchema Maven / Gradle / Ivy

Go to download
/*
    SpecRunner - Acceptance Test Driven Development Tool
    Copyright (C) 2011-2016  Thiago Santos

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see 
 */
package org.specrunner.sql.meta.impl;

import org.specrunner.comparators.ComparatorException;
import org.specrunner.context.IContext;
import org.specrunner.converters.ConverterException;
import org.specrunner.formatters.FormatterException;
import org.specrunner.plugins.PluginException;
import org.specrunner.readers.ReaderException;
import org.specrunner.sql.meta.Column;
import org.specrunner.sql.meta.ISchemaLoaderXML;
import org.specrunner.sql.meta.Table;
import org.specrunner.util.xom.node.INodeHolder;

/**
 * Utility class for Schema manipulation.
 * 
 * @author Thiago Santos
 * 
 */
public final class UtilSchema {

    /**
     * Default constructor.
     */
    private UtilSchema() {
    }

    /**
     * Setup a column in a conservative way. Only filled attributes are replaced
     * the old ones remain unchanged.
     * 
     * @param context
     *            The test context.
     * @param table
     *            The table where column is expected to be.
     * @param column
     *            The column to fill.
     * @param holder
     *            The node holder.
     * @throws ConverterException
     *             On default value conversion errors.
     * @throws ComparatorException
     *             On comparator lookup errors.
     * @throws ReaderException
     *             On reader lookup errors.
     * @throws FormatterException On formatter lookup errors.
     */
    public static void setupColumn(IContext context, Table table, Column column, INodeHolder holder) throws ConverterException, ComparatorException, ReaderException, FormatterException {
        column.setName(holder.getAttribute(ISchemaLoaderXML.ATTR_NAME, column.getName()));
        column.setAlias(holder.getAttribute(ISchemaLoaderXML.ATTR_ALIAS, column.getAlias() != null ? column.getAlias() : column.getName()));
        column.setTable(holder.getAttribute(ISchemaLoaderXML.ATTR_TABLE, column.getTable()));
        column.setKey(column.isKey() || Boolean.parseBoolean(holder.getAttribute(ISchemaLoaderXML.ATT_KEY, ISchemaLoaderXML.DEFAULT_FALSE)));
        column.setSequence(holder.getAttribute(ISchemaLoaderXML.ATT_SEQUENCE, column.getSequence()));
        column.setDate(column.isDate() || Boolean.parseBoolean(holder.getAttribute(ISchemaLoaderXML.ATT_DATE, ISchemaLoaderXML.DEFAULT_FALSE)));
        column.setReader(holder.getReader(column.getReader()));
        column.setConverter(holder.getConverter(column.getConverter()));
        column.setArguments(holder.getArguments(column.getArguments()));
        column.setFormatter(holder.getFormatter(column.getFormatter()));
        column.setFormatterArguments(holder.getFormatterArguments(column.getFormatterArguments()));
        column.setComparator(holder.getComparator(column.getComparator()));
        if (holder.hasAttribute(ISchemaLoaderXML.ATT_DEFAULT)) {
            String value = holder.getAttribute(ISchemaLoaderXML.ATT_DEFAULT);
            try {
                // set value to enable use
                holder.setAttribute(INodeHolder.ATTRIBUTE_VALUE, value);
                column.setDefaultExpression(holder);
                // try convert, to be sure expression is valid from start
                Object defValue = holder.getObject(context, true);
                // set default expression
                column.setDefaultValue(defValue);
            } catch (PluginException e) {
                throw new ConverterException(e);
            }
        }
        column.setVirtual(column.isVirtual() || Boolean.parseBoolean(holder.getAttribute(ISchemaLoaderXML.ATT_VIRTUAL, ISchemaLoaderXML.DEFAULT_FALSE)));
        column.setReference(column.isReference() || Boolean.parseBoolean(holder.getAttribute(ISchemaLoaderXML.ATT_REFERENCE, ISchemaLoaderXML.DEFAULT_FALSE)));
        column.setPointer(holder.getAttribute(ISchemaLoaderXML.ATTR_POINTER, column.getPointer()));
    }
}