fix: Reuse DataProvider container for retries and handle suppressed failures

- Changed dataProviderContainerUuidStorage to use computeIfAbsent to reuse the existing container for a test method if it already exists (e.g. during retry).
- Added currentExecutable.remove() in beforeDataProviderExecution to force generation of a new UUID for each execution attempt, ensuring retries are captured as distinct fixtures even if TestNG suppresses failure events or we missed cleanup.
- Added FlakyDataProvider sample and test case to verify retry reporting.

Author: baflQA

allure-testng/src/main/java/io/qameta/allure/testng/AllureTestNg.java

@@ -649,13 +649,19 @@ public void onConfigurationSkip(final ITestResult itr) {
     public void beforeDataProviderExecution(final IDataProviderMethod dataProviderMethod,
                                             final ITestNGMethod method,
                                             final ITestContext iTestContext) {
-        final String containerUuid = UUID.randomUUID().toString();
-        getLifecycle().startTestContainer(
-                new TestResultContainer()
-                        .setUuid(containerUuid)
-                        .setName(method.getMethodName())
+        currentExecutable.remove();
+        final String containerUuid = dataProviderContainerUuidStorage.computeIfAbsent(
+                method,
+                key -> {
+                    final String uuid = UUID.randomUUID().toString();
+                    getLifecycle().startTestContainer(
+                            new TestResultContainer()
+                                    .setUuid(uuid)
+                                    .setName(method.getMethodName())
+                    );
+                    return uuid;
+                }
         );
-        dataProviderContainerUuidStorage.put(method, containerUuid);
 
         final String uuid = currentExecutable.get();
         final FixtureResult result = new FixtureResult()

allure-testng/src/test/java/io/qameta/allure/testng/AllureTestNgTest.java

@@ -1545,6 +1545,20 @@ public void shouldProcessFailedDataProviderInSetup() {
                 .contains(Tuple.tuple("dataProvider", Status.BROKEN));
     }
 
+    @AllureFeatures.Fixtures
+    @Test(description = "Should process flaky data provider in setup")
+    public void shouldProcessFlakyDataProvider() {
+        final AllureResults results = runTestNgSuites("suites/flaky-data-provider.xml");
+
+        assertThat(results.getTestResultContainers())
+                .flatExtracting(TestResultContainer::getBefores)
+                .extracting(FixtureResult::getName, FixtureResult::getStatus)
+                .containsSubsequence(
+                        Tuple.tuple("provide", Status.BROKEN),
+                        Tuple.tuple("provide", Status.PASSED)
+                );
+    }
+
     private Integer getOrderParameter(final TestResult result) {
         return result.getParameters().stream()
                 .filter(p -> p.getName().equals("order"))

allure-testng/src/test/java/io/qameta/allure/testng/samples/FlakyDataProvider.java

@@ -0,0 +1,39 @@
+package io.qameta.allure.testng.samples;
+
+import org.testng.IDataProviderMethod;
+import org.testng.IRetryDataProvider;
+import org.testng.ITestResult;
+import org.testng.annotations.BeforeClass;
+import org.testng.annotations.DataProvider;
+import org.testng.annotations.Test;
+
+import java.util.concurrent.atomic.AtomicInteger;
+
+public class FlakyDataProvider {
+
+    private static final AtomicInteger COUNTER = new AtomicInteger(0);
+
+    @BeforeClass
+    public void reset() {
+        COUNTER.set(0);
+    }
+
+    public static class Retry implements IRetryDataProvider {
+        @Override
+        public boolean retry(IDataProviderMethod method) {
+            return COUNTER.incrementAndGet() < 2;
+        }
+    }
+
+    @DataProvider(retryUsing = Retry.class)
+    public Object[][] provide() {
+        if (COUNTER.get() == 0) {
+            throw new RuntimeException("Simulated DataProvider Failure");
+        }
+        return new Object[][]{{"data"}};
+    }
+
+    @Test(dataProvider = "provide")
+    public void test(String s) {
+    }
+}

allure-testng/src/test/resources/suites/flaky-data-provider.xml

@@ -0,0 +1,8 @@
+<!DOCTYPE suite SYSTEM "http://testng.org/testng-1.0.dtd" >
+<suite name="Flaky Data Provider Suite" verbose="1" parallel="false" thread-count="1">
+    <test name="Flaky Data Provider Test">
+        <classes>
+            <class name="io.qameta.allure.testng.samples.FlakyDataProvider" />
+        </classes>
+    </test>
+</suite>