diff --git a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/it/ITBatchDmlTest.java b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/it/ITBatchDmlTest.java index a080d1c3968..f45fa2c427c 100644 --- a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/it/ITBatchDmlTest.java +++ b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/it/ITBatchDmlTest.java @@ -195,10 +195,7 @@ public long[] run(TransactionContext transaction) { runner.run(callable); Assert.fail("Expecting an exception."); } catch (SpannerBatchUpdateException e) { - // TODO: Remove if-statement when emulator returns the same error code as Cloud Spanner. - if (!env.getTestHelper().isEmulator()) { - assertThat(e.getErrorCode()).isEqualTo(ErrorCode.ALREADY_EXISTS); - } + assertThat(e.getErrorCode()).isEqualTo(ErrorCode.ALREADY_EXISTS); long[] rowCounts = e.getUpdateCounts(); assertThat(rowCounts.length).isEqualTo(1); for (long rc : rowCounts) { diff --git a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/it/ITBatchReadTest.java b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/it/ITBatchReadTest.java index a6aa4f715d6..9c3f11d3eeb 100644 --- a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/it/ITBatchReadTest.java +++ b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/it/ITBatchReadTest.java @@ -17,7 +17,6 @@ package com.google.cloud.spanner.it; import static com.google.common.truth.Truth.assertThat; -import static org.junit.Assume.assumeFalse; import com.google.cloud.ByteArray; import com.google.cloud.Timestamp; @@ -89,9 +88,6 @@ private static List manyRows() { @BeforeClass public static void setUpDatabase() throws Exception { - assumeFalse( - "BatchReadOnlyTransactions are not supported on the emulator", - env.getTestHelper().isEmulator()); db = env.getTestHelper() .createTestDatabase( diff --git a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/it/ITDMLTest.java b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/it/ITDMLTest.java index 9af4a22f76b..aabf93b3a6c 100644 --- a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/it/ITDMLTest.java +++ b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/it/ITDMLTest.java @@ -18,7 +18,6 @@ import static com.google.common.truth.Truth.assertThat; import static org.junit.Assert.fail; -import static org.junit.Assume.assumeFalse; import com.google.cloud.spanner.AbortedException; import com.google.cloud.spanner.Database; @@ -137,8 +136,6 @@ public void abortOnceShouldSucceedAfterRetry() { @Test public void partitionedDML() { - assumeFalse("The emulator does not support partitioned DML", env.getTestHelper().isEmulator()); - executeUpdate(DML_COUNT, insertDml()); assertThat( client