fixes #2

Author: anshooarora

src/main/java/com/aventstack/extentreports/service/ExtentTestManager.java

@@ -8,6 +8,7 @@
 import org.testng.ITestResult;
 
 import com.aventstack.extentreports.ExtentTest;
+import com.aventstack.extentreports.Status;
 import com.aventstack.extentreports.model.Test;
 import com.aventstack.extentreports.testng.listener.commons.ExtentTestCommons;
 
@@ -94,5 +95,28 @@ private static synchronized ExtentTest createTest(ITestResult result, ExtentTest
         ExtentTestCommons.assignGroups(test, groups);
         return test;
     }
+    
+    public static synchronized void log(ITestResult result) {
+        String msg = "Test ";
+        Status status = Status.PASS;
+        switch (result.getStatus()) {
+            case ITestResult.SKIP:
+                status = Status.SKIP;
+                msg += "skipped";
+                break;
+            case ITestResult.FAILURE:
+                status = Status.FAIL;
+                msg += "failed";
+                break;
+            default:
+                msg += "passed";
+                break;
+        }
+        if (result.getThrowable() != null) {
+            ExtentTestManager.getTest(result).log(status, result.getThrowable());
+            return;
+        }
+        ExtentTestManager.getTest(result).log(status, msg);
+    }
 
 }

src/main/java/com/aventstack/extentreports/testng/listener/ExtentITestListenerAdapter.java

@@ -27,21 +27,17 @@ public synchronized void onTestStart(ITestResult result) {
 
     @Override
     public synchronized void onTestSuccess(ITestResult result) {
-        ExtentTestManager.getTest(result).pass("Test passed");
+        ExtentTestManager.log(result);
     }
 
     @Override
     public synchronized void onTestFailure(ITestResult result) {
-        ExtentTestManager.getTest(result).fail(result.getThrowable());
+        ExtentTestManager.log(result);
     }
 
     @Override
     public synchronized void onTestSkipped(ITestResult result) {
-        if (result.getThrowable() == null) {
-            ExtentTestManager.getTest(result).skip("Test skipped");
-            return;
-        }
-        ExtentTestManager.getTest(result).skip(result.getThrowable());
+        ExtentTestManager.log(result);
     }
 
     @Override

src/main/java/com/aventstack/extentreports/testng/listener/ExtentITestListenerClassAdapter.java

@@ -27,17 +27,17 @@ public synchronized void onTestStart(ITestResult result) {
 
     @Override
     public synchronized void onTestSuccess(ITestResult result) {
-        ExtentTestManager.getTest(result).pass("Test passed");
+        ExtentTestManager.log(result);
     }
 
     @Override
     public synchronized void onTestFailure(ITestResult result) {
-        ExtentTestManager.getTest(result).fail(result.getThrowable());
+        ExtentTestManager.log(result);
     }
 
     @Override
     public synchronized void onTestSkipped(ITestResult result) {
-        ExtentTestManager.getTest(result).skip(result.getThrowable());
+        ExtentTestManager.log(result);
     }
 
     @Override