From c02ecb3563c5486ef28c5c814251db9cd966af62 Mon Sep 17 00:00:00 2001 From: Piotr Findeisen Date: Thu, 23 Mar 2023 13:16:58 +0100 Subject: [PATCH] Add retries in TestImpersonation --- .../main/java/io/trino/tests/product/TestImpersonation.java | 5 +++++ 1 file changed, 5 insertions(+) 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 index 857064506633..7ab1f6673c50 100644 --- 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,6 +69,7 @@ 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"; @@ -73,6 +77,7 @@ public void testHdfsImpersonationDisabled() } @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";