/* * Copyright 2008, Unitils.org * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package org.unitils.dbunit.datasetfactory.impl; import org.unitils.core.UnitilsException; import org.unitils.dbunit.datasetfactory.DataSetFactory; import org.unitils.dbunit.util.MultiSchemaDataSet; import org.unitils.dbunit.util.MultiSchemaXmlDataSetReader; import java.io.File; import java.util.Arrays; import java.util.Properties; /** * A data set factory that can handle data set definitions for multiple database schemas. * * @author Filip Neven * @author Tim Ducheyne */ public class MultiSchemaXmlDataSetFactory implements DataSetFactory { /** * The schema name to use when no name was explicitly specified. */ protected String defaultSchemaName; /** * Initializes this DataSetFactory * * @param configuration The configuration, not null * @param defaultSchemaName The name of the default schema of the test database, not null */ public void init(Properties configuration, String defaultSchemaName) { this.defaultSchemaName = defaultSchemaName; } /** * Creates a {@link MultiSchemaDataSet} using the given file. * * @param dataSetFiles The dataset files, not null * @return A {@link MultiSchemaDataSet} containing the datasets per schema, not null */ public MultiSchemaDataSet createDataSet(File... dataSetFiles) { try { MultiSchemaXmlDataSetReader multiSchemaXmlDataSetReader = new MultiSchemaXmlDataSetReader(defaultSchemaName); return multiSchemaXmlDataSetReader.readDataSetXml(dataSetFiles); } catch (Exception e) { throw new UnitilsException("Unable to create DbUnit dataset for data set files: " + Arrays.toString(dataSetFiles), e); } } /** * @return The extension that files which can be interpreted by this factory must have */ public String getDataSetFileExtension() { return "xml"; } }