DataSetModifiers.java

/*
 * Copyright 2002-2016 the original author or authors
 *
 * 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 com.github.springtestdbunit;

import java.lang.reflect.Constructor;
import java.lang.reflect.Modifier;
import java.util.ArrayList;
import java.util.List;

import org.dbunit.dataset.IDataSet;

import com.github.springtestdbunit.dataset.DataSetModifier;

/**
 * A collection of {@link DataSetModifier} items loaded during testing.
 *
 * @author Phillip Webb
 */
class DataSetModifiers implements DataSetModifier {

	private final List<DataSetModifier> modifiers = new ArrayList<DataSetModifier>();

	public IDataSet modify(IDataSet dataSet) {
		for (DataSetModifier modifier : this.modifiers) {
			dataSet = modifier.modify(dataSet);
		}
		return dataSet;
	}

	public void add(Object testInstance, Class<? extends DataSetModifier> modifierClass) {
		try {
			Class<?> enclosingClass = modifierClass.getEnclosingClass();
			if ((enclosingClass == null) || Modifier.isStatic(modifierClass.getModifiers())) {
				add(modifierClass.getDeclaredConstructor());
			} else {
				add(modifierClass.getDeclaredConstructor(enclosingClass), testInstance);
			}
		} catch (Exception ex) {
			throw new IllegalStateException(ex);
		}
	}

	private void add(Constructor<? extends DataSetModifier> constructor, Object... args) throws Exception {
		constructor.setAccessible(true);
		this.modifiers.add(constructor.newInstance(args));
	}

}