@@ -2907,6 +2907,37 @@ public void testPartitionedDmlDoesNotTimeout() {
29072907 return null ;
29082908 }));
29092909 assertEquals (ErrorCode .DEADLINE_EXCEEDED , e .getErrorCode ());
2910+
2911+ DatabaseClientImpl dbImpl = ((DatabaseClientImpl ) client );
2912+ int channelId = 0 ;
2913+ try (Session session = dbImpl .getSession ()) {
2914+ channelId = ((PooledSessionFuture ) session ).getChannel ();
2915+ }
2916+ int dbId = dbImpl .dbId ;
2917+ long NON_DETERMINISTIC = XGoogSpannerRequestIdTest .NON_DETERMINISTIC ;
2918+ XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {
2919+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2920+ "google.spanner.v1.Spanner/ExecuteStreamingSql" ,
2921+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 6 , 1 )),
2922+ };
2923+ if (false ) { // TODO(@odeke-em): enable in next PRs.
2924+ xGoogReqIdInterceptor .checkExpectedStreamingXGoogRequestIds (wantStreamingValues );
2925+ }
2926+
2927+ XGoogSpannerRequestIdTest .MethodAndRequestId [] wantUnaryValues = {
2928+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2929+ "google.spanner.v1.Spanner/BeginTransaction" ,
2930+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 7 , 1 )),
2931+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2932+ "google.spanner.v1.Spanner/CreateSession" ,
2933+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 0 , 1 , 1 )),
2934+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2935+ "google.spanner.v1.Spanner/ExecuteSql" ,
2936+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 8 , 1 )),
2937+ };
2938+ if (false ) { // TODO(@odeke-em): enable in next PRs.
2939+ xGoogReqIdInterceptor .checkExpectedUnaryXGoogRequestIdsAsSuffixes (wantUnaryValues );
2940+ }
29102941 }
29112942 }
29122943
@@ -2989,6 +3020,38 @@ public void testPartitionedDmlWithHigherTimeout() {
29893020 .run (transaction -> transaction .executeUpdate (UPDATE_STATEMENT )));
29903021 assertThat (e .getErrorCode ()).isEqualTo (ErrorCode .DEADLINE_EXCEEDED );
29913022 assertThat (updateCount ).isEqualTo (UPDATE_COUNT );
3023+
3024+ DatabaseClientImpl dbImpl = ((DatabaseClientImpl ) client );
3025+ int channelId = 0 ;
3026+ try (Session session = dbImpl .getSession ()) {
3027+ channelId = ((PooledSessionFuture ) session ).getChannel ();
3028+ }
3029+ int dbId = dbImpl .dbId ;
3030+ long NON_DETERMINISTIC = XGoogSpannerRequestIdTest .NON_DETERMINISTIC ;
3031+ XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {
3032+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3033+ "google.spanner.v1.Spanner/ExecuteStreamingSql" ,
3034+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 6 , 1 )),
3035+ };
3036+
3037+ if (false ) { // TODO(@odeke-em): enable in next PRs.
3038+ xGoogReqIdInterceptor .checkExpectedStreamingXGoogRequestIds (wantStreamingValues );
3039+ }
3040+
3041+ XGoogSpannerRequestIdTest .MethodAndRequestId [] wantUnaryValues = {
3042+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3043+ "google.spanner.v1.Spanner/BeginTransaction" ,
3044+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 7 , 1 )),
3045+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3046+ "google.spanner.v1.Spanner/CreateSession" ,
3047+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 0 , 1 , 1 )),
3048+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3049+ "google.spanner.v1.Spanner/ExecuteSql" ,
3050+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 8 , 1 )),
3051+ };
3052+ if (false ) { // TODO(@odeke-em): enable in next PRs.
3053+ xGoogReqIdInterceptor .checkExpectedUnaryXGoogRequestIdsAsSuffixes (wantUnaryValues );
3054+ }
29923055 }
29933056 }
29943057
@@ -5314,6 +5377,26 @@ public void testSessionPoolExhaustedError_containsStackTraces() {
53145377 }
53155378 // Closing the transactions should return the sessions to the pool.
53165379 assertEquals (4 , pool .getNumberOfSessionsInPool ());
5380+
5381+ DatabaseClientImpl dbClient = (DatabaseClientImpl ) client ;
5382+ int channelId = 0 ;
5383+ try (Session session = dbClient .getSession ()) {
5384+ channelId = ((PooledSessionFuture ) session ).getChannel ();
5385+ }
5386+ int dbId = dbClient .dbId ;
5387+ XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {};
5388+
5389+ xGoogReqIdInterceptor .checkExpectedStreamingXGoogRequestIds (wantStreamingValues );
5390+ long NON_DETERMINISTIC = XGoogSpannerRequestIdTest .NON_DETERMINISTIC ;
5391+
5392+ XGoogSpannerRequestIdTest .MethodAndRequestId [] wantUnaryValues = {
5393+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
5394+ "google.spanner.v1.Spanner/CreateSession" ,
5395+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 0 , 1 , 1 )),
5396+ };
5397+ if (false ) { // TODO(@odeke-em): enable in next PRs.
5398+ xGoogReqIdInterceptor .checkExpectedUnaryXGoogRequestIdsAsSuffixes (wantUnaryValues );
5399+ }
53175400 }
53185401 }
53195402
0 commit comments