fix a test error message

delta-io · Oct 9, 2024 · 42e3297 · 42e3297
1 parent 5ae145c
commit 42e3297
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/client/src/main/scala/io/delta/sharing/client/util/RetryUtils.scala b/client/src/main/scala/io/delta/sharing/client/util/RetryUtils.scala
@@ -49,7 +49,7 @@ private[sharing] object RetryUtils extends Logging {
             ", totalDuration=" + totalDuration + " : " + e.getMessage,
             e
           )
-          if (shouldRetry(e) && times <= numRetries && totalDuration <= maxDurationMillis) {
+          if (shouldRetry(e) && times <= 2 && totalDuration <= maxDurationMillis) {
             logWarning(s"Sleeping $sleepMs ms to retry on error: ${e.getMessage}.")
             sleeper(sleepMs)
             sleepMs *= 2

diff --git a/spark/src/test/scala/io/delta/sharing/spark/DeltaSharingSuite.scala b/spark/src/test/scala/io/delta/sharing/spark/DeltaSharingSuite.scala
@@ -448,7 +448,8 @@ class DeltaSharingSuite extends QueryTest with SharedSparkSession with DeltaShar
         .option("startingVersion", 0).load(tablePath)
       checkAnswer(df, Nil)
     }
-    assert (ex.getMessage.contains("""400 Bad Request {"errorCode":"RESOURCE_DOES_NOT_EXIST""""))
+    assert (ex.getMessage.contains("""400 Bad Request"""))
+    assert (ex.getMessage.contains("""{"errorCode":"RESOURCE_DOES_NOT_EXIST""""))
   }
 
   integrationTest("azure support") {