[SPR-4702] now updating TestContext state before calling before/after test...

[SPR-4702] now updating TestContext state before calling before/after test class life cycle callbacks.

org.springframework.test/src/main/java/org/springframework/test/context/TestContextManager.java

@@ -47,10 +47,9 @@ import org.springframework.util.Assert;
  * execution points:
  * </p>
  * <ul>
- * <li>{@link #beforeTestClass() before test class execution}:
- * prior to any <em>before class methods</em> of a particular testing
- * framework (e.g., JUnit 4's {@link org.junit.BeforeClass
- * &#064;BeforeClass})</li>
+ * <li>{@link #beforeTestClass() before test class execution}: prior to any
+ * <em>before class methods</em> of a particular testing framework (e.g., JUnit
+ * 4's {@link org.junit.BeforeClass &#064;BeforeClass})</li>
  * <li>{@link #prepareTestInstance(Object) test instance preparation}:
  * immediately following instantiation of the test instance</li>
  * <li>{@link #beforeTestMethod(Object,Method) before test method execution}:
@@ -60,9 +59,8 @@ import org.springframework.util.Assert;
  * execution}: after any <em>after methods</em> of a particular testing
  * framework (e.g., JUnit 4's {@link org.junit.After &#064;After})</li>
  * <li>{@link #afterTestClass() after test class execution}: after any
- * <em>after class methods</em> of a particular testing
- * framework (e.g., JUnit 4's {@link org.junit.AfterClass
- * &#064;AfterClass})</li>
+ * <em>after class methods</em> of a particular testing framework (e.g., JUnit
+ * 4's {@link org.junit.AfterClass &#064;AfterClass})</li>
  * </ul>
  * 
  * @author Sam Brannen
@@ -152,6 +150,17 @@ public class TestContextManager {
 		return Collections.unmodifiableList(this.testExecutionListeners);
 	}
 
+	/**
+	 * Gets a copy of the {@link TestExecutionListener TestExecutionListeners}
+	 * registered for this <code>TestContextManager</code> in reverse order.
+	 */
+	private List<TestExecutionListener> getReversedTestExecutionListeners() {
+		List<TestExecutionListener> listenersReversed = new ArrayList<TestExecutionListener>(
+			getTestExecutionListeners());
+		Collections.reverse(listenersReversed);
+		return listenersReversed;
+	}
+
 	/**
 	 * Retrieves an array of newly instantiated {@link TestExecutionListener
 	 * TestExecutionListeners} for the specified {@link Class class}. If
@@ -264,6 +273,7 @@ public class TestContextManager {
 		if (logger.isTraceEnabled()) {
 			logger.trace("beforeTestClass(): class [" + testClass + "]");
 		}
+		getTestContext().updateState(null, null, null);
 
 		for (TestExecutionListener testExecutionListener : getTestExecutionListeners()) {
 			try {
@@ -393,14 +403,10 @@ public class TestContextManager {
 		}
 		getTestContext().updateState(testInstance, testMethod, exception);
 
+		Exception afterTestMethodException = null;
 		// Traverse the TestExecutionListeners in reverse order to ensure proper
 		// "wrapper"-style execution of listeners.
-		List<TestExecutionListener> listenersReversed = new ArrayList<TestExecutionListener>(
-			getTestExecutionListeners());
-		Collections.reverse(listenersReversed);
-
-		Exception afterTestMethodException = null;
-		for (TestExecutionListener testExecutionListener : listenersReversed) {
+		for (TestExecutionListener testExecutionListener : getReversedTestExecutionListeners()) {
 			try {
 				testExecutionListener.afterTestMethod(getTestContext());
 			}
@@ -440,15 +446,12 @@ public class TestContextManager {
 		if (logger.isTraceEnabled()) {
 			logger.trace("afterTestClass(): class [" + testClass + "]");
 		}
+		getTestContext().updateState(null, null, null);
 
+		Exception afterTestClassException = null;
 		// Traverse the TestExecutionListeners in reverse order to ensure proper
 		// "wrapper"-style execution of listeners.
-		List<TestExecutionListener> listenersReversed = new ArrayList<TestExecutionListener>(
-			getTestExecutionListeners());
-		Collections.reverse(listenersReversed);
-
-		Exception afterTestClassException = null;
-		for (TestExecutionListener testExecutionListener : listenersReversed) {
+		for (TestExecutionListener testExecutionListener : getReversedTestExecutionListeners()) {
 			try {
 				testExecutionListener.afterTestClass(getTestContext());
 			}
@@ -465,4 +468,4 @@ public class TestContextManager {
 		}
 	}
 
-}
+}
\ No newline at end of file