Scan AfterFailedUnit inside test

src/com/artos/framework/infra/RunnerTestUnits.java

@@ -268,6 +268,16 @@ private void runIndividualUnitTest(TestUnitObjectWrapper unit) {
 				context.getAfterTestUnit().invoke(context.getPrePostRunnableObj().getDeclaredConstructor().newInstance(), context);
 				notifyGlobalAfterTestUnitMethodExecutionFinished(unit);
 			}
+
+			// Run custom after unit failed method post each failed test unit execution
+			// If KTF marked test unit is passing then also execute this method because outcome of this unit will be failed
+			if (context.getCurrentUnitTestStatus() == TestStatus.FAIL || (unit.isKTF() && context.getCurrentUnitTestStatus() == TestStatus.PASS)) {
+				if (null != t.getMethodAfterFailedUnit()) {
+					notifyLocalAfterFailedUnitMethodExecutionStarted(t, unit);
+					t.getMethodAfterFailedUnit().invoke(t.getTestClassObject().newInstance(), context);
+					notifyLocalAfterFailedUnitMethodExecutionFinished(unit);
+				}
+			}
 
 			// Run global after unit failed method post each failed test unit execution
 			// If KTF marked test unit is passing then also execute this method because outcome of this unit will be failed
@@ -603,6 +613,18 @@ void notifyLocalAfterTestUnitMethodExecutionFinished(TestUnitObjectWrapper unit)
 			listener.localAfterTestUnitMethodExecutionFinished(unit);
 		}
 	}
