@@ -3601,9 +3601,16 @@ public void testBackendPartitionQueryOptions() {
36013601 // statistics package and directed read options.
36023602 List <AbstractMessage > requests = mockSpanner .getRequests ();
36033603 assert requests .size () >= 2 : "required to have at least 2 requests" ;
3604- assertThat (requests .get (requests .size () - 1 )).isInstanceOf (DeleteSessionRequest .class );
3605- assertThat (requests .get (requests .size () - 2 )).isInstanceOf (ExecuteSqlRequest .class );
3606- ExecuteSqlRequest executeSqlRequest = (ExecuteSqlRequest ) requests .get (requests .size () - 2 );
3604+ if (isMultiplexedSessionsEnabled ()) {
3605+ assertThat (requests .get (requests .size () - 1 )).isInstanceOf (ExecuteSqlRequest .class );
3606+ } else {
3607+ assertThat (requests .get (requests .size () - 1 )).isInstanceOf (DeleteSessionRequest .class );
3608+ assertThat (requests .get (requests .size () - 2 )).isInstanceOf (ExecuteSqlRequest .class );
3609+ }
3610+ ExecuteSqlRequest executeSqlRequest =
3611+ (ExecuteSqlRequest )
3612+ requests .get (
3613+ isMultiplexedSessionsEnabled () ? requests .size () - 1 : requests .size () - 2 );
36073614 assertThat (executeSqlRequest .getQueryOptions ()).isNotNull ();
36083615 assertThat (executeSqlRequest .getQueryOptions ().getOptimizerVersion ()).isEqualTo ("1" );
36093616 assertThat (executeSqlRequest .getQueryOptions ().getOptimizerStatisticsPackage ())
@@ -3651,9 +3658,16 @@ public void testBackendPartitionQueryOptions() {
36513658 // statistics package and directed read options.
36523659 List <AbstractMessage > requests = mockSpanner .getRequests ();
36533660 assert requests .size () >= 2 : "required to have at least 2 requests" ;
3654- assertThat (requests .get (requests .size () - 1 )).isInstanceOf (DeleteSessionRequest .class );
3655- assertThat (requests .get (requests .size () - 2 )).isInstanceOf (ExecuteSqlRequest .class );
3656- ExecuteSqlRequest executeSqlRequest = (ExecuteSqlRequest ) requests .get (requests .size () - 2 );
3661+ if (isMultiplexedSessionsEnabled ()) {
3662+ assertThat (requests .get (requests .size () - 1 )).isInstanceOf (ExecuteSqlRequest .class );
3663+ } else {
3664+ assertThat (requests .get (requests .size () - 1 )).isInstanceOf (DeleteSessionRequest .class );
3665+ assertThat (requests .get (requests .size () - 2 )).isInstanceOf (ExecuteSqlRequest .class );
3666+ }
3667+ ExecuteSqlRequest executeSqlRequest =
3668+ (ExecuteSqlRequest )
3669+ requests .get (
3670+ isMultiplexedSessionsEnabled () ? requests .size () - 1 : requests .size () - 2 );
36573671 assertThat (executeSqlRequest .getQueryOptions ()).isNotNull ();
36583672 assertThat (executeSqlRequest .getQueryOptions ().getOptimizerVersion ()).isEqualTo ("1" );
36593673 assertThat (executeSqlRequest .getQueryOptions ().getOptimizerStatisticsPackage ())
@@ -3697,9 +3711,16 @@ public void testBackendPartitionReadOptions() {
36973711 // statistics package and directed read options.
36983712 List <AbstractMessage > requests = mockSpanner .getRequests ();
36993713 assert requests .size () >= 2 : "required to have at least 2 requests" ;
3700- assertThat (requests .get (requests .size () - 1 )).isInstanceOf (DeleteSessionRequest .class );
3701- assertThat (requests .get (requests .size () - 2 )).isInstanceOf (ReadRequest .class );
3702- ReadRequest readRequest = (ReadRequest ) requests .get (requests .size () - 2 );
3714+ if (isMultiplexedSessionsEnabled ()) {
3715+ assertThat (requests .get (requests .size () - 1 )).isInstanceOf (ReadRequest .class );
3716+ } else {
3717+ assertThat (requests .get (requests .size () - 1 )).isInstanceOf (DeleteSessionRequest .class );
3718+ assertThat (requests .get (requests .size () - 2 )).isInstanceOf (ReadRequest .class );
3719+ }
3720+ ReadRequest readRequest =
3721+ (ReadRequest )
3722+ requests .get (
3723+ isMultiplexedSessionsEnabled () ? requests .size () - 1 : requests .size () - 2 );
37033724 assertThat (readRequest .getDirectedReadOptions ()).isEqualTo (DIRECTED_READ_OPTIONS1 );
37043725 }
37053726 }
@@ -3740,9 +3761,16 @@ public void testBackendPartitionReadOptions() {
37403761 // statistics package and directed read options.
37413762 List <AbstractMessage > requests = mockSpanner .getRequests ();
37423763 assert requests .size () >= 2 : "required to have at least 2 requests" ;
3743- assertThat (requests .get (requests .size () - 1 )).isInstanceOf (DeleteSessionRequest .class );
3744- assertThat (requests .get (requests .size () - 2 )).isInstanceOf (ReadRequest .class );
3745- ReadRequest readRequest = (ReadRequest ) requests .get (requests .size () - 2 );
3764+ if (isMultiplexedSessionsEnabled ()) {
3765+ assertThat (requests .get (requests .size () - 1 )).isInstanceOf (ReadRequest .class );
3766+ } else {
3767+ assertThat (requests .get (requests .size () - 1 )).isInstanceOf (DeleteSessionRequest .class );
3768+ assertThat (requests .get (requests .size () - 2 )).isInstanceOf (ReadRequest .class );
3769+ }
3770+ ReadRequest readRequest =
3771+ (ReadRequest )
3772+ requests .get (
3773+ isMultiplexedSessionsEnabled () ? requests .size () - 1 : requests .size () - 2 );
37463774 assertThat (readRequest .getDirectedReadOptions ()).isEqualTo (DIRECTED_READ_OPTIONS2 );
37473775 }
37483776 }
0 commit comments