package junit.extensions;

import junit.framework.Test;
import junit.framework.TestResult;

/**
 * A Decorator that runs a test repeatedly.
 * 
 */
public class RepeatedTest extends TestDecorator {
	private int fTimesRepeat;

	public RepeatedTest(Test test, int repeat) {
		super(test);
		if (repeat < 0)
			throw new IllegalArgumentException("Repetition count must be >= 0");
		fTimesRepeat= repeat;
	}

	@Override
	public int countTestCases() {
		return super.countTestCases() * fTimesRepeat;
	}

	@Override
	public void run(TestResult result) {
		for (int i= 0; i < fTimesRepeat; i++) {
			if (result.shouldStop())
				break;
			super.run(result);
		}
	}

	@Override
	public String toString() {
		return super.toString() + "(repeated)";
	}
}