skip test fix and new code to skip & add dt count

Author: Abhishek

src/main/java/com/abhishek/listener/TestNGCustomReportListener.java

@@ -449,13 +449,13 @@ public void generateSuiteSummaryReport(List<ISuite> suites) {
                 q = getMethodSet(overview.getSkippedTests(), suite).size();
                 qty_skip += q;
                 String skipColor = "";
-                if(qty_skip > 0)
+                if(q > 0)
                     skipColor = "#feff65";
                 summaryCell(q, 0, skipColor);
                 q = getMethodSet(overview.getFailedTests(), suite).size();
                 qty_fail += q;
                 String failedColor = "";
-                if(qty_fail > 0)
+                if(q > 0)
                     failedColor = "#fa8072";
                 summaryCell(q, 0, failedColor);
 
@@ -691,22 +691,31 @@ public void generateCustomTopBand(List<ISuite> suites) {
             Map<String, ISuiteResult> suiteResults = suite.getResults();
             for (ISuiteResult sr : suiteResults.values()) {
                 ITestContext tc = sr.getTestContext();
+
+                //To get Dataprovider test Count
                 passed += tc.getPassedTests().getAllResults().size();
                 failed += tc.getFailedTests().getAllResults().size();
-                failed += tc.getFailedTests().getAllResults().size();
+                skipped += tc.getSkippedTests().getAllResults().size();
+
+                //To skip Dataprovider test Count
+               /* passed += tc.getPassedTests().getAllMethods().size();
+                failed += tc.getFailedTests().getAllMethods().size();
+                skipped += tc.getSkippedTests().getAllMethods().size();*/
+
 
                 System.out.println("Passed tests for suite '" + suiteName +
-                        "' is:" + tc.getPassedTests().getAllResults().size());
+                        "' is:" + tc.getPassedTests().getAllMethods().size());
                 System.out.println("Failed tests for suite '" + suiteName +
                         "' is:" +
-                        tc.getFailedTests().getAllResults().size());
+                        tc.getFailedTests().getAllMethods().size());
                 System.out.println("Skipped tests for suite '" + suiteName +
                         "' is:" +
-                        tc.getFailedTests().getAllResults().size());
+                        tc.getFailedTests().getAllMethods().size());
                 System.out.println("Total excution time for test '" + tc.getName() +
                         "' is:" + (tc.getEndDate().getTime() - tc.getStartDate().getTime()));
             }
         }
+
         createResultTable(passed, failed, skipped);
         createChartGraph(passed, failed, skipped);
         createBrowserInfo();