2020import static org .junit .Assert .assertEquals ;
2121import static org .junit .Assert .assertFalse ;
2222import static org .junit .Assert .assertTrue ;
23+ import static org .junit .Assume .assumeFalse ;
2324
2425import com .google .cloud .spanner .MockSpannerServiceImpl ;
2526import com .google .cloud .spanner .ResultSet ;
@@ -149,6 +150,9 @@ public boolean isEnableExtendedTracing() {
149150
150151 try (Connection connection = createTestConnection (getBaseUrl ())) {
151152 connection .setAutocommit (true );
153+ assumeFalse (
154+ "OpenTelemetryTracingTest handler is not implemented for read-write with multiplexed sessions" ,
155+ isMultiplexedSessionsEnabledForRW (connection .getSpanner ()));
152156 try (ResultSet resultSet = connection .executeQuery (SELECT1_STATEMENT )) {
153157 assertTrue (resultSet .next ());
154158 assertFalse (resultSet .next ());
@@ -185,6 +189,9 @@ public boolean isEnableExtendedTracing() {
185189 public void testSingleUseQuery () {
186190 try (Connection connection = createTestConnection ()) {
187191 connection .setAutocommit (true );
192+ assumeFalse (
193+ "OpenTelemetryTracingTest handler is not implemented for read-write with multiplexed sessions" ,
194+ isMultiplexedSessionsEnabledForRW (connection .getSpanner ()));
188195 try (ResultSet resultSet = connection .executeQuery (SELECT1_STATEMENT )) {
189196 assertTrue (resultSet .next ());
190197 assertFalse (resultSet .next ());
@@ -220,6 +227,9 @@ public void testSingleUseUpdate() {
220227 try (Connection connection = createTestConnection ()) {
221228 connection .setAutocommit (true );
222229 connection .executeUpdate (INSERT_STATEMENT );
230+ assumeFalse (
231+ "OpenTelemetryTracingTest handler is not implemented for read-write with multiplexed sessions" ,
232+ isMultiplexedSessionsEnabledForRW (connection .getSpanner ()));
223233 }
224234 assertEquals (CompletableResultCode .ofSuccess (), spanExporter .flush ());
225235 List <SpanData > spans = spanExporter .getFinishedSpanItems ();
@@ -256,6 +266,9 @@ public void testSingleUseBatchUpdate() {
256266 connection .executeUpdate (INSERT_STATEMENT );
257267 connection .executeUpdate (INSERT_STATEMENT );
258268 connection .runBatch ();
269+ assumeFalse (
270+ "OpenTelemetryTracingTest handler is not implemented for read-write with multiplexed sessions" ,
271+ isMultiplexedSessionsEnabledForRW (connection .getSpanner ()));
259272 }
260273 assertEquals (CompletableResultCode .ofSuccess (), spanExporter .flush ());
261274 List <SpanData > spans = spanExporter .getFinishedSpanItems ();
@@ -297,6 +310,9 @@ public void testSingleUseDdl() {
297310
298311 try (Connection connection = createTestConnection ()) {
299312 connection .setAutocommit (true );
313+ assumeFalse (
314+ "OpenTelemetryTracingTest handler is not implemented for read-write with multiplexed sessions" ,
315+ isMultiplexedSessionsEnabledForRW (connection .getSpanner ()));
300316 connection .execute (Statement .of (ddl ));
301317 }
302318 assertEquals (CompletableResultCode .ofSuccess (), spanExporter .flush ());
@@ -315,6 +331,9 @@ public void testSingleUseDdlBatch() {
315331
316332 try (Connection connection = createTestConnection ()) {
317333 connection .setAutocommit (true );
334+ assumeFalse (
335+ "OpenTelemetryTracingTest handler is not implemented for read-write with multiplexed sessions" ,
336+ isMultiplexedSessionsEnabledForRW (connection .getSpanner ()));
318337 connection .startBatchDdl ();
319338 connection .execute (Statement .of (ddl1 ));
320339 connection .execute (Statement .of (ddl2 ));
@@ -332,6 +351,9 @@ public void testSingleUseDdlBatch() {
332351 public void testMultiUseReadOnlyQueries () {
333352 try (Connection connection = createTestConnection ()) {
334353 connection .setAutocommit (false );
354+ assumeFalse (
355+ "OpenTelemetryTracingTest handler is not implemented for read-write with multiplexed sessions" ,
356+ isMultiplexedSessionsEnabledForRW (connection .getSpanner ()));
335357 connection .setReadOnly (true );
336358 twice (
337359 () -> {
@@ -363,6 +385,9 @@ public void testMultiUseReadOnlyQueries() {
363385 public void testMultiUseReadWriteQueries () {
364386 try (Connection connection = createTestConnection ()) {
365387 connection .setAutocommit (false );
388+ assumeFalse (
389+ "OpenTelemetryTracingTest handler is not implemented for read-write with multiplexed sessions" ,
390+ isMultiplexedSessionsEnabledForRW (connection .getSpanner ()));
366391 connection .setReadOnly (false );
367392 twice (
368393 () -> {
@@ -397,6 +422,9 @@ public void testMultiUseReadWriteQueries() {
397422 public void testMultiUseReadWriteUpdates () {
398423 try (Connection connection = createTestConnection ()) {
399424 connection .setAutocommit (false );
425+ assumeFalse (
426+ "OpenTelemetryTracingTest handler is not implemented for read-write with multiplexed sessions" ,
427+ isMultiplexedSessionsEnabledForRW (connection .getSpanner ()));
400428 connection .setReadOnly (false );
401429 assertEquals (1L , connection .executeUpdate (INSERT_STATEMENT ));
402430 assertEquals (1L , connection .executeUpdate (INSERT_STATEMENT ));
@@ -426,6 +454,9 @@ public void testMultiUseReadWriteUpdates() {
426454 public void testMultiUseReadWriteBatchUpdates () {
427455 try (Connection connection = createTestConnection ()) {
428456 connection .setAutocommit (false );
457+ assumeFalse (
458+ "OpenTelemetryTracingTest handler is not implemented for read-write with multiplexed sessions" ,
459+ isMultiplexedSessionsEnabledForRW (connection .getSpanner ()));
429460 connection .setReadOnly (false );
430461
431462 twice (
@@ -466,6 +497,9 @@ public void testMultiUseReadWriteBatchUpdates() {
466497 public void testMultiUseReadWriteAborted () {
467498 try (Connection connection = createTestConnection ()) {
468499 connection .setAutocommit (false );
500+ assumeFalse (
501+ "OpenTelemetryTracingTest handler is not implemented for read-write with multiplexed sessions" ,
502+ isMultiplexedSessionsEnabledForRW (connection .getSpanner ()));
469503 connection .setReadOnly (false );
470504 assertEquals (1L , connection .executeUpdate (INSERT_STATEMENT ));
471505 mockSpanner .abortNextStatement ();
@@ -514,6 +548,9 @@ public void testSavepoint() {
514548
515549 try (Connection connection = createTestConnection ()) {
516550 connection .setAutocommit (false );
551+ assumeFalse (
552+ "OpenTelemetryTracingTest handler is not implemented for read-write with multiplexed sessions" ,
553+ isMultiplexedSessionsEnabledForRW (connection .getSpanner ()));
517554 connection .setReadOnly (false );
518555 connection .setSavepointSupport (SavepointSupport .ENABLED );
519556 assertEquals (1L , connection .executeUpdate (statement1 ));
@@ -563,6 +600,9 @@ public void testTransactionTag() {
563600 try (Connection connection = createTestConnection ()) {
564601 connection .setAutocommit (false );
565602 connection .setReadOnly (false );
603+ assumeFalse (
604+ "OpenTelemetryTracingTest handler is not implemented for read-write with multiplexed sessions" ,
605+ isMultiplexedSessionsEnabledForRW (connection .getSpanner ()));
566606 connection .setTransactionTag ("my_tag" );
567607 assertEquals (1L , connection .executeUpdate (INSERT_STATEMENT ));
568608 connection .commit ();
0 commit comments