@@ -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