1 files changed, 65 insertions, 0 deletions
diff --git a/junit4/src/test/java/org/junit/tests/experimental/theories/extendingwithstubs/StubbedTheories.java b/junit4/src/test/java/org/junit/tests/experimental/theories/extendingwithstubs/StubbedTheories.java
new file mode 100644
index 0000000..91c0b42
--- /dev/null
+++ b/junit4/src/test/java/org/junit/tests/experimental/theories/extendingwithstubs/StubbedTheories.java
@@ -0,0 +1,65 @@
+package org.junit.tests.experimental.theories.extendingwithstubs;
+
+
+import java.util.ArrayList;
+import java.util.List;
+
+import org.junit.experimental.theories.ParameterSignature;
+import org.junit.experimental.theories.Theories;
+import org.junit.experimental.theories.internal.Assignments;
+import org.junit.internal.AssumptionViolatedException;
+import org.junit.runners.model.FrameworkMethod;
+import org.junit.runners.model.InitializationError;
+import org.junit.runners.model.Statement;
+import org.junit.runners.model.TestClass;
+
+public class StubbedTheories extends Theories {
+	public StubbedTheories(Class<?> klass) throws InitializationError {
+		super(klass);
+	}
+
+	@Override
+	public Statement methodBlock(FrameworkMethod method) {
+		return new StubbedTheoryAnchor(method, getTestClass());
+	}
+	
+	public static class StubbedTheoryAnchor extends TheoryAnchor {
+		public StubbedTheoryAnchor(FrameworkMethod method, TestClass testClass) {
+			super(method, testClass);
+		}
+
+		private List<GuesserQueue> queues= new ArrayList<GuesserQueue>();
+
+		@Override
+		protected void handleAssumptionViolation(AssumptionViolatedException e) {
+			super.handleAssumptionViolation(e);
+			for (GuesserQueue queue : queues)
+				queue.update(e);
+		}
+
+		@Override
+		protected void runWithIncompleteAssignment(Assignments incomplete)
+				throws InstantiationException, IllegalAccessException,
+				Throwable {
+			GuesserQueue guessers= createGuesserQueue(incomplete);
+			queues.add(guessers);
+			while (!guessers.isEmpty())
+				runWithAssignment(incomplete.assignNext(guessers.remove(0)));
+			queues.remove(guessers);
+		}
+
+		private GuesserQueue createGuesserQueue(Assignments incomplete)
+				throws InstantiationException, IllegalAccessException {
+			ParameterSignature nextUnassigned= incomplete.nextUnassigned();
+
+			if (nextUnassigned.hasAnnotation(Stub.class)) {
+				GuesserQueue queue= new GuesserQueue();
+				queue.add(new Guesser<Object>(nextUnassigned.getType()));
+				return queue;
+			}
+
+			return GuesserQueue.forSingleValues(incomplete.potentialsForNextUnassigned());
+		}
+	}
+
+}