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

Author: harshachinta

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");
  }
 

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()) {