test(spanner): update non-retryable error code in unit tests (#3176)

googleapis · Jun 25, 2024 · b8b5aed · b8b5aed
1 parent 89c09ce
commit b8b5aed
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/AsyncRunnerTest.java b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/AsyncRunnerTest.java
@@ -229,7 +229,7 @@ public void asyncRunnerUpdateAbortedWithoutGettingResult() throws Exception {
   public void asyncRunnerCommitFails() throws Exception {
     mockSpanner.setCommitExecutionTime(
         SimulatedExecutionTime.ofException(
-            Status.RESOURCE_EXHAUSTED
+            Status.INVALID_ARGUMENT
                 .withDescription("mutation limit exceeded")
                 .asRuntimeException()));
     AsyncRunner runner = client().runAsync();
@@ -245,7 +245,7 @@ public void asyncRunnerCommitFails() throws Exception {
     ExecutionException e = assertThrows(ExecutionException.class, () -> updateCount.get());
     assertThat(e.getCause()).isInstanceOf(SpannerException.class);
     SpannerException se = (SpannerException) e.getCause();
-    assertThat(se.getErrorCode()).isEqualTo(ErrorCode.RESOURCE_EXHAUSTED);
+    assertThat(se.getErrorCode()).isEqualTo(ErrorCode.INVALID_ARGUMENT);
     assertThat(se.getMessage()).contains("mutation limit exceeded");
   }
 
@@ -432,7 +432,7 @@ public void asyncRunnerBatchUpdateAbortedWithoutGettingResult() throws Exception
   public void asyncRunnerWithBatchUpdateCommitFails() throws Exception {
     mockSpanner.setCommitExecutionTime(
         SimulatedExecutionTime.ofException(
-            Status.RESOURCE_EXHAUSTED
+            Status.INVALID_ARGUMENT
                 .withDescription("mutation limit exceeded")
                 .asRuntimeException()));
     AsyncRunner runner = client().runAsync();
@@ -448,7 +448,7 @@ public void asyncRunnerWithBatchUpdateCommitFails() throws Exception {
     ExecutionException e = assertThrows(ExecutionException.class, () -> updateCount.get());
     assertThat(e.getCause()).isInstanceOf(SpannerException.class);
     SpannerException se = (SpannerException) e.getCause();
-    assertThat(se.getErrorCode()).isEqualTo(ErrorCode.RESOURCE_EXHAUSTED);
+    assertThat(se.getErrorCode()).isEqualTo(ErrorCode.INVALID_ARGUMENT);
     assertThat(se.getMessage()).contains("mutation limit exceeded");
   }
 

diff --git a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/AsyncTransactionManagerTest.java b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/AsyncTransactionManagerTest.java
@@ -531,7 +531,7 @@ public void asyncTransactionManagerUpdateAbortedWithoutGettingResult() throws Ex
   public void asyncTransactionManagerCommitFails() throws Exception {
     mockSpanner.setCommitExecutionTime(
         SimulatedExecutionTime.ofException(
-            Status.RESOURCE_EXHAUSTED
+            Status.INVALID_ARGUMENT
                 .withDescription("mutation limit exceeded")
                 .asRuntimeException()));
     try (AsyncTransactionManager mgr = client().transactionManagerAsync()) {
@@ -545,7 +545,7 @@ public void asyncTransactionManagerCommitFails() throws Exception {
                               AsyncTransactionManagerHelper.executeUpdateAsync(UPDATE_STATEMENT),
                               executor)
                           .commitAsync()));
-      assertThat(e.getErrorCode()).isEqualTo(ErrorCode.RESOURCE_EXHAUSTED);
+      assertThat(e.getErrorCode()).isEqualTo(ErrorCode.INVALID_ARGUMENT);
       assertThat(e.getMessage()).contains("mutation limit exceeded");
     }
   }
@@ -928,7 +928,7 @@ public void asyncTransactionManagerBatchUpdateAbortedWithoutGettingResult() thro
   public void asyncTransactionManagerWithBatchUpdateCommitFails() throws Exception {
     mockSpanner.setCommitExecutionTime(
         SimulatedExecutionTime.ofException(
-            Status.RESOURCE_EXHAUSTED
+            Status.INVALID_ARGUMENT
                 .withDescription("mutation limit exceeded")
                 .asRuntimeException()));
     try (AsyncTransactionManager manager = clientWithEmptySessionPool().transactionManagerAsync()) {
@@ -945,7 +945,7 @@ public void asyncTransactionManagerWithBatchUpdateCommitFails() throws Exception
                                       ImmutableList.of(UPDATE_STATEMENT, UPDATE_STATEMENT)),
                               executor)
                           .commitAsync()));
-      assertThat(e.getErrorCode()).isEqualTo(ErrorCode.RESOURCE_EXHAUSTED);
+      assertThat(e.getErrorCode()).isEqualTo(ErrorCode.INVALID_ARGUMENT);
       assertThat(e.getMessage()).contains("mutation limit exceeded");
     }
     if (isMultiplexedSessionsEnabled()) {