sayems
diff --git a/‎src/main/java/org/sayem/webdriver/TestBase.java
Copy file name to clipboardExpand all lines: src/main/java/org/sayem/webdriver/TestBase.java
+2-1Lines changed: 2 additions & 1 deletion b/‎src/main/java/org/sayem/webdriver/TestBase.java
Copy file name to clipboardExpand all lines: src/main/java/org/sayem/webdriver/TestBase.java
+2-1Lines changed: 2 additions & 1 deletion
diff --git a/‎src/main/java/org/sayem/webdriver/listeners/RetryListener.java
Copy file name to clipboard
+24Lines changed: 24 additions & 0 deletions b/‎src/main/java/org/sayem/webdriver/listeners/RetryListener.java
Copy file name to clipboard
+24Lines changed: 24 additions & 0 deletions
diff --git a/‎src/main/java/org/sayem/webdriver/listeners/TestNGRetry.java
Copy file name to clipboard
+35Lines changed: 35 additions & 0 deletions b/‎src/main/java/org/sayem/webdriver/listeners/TestNGRetry.java
Copy file name to clipboard
+35Lines changed: 35 additions & 0 deletions
@@ -4,6 +4,7 @@
 import org.openqa.selenium.WebDriver;
 import org.openqa.selenium.support.PageFactory;
 import org.sayem.webdriver.browsers.config.BrowserThreads;
+import org.sayem.webdriver.listeners.RetryListener;
 import org.sayem.webdriver.listeners.ScreenshotListener;
 import org.sayem.webdriver.properties.PropertiesUtil;
 import org.sayem.webdriver.properties.Repository;
@@ -19,7 +20,7 @@
 /**
  * Created by sayem on 10/05/15.
  */
-@Listeners({ScreenshotListener.class})
+@Listeners({ScreenshotListener.class, RetryListener.class})
 public abstract class TestBase {
 
     private static List<BrowserThreads> webDriverThreadPool = Collections.synchronizedList(new ArrayList<>());
 
@@ -0,0 +1,24 @@
+package org.sayem.webdriver.listeners;
+
+import java.lang.reflect.Constructor;
+import java.lang.reflect.Method;
+
+import org.testng.IAnnotationTransformer;
+import org.testng.IRetryAnalyzer;
+import org.testng.annotations.ITestAnnotation;
+
+/**
+ * Created by sayem on 2/2/16.
+ */
+public class RetryListener implements IAnnotationTransformer {
+
+    @Override
+    public void transform(ITestAnnotation testannotation, Class testClass,
+                          Constructor testConstructor, Method testMethod)	{
+        IRetryAnalyzer retry = testannotation.getRetryAnalyzer();
+
+        if (retry == null)	{
+            testannotation.setRetryAnalyzer(TestNGRetry.class);
+        }
+    }
+}
@@ -0,0 +1,35 @@
+package org.sayem.webdriver.listeners;
+
+import org.testng.IRetryAnalyzer;
+import org.testng.ITestResult;
+
+/**
+ * Created by sayem on 2/2/16.
+ */
+public class TestNGRetry implements IRetryAnalyzer {
+    private int retryCount = 0;
+    private int maxRetryCount = 1;
+
+    // Below method returns 'true' if the test method has to be retried else 'false'
+    // and it takes the 'Result' as parameter of the test method that just ran
+    public boolean retry(ITestResult result) {
+        if (retryCount < maxRetryCount) {
+            System.out.println("Retrying test " + result.getName() + " with status "
+                    + getResultStatusName(result.getStatus()) + " for the " + (retryCount+1) + " time(s).");
+            retryCount++;
+            return true;
+        }
+        return false;
+    }
+
+    public String getResultStatusName(int status) {
+        String resultName = null;
+        if(status==1)
+            resultName = "SUCCESS";
+        if(status==2)
+            resultName = "FAILURE";
+        if(status==3)
+            resultName = "SKIP";
+        return resultName;
+    }
+}