add retry test

baev · baev · commit 8225ead07f49 · 2017-03-01T17:38:00.000+03:00
diff --git a/allure-testng/src/test/java/io/qameta/allure/samples/RetryTest.java b/allure-testng/src/test/java/io/qameta/allure/samples/RetryTest.java
@@ -0,0 +1,28 @@
+package io.qameta.allure.samples;
+
+import org.testng.ITestResult;
+import org.testng.annotations.Test;
+import org.testng.util.RetryAnalyzerCount;
+
+/**
+ * @author charlie (Dmitry Baev).
+ */
+public class RetryTest {
+
+    @Test(retryAnalyzer = Retry.class)
+    public void testWithRetry() throws Exception {
+        throw new RuntimeException("Unexpected failure");
+    }
+
+    public static class Retry extends RetryAnalyzerCount {
+
+        @Override
+        public boolean retryMethod(ITestResult result) {
+            boolean willRetry = !result.isSuccess();
+            if (willRetry) {
+                result.setAttribute("retry", true);
+            }
+            return willRetry;
+        }
+    }
+}
diff --git a/allure-testng/src/test/java/io/qameta/allure/test/FeatureCombinationsTest.java b/allure-testng/src/test/java/io/qameta/allure/test/FeatureCombinationsTest.java
@@ -400,6 +400,14 @@ public void bddAnnotationsTest() throws Exception {
                 );
     }
 
+    @Test
+    public void retryTest() throws Exception {
+        runTestNgSuites("suites/retry.xml");
+        List<TestResult> testResults = results.getTestResults();
+        assertThat(testResults)
+                .hasSize(2);
+    }
+
     private Predicate<TestResult> hasLinks() {
         return testResult -> !testResult.getLinks().isEmpty();
     }
diff --git a/allure-testng/src/test/resources/suites/retry.xml b/allure-testng/src/test/resources/suites/retry.xml
@@ -0,0 +1,10 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE suite SYSTEM "http://testng.org/testng-1.0.dtd" >
+
+<suite name="Retry suite">
+    <test name="Retry tests">
+        <classes>
+            <class name="io.qameta.allure.samples.RetryTest"/>
+        </classes>
+    </test>
+</suite>

Original file line number	Diff line number	Diff line change
`@@ -400,6 +400,14 @@ public void bddAnnotationsTest() throws Exception {`
`400`	`400`	`);`
`401`	`401`	`}`
`402`	`402`
	`403`	`+ @Test`
	`404`	`+ public void retryTest() throws Exception {`
	`405`	`+ runTestNgSuites("suites/retry.xml");`
	`406`	`+ List<TestResult> testResults = results.getTestResults();`
	`407`	`+ assertThat(testResults)`
	`408`	`+ .hasSize(2);`
	`409`	`+ }`
	`410`	`+`
`403`	`411`	`private Predicate<TestResult> hasLinks() {`
`404`	`412`	`return testResult -> !testResult.getLinks().isEmpty();`
`405`	`413`	`}`