Add retries in TestImpersonation

trinodb · Mar 23, 2023 · c5d4993 · c5d4993
1 parent 1d400c3
commit c5d4993
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/testing/trino-product-tests/src/main/java/io/trino/tests/product/TestImpersonation.java b/testing/trino-product-tests/src/main/java/io/trino/tests/product/TestImpersonation.java
@@ -20,6 +20,7 @@
 import io.trino.tempto.ProductTest;
 import io.trino.tempto.hadoop.hdfs.HdfsClient;
 import io.trino.tempto.query.QueryExecutor;
+import io.trino.testng.services.Flaky;
 import org.testng.annotations.Test;
 
 import java.net.URI;
@@ -29,6 +30,8 @@
 import static io.trino.tests.product.TestGroups.HDFS_IMPERSONATION;
 import static io.trino.tests.product.TestGroups.HDFS_NO_IMPERSONATION;
 import static io.trino.tests.product.TestGroups.PROFILE_SPECIFIC_TESTS;
+import static io.trino.tests.product.utils.HadoopTestUtils.RETRYABLE_FAILURES_ISSUES;
+import static io.trino.tests.product.utils.HadoopTestUtils.RETRYABLE_FAILURES_MATCH;
 import static io.trino.tests.product.utils.QueryExecutors.connectToTrino;
 import static java.lang.String.format;
 import static org.testng.Assert.assertEquals;
@@ -66,13 +69,15 @@ public void cleanup()
     }
 
     @Test(groups = {HDFS_NO_IMPERSONATION, PROFILE_SPECIFIC_TESTS})
+    @Flaky(issue = RETRYABLE_FAILURES_ISSUES, match = RETRYABLE_FAILURES_MATCH)
     public void testHdfsImpersonationDisabled()
     {
         String tableName = "check_hdfs_impersonation_disabled";
         checkTableOwner(tableName, configuredHdfsUser, aliceExecutor);
     }
 
     @Test(groups = {HDFS_IMPERSONATION, PROFILE_SPECIFIC_TESTS})
+    @Flaky(issue = RETRYABLE_FAILURES_ISSUES, match = RETRYABLE_FAILURES_MATCH)
     public void testHdfsImpersonationEnabled()
     {
         String tableName = "check_hdfs_impersonation_enabled";