+
+	void notifyLocalAfterFailedUnitMethodExecutionStarted(TestObjectWrapper t, TestUnitObjectWrapper unit) {
+		for (TestProgress listener : listenerList) {
+			listener.localAfterFailedUnitMethodExecutionStarted(t, unit);
+		}
+	}
+
+	void notifyLocalAfterFailedUnitMethodExecutionFinished(TestUnitObjectWrapper unit) {
+		for (TestProgress listener : listenerList) {
+			listener.localAfterFailedUnitMethodExecutionFinished(unit);
+		}
+	}
 
 	void notifyTestUnitExecutionStarted(TestUnitObjectWrapper unit) {
 		for (TestProgress listener : listenerList) {

src/com/artos/framework/infra/ScanTestCase.java

@@ -29,6 +29,7 @@
 import java.util.List;
 import java.util.stream.Collectors;
 
+import com.artos.annotation.AfterFailedUnit;
 import com.artos.annotation.AfterTest;
 import com.artos.annotation.AfterTestUnit;
 import com.artos.annotation.BeforeTest;
@@ -110,6 +111,11 @@ private void scanForTestUnits(TestObjectWrapper testObj) {
 					testObj.setMethodAfterTestCase(method);
 					continue;
 				}
+
+				// If one method is found then do not accept any other
+				if (null == testObj.getMethodAfterFailedUnit() && isValidMethod(method, AfterFailedUnit.class)) {
+					testObj.setMethodAfterFailedUnit(method);
+				}
 			}
 			// move to the upper class in the hierarchy in search for more methods
 			klass = klass.getSuperclass();

src/com/artos/framework/infra/TestObjectWrapper.java

@@ -63,6 +63,7 @@ public class TestObjectWrapper {
 	Method methodAfterTestUnit = null;
 	Method methodBeforeTestCase = null;
 	Method methodAfterTestCase = null;
+	Method methodAfterFailedUnit = null;
 	List<TestUnitObjectWrapper> testUnitList = null;
 
 	// TestPlan
@@ -324,6 +325,14 @@ public Method getMethodBeforeTestCase() {
 	public void setMethodBeforeTestCase(Method methodBeforeTestCase) {
 		this.methodBeforeTestCase = methodBeforeTestCase;
 	}
+
+	public Method getMethodAfterFailedUnit() {
+		return methodAfterFailedUnit;
+	}
+
+	public void setMethodAfterFailedUnit(Method methodAfterFailedUnit) {
+		this.methodAfterFailedUnit = methodAfterFailedUnit;
+	}
 
 	public Method getMethodAfterTestCase() {
 		return methodAfterTestCase;

src/com/artos/framework/listener/ExtentReportListener.java

@@ -484,6 +484,18 @@ public void localAfterTestUnitMethodExecutionFinished(TestUnitObjectWrapper unit
 		// TODO Auto-generated method stub
 
 	}
+
+	@Override
+	public void localAfterFailedUnitMethodExecutionStarted(TestObjectWrapper t, TestUnitObjectWrapper unit) {
+		// TODO Auto-generated method stub
+
+	}
+
+	@Override
+	public void localAfterFailedUnitMethodExecutionFinished(TestUnitObjectWrapper unit) {
+		// TODO Auto-generated method stub
+
+	}
 
 	@Override
 	public void testUnitExecutionStarted(TestUnitObjectWrapper unit) {

src/com/artos/framework/listener/JUnitReportListener.java

@@ -285,6 +285,18 @@ public void localAfterTestUnitMethodExecutionFinished(TestUnitObjectWrapper unit
 		// TODO Auto-generated method stub
 
 	}
+
+	@Override
+	public void localAfterFailedUnitMethodExecutionStarted(TestObjectWrapper t, TestUnitObjectWrapper unit) {
+		// TODO Auto-generated method stub
+
+	}
+
+	@Override
+	public void localAfterFailedUnitMethodExecutionFinished(TestUnitObjectWrapper unit) {
+		// TODO Auto-generated method stub
+
+	}
 
 	@Override
 	public void globalBeforeTestCaseMethodExecutionStarted(String methodName, TestObjectWrapper t) {

src/com/artos/framework/listener/TestExecutionEventListener.java

@@ -398,6 +398,16 @@ public void localAfterTestUnitMethodExecutionStarted(TestObjectWrapper t, TestUn
 	public void localAfterTestUnitMethodExecutionFinished(TestUnitObjectWrapper unit) {
 		// logger.trace("\n---------------- Local After Test Unit Method Finished -------------------");
 	}
+
+	@Override
+	public void localAfterFailedUnitMethodExecutionStarted(TestObjectWrapper t, TestUnitObjectWrapper unit) {
+		logger.info("\n=> " + t.getMethodAfterFailedUnit().getName() + "(context)");
+	}
+
+	@Override
+	public void localAfterFailedUnitMethodExecutionFinished(TestUnitObjectWrapper unit) {
+		// logger.trace("\n---------------- Local After Failed Unit Method Finished -------------------");
+	}
 
 	@Override
 	public void testUnitExecutionStarted(TestUnitObjectWrapper unit) {

src/com/artos/framework/listener/UDPReportListener.java

@@ -163,6 +163,18 @@ public void localAfterTestUnitMethodExecutionFinished(TestUnitObjectWrapper unit
 		// TODO Auto-generated method stub
 
 	}
+
+	@Override
+	public void localAfterFailedUnitMethodExecutionStarted(TestObjectWrapper t, TestUnitObjectWrapper unit) {
+		// TODO Auto-generated method stub
+
+	}
+
+	@Override
+	public void localAfterFailedUnitMethodExecutionFinished(TestUnitObjectWrapper unit) {
+		// TODO Auto-generated method stub
+
+	}
 
 	@Override
 	public void globalBeforeTestCaseMethodExecutionStarted(String methodName, TestObjectWrapper t) {

src/com/artos/interfaces/TestProgress.java

@@ -256,6 +256,23 @@ public interface TestProgress {
 	 * @see TestUnitObjectWrapper
 	 */
 	public void localAfterTestUnitMethodExecutionFinished(TestUnitObjectWrapper unit);
+
+	/**
+	 * Method is called after local {@code AfterFailedUnit} method execution starts
+	 * 
+	 * @param t test object wrapper
+	 * @param unit test unit object wrapper
+	 * @see TestUnitObjectWrapper
+	 */
+	public void localAfterFailedUnitMethodExecutionStarted(TestObjectWrapper t, TestUnitObjectWrapper unit);
+
+	/**
+	 * Method is called after local {@code AfterTestUnit} method execution starts
+	 * 
+	 * @param unit test unit object wrapper
+	 * @see TestUnitObjectWrapper
+	 */
+	public void localAfterFailedUnitMethodExecutionFinished(TestUnitObjectWrapper unit);
 
 	// ==========================================================
 	// Test Case Before and After