From ecfa16c2584ded11d968c848dc477599502114fe Mon Sep 17 00:00:00 2001 From: Yuya Ebihara Date: Thu, 21 Sep 2023 17:48:46 +0900 Subject: [PATCH] Retry on "HTTP Response code: 504" in Databricks tests --- .../tests/product/deltalake/util/DeltaLakeTestUtils.java | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/testing/trino-product-tests/src/main/java/io/trino/tests/product/deltalake/util/DeltaLakeTestUtils.java b/testing/trino-product-tests/src/main/java/io/trino/tests/product/deltalake/util/DeltaLakeTestUtils.java index c3d9eb1ceec1..680aeca64c1e 100644 --- a/testing/trino-product-tests/src/main/java/io/trino/tests/product/deltalake/util/DeltaLakeTestUtils.java +++ b/testing/trino-product-tests/src/main/java/io/trino/tests/product/deltalake/util/DeltaLakeTestUtils.java @@ -49,7 +49,9 @@ public final class DeltaLakeTestUtils public static final String DATABRICKS_COMMUNICATION_FAILURE_ISSUE = "https://github.com/trinodb/trino/issues/14391"; @Language("RegExp") public static final String DATABRICKS_COMMUNICATION_FAILURE_MATCH = - "\\Q[Databricks][DatabricksJDBCDriver](500593) Communication link failure. Failed to connect to server. Reason: HTTP retry after response received with no Retry-After header, error: HTTP Response code: 503, Error message: Unknown."; + "\\Q[Databricks][DatabricksJDBCDriver](500593) Communication link failure. Failed to connect to server. Reason: \\E" + + "(HTTP retry after response received with no Retry-After header, error: HTTP Response code: 503|HTTP Response code: 504)" + + ", Error message: Unknown."; private static final RetryPolicy CONCURRENT_MODIFICATION_EXCEPTION_RETRY_POLICY = RetryPolicy.builder() .handleIf(throwable -> Throwables.getRootCause(throwable) instanceof ConcurrentModificationException) .handleIf(throwable -> throwable.getMessage() != null && throwable.getMessage().contains("Table being modified concurrently"))