@@ -2910,32 +2910,33 @@ public void testPartitionedDmlDoesNotTimeout() {
2910
2910
DatabaseClientImpl dbImpl = ((DatabaseClientImpl ) client );
2911
2911
int channelId = dbImpl .getSession ().getChannel ();
2912
2912
int dbId = dbImpl .dbId ;
2913
+ long NON_DETERMINISTIC = XGoogSpannerRequestIdTest .NON_DETERMINISTIC ;
2913
2914
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {
2914
2915
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2915
2916
"google.spanner.v1.Spanner/ExecuteStreamingSql" ,
2916
- new XGoogSpannerRequestId (dbId , channelId , 5 , 1 )),
2917
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 5 , 1 )),
2917
2918
};
2918
2919
xGoogReqIdInterceptor .checkExpectedStreamingXGoogRequestIds (wantStreamingValues );
2919
2920
2920
2921
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantUnaryValues = {
2921
2922
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2922
2923
"google.spanner.v1.Spanner/BatchCreateSessions" ,
2923
- new XGoogSpannerRequestId (dbId , 0 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
2924
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 0 , NON_DETERMINISTIC , 1 )),
2924
2925
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2925
2926
"google.spanner.v1.Spanner/BatchCreateSessions" ,
2926
- new XGoogSpannerRequestId (dbId , 1 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
2927
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , NON_DETERMINISTIC , 1 )),
2927
2928
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2928
2929
"google.spanner.v1.Spanner/BatchCreateSessions" ,
2929
- new XGoogSpannerRequestId (dbId , 2 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
2930
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 2 , NON_DETERMINISTIC , 1 )),
2930
2931
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2931
2932
"google.spanner.v1.Spanner/BatchCreateSessions" ,
2932
- new XGoogSpannerRequestId (dbId , 3 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
2933
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 3 , NON_DETERMINISTIC , 1 )),
2933
2934
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2934
2935
"google.spanner.v1.Spanner/BeginTransaction" ,
2935
- new XGoogSpannerRequestId (dbId , channelId , 6 , 1 )),
2936
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 6 , 1 )),
2936
2937
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2937
2938
"google.spanner.v1.Spanner/ExecuteSql" ,
2938
- new XGoogSpannerRequestId (dbId , channelId , 7 , 1 )),
2939
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 7 , 1 )),
2939
2940
};
2940
2941
xGoogReqIdInterceptor .checkExpectedUnaryXGoogRequestIds (wantUnaryValues );
2941
2942
}
@@ -2976,30 +2977,36 @@ public void testPartitionedDmlWithLowerTimeout() {
2976
2977
DatabaseClientImpl dbImpl = ((DatabaseClientImpl ) client );
2977
2978
int channelId = dbImpl .getSession ().getChannel ();
2978
2979
int dbId = dbImpl .dbId ;
2979
- XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {};
2980
+ long NON_DETERMINISTIC = XGoogSpannerRequestIdTest .NON_DETERMINISTIC ;
2981
+ XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {
2982
+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2983
+ "google.spanner.v1.Spanner/ExecuteStreamingSql" ,
2984
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 5 , 1 )),
2985
+ };
2980
2986
xGoogReqIdInterceptor .checkExpectedStreamingXGoogRequestIds (wantStreamingValues );
2981
2987
2982
2988
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantUnaryValues = {
2983
2989
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2984
2990
"google.spanner.v1.Spanner/BatchCreateSessions" ,
2985
- new XGoogSpannerRequestId (dbId , 0 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
2991
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 0 , NON_DETERMINISTIC , 1 )),
2986
2992
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2987
2993
"google.spanner.v1.Spanner/BatchCreateSessions" ,
2988
- new XGoogSpannerRequestId (dbId , 1 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
2994
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , NON_DETERMINISTIC , 1 )),
2989
2995
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2990
2996
"google.spanner.v1.Spanner/BatchCreateSessions" ,
2991
- new XGoogSpannerRequestId (dbId , 2 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
2997
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 2 , NON_DETERMINISTIC , 1 )),
2992
2998
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2993
2999
"google.spanner.v1.Spanner/BatchCreateSessions" ,
2994
- new XGoogSpannerRequestId (dbId , 3 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
3000
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 3 , NON_DETERMINISTIC , 1 )),
2995
3001
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2996
3002
"google.spanner.v1.Spanner/BeginTransaction" ,
2997
- new XGoogSpannerRequestId (dbId , channelId , 6 , 1 )),
3003
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 6 , 1 )),
2998
3004
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2999
- "google.spanner.v1.Spanner/Commit" , new XGoogSpannerRequestId (dbId , channelId , 8 , 1 )),
3005
+ "google.spanner.v1.Spanner/Commit" ,
3006
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 8 , 1 )),
3000
3007
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3001
3008
"google.spanner.v1.Spanner/ExecuteSql" ,
3002
- new XGoogSpannerRequestId (dbId , channelId , 7 , 1 )),
3009
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 7 , 1 )),
3003
3010
};
3004
3011
xGoogReqIdInterceptor .checkExpectedUnaryXGoogRequestIds (wantUnaryValues );
3005
3012
}
@@ -3054,33 +3061,34 @@ public void testPartitionedDmlWithHigherTimeout() {
3054
3061
DatabaseClientImpl dbImpl = ((DatabaseClientImpl ) client );
3055
3062
int channelId = dbImpl .getSession ().getChannel ();
3056
3063
int dbId = dbImpl .dbId ;
3064
+ long NON_DETERMINISTIC = XGoogSpannerRequestIdTest .NON_DETERMINISTIC ;
3057
3065
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {
3058
3066
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3059
3067
"google.spanner.v1.Spanner/ExecuteStreamingSql" ,
3060
- new XGoogSpannerRequestId (dbId , channelId , 5 , 1 )),
3068
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 5 , 1 )),
3061
3069
};
3062
3070
3063
3071
xGoogReqIdInterceptor .checkExpectedStreamingXGoogRequestIds (wantStreamingValues );
3064
3072
3065
3073
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantUnaryValues = {
3066
3074
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3067
3075
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3068
- new XGoogSpannerRequestId (dbId , 0 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
3076
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 0 , NON_DETERMINISTIC , 1 )),
3069
3077
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3070
3078
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3071
- new XGoogSpannerRequestId (dbId , 1 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
3079
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , NON_DETERMINISTIC , 1 )),
3072
3080
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3073
3081
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3074
- new XGoogSpannerRequestId (dbId , 2 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
3082
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 2 , NON_DETERMINISTIC , 1 )),
3075
3083
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3076
3084
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3077
- new XGoogSpannerRequestId (dbId , 3 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
3085
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 3 , NON_DETERMINISTIC , 1 )),
3078
3086
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3079
3087
"google.spanner.v1.Spanner/BeginTransaction" ,
3080
- new XGoogSpannerRequestId (dbId , channelId , 6 , 1 )),
3088
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 6 , 1 )),
3081
3089
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3082
3090
"google.spanner.v1.Spanner/ExecuteSql" ,
3083
- new XGoogSpannerRequestId (dbId , channelId , 7 , 1 )),
3091
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 7 , 1 )),
3084
3092
};
3085
3093
xGoogReqIdInterceptor .checkExpectedUnaryXGoogRequestIds (wantUnaryValues );
3086
3094
}
@@ -3104,30 +3112,31 @@ public void testPartitionedDmlRetriesOnUnavailable() {
3104
3112
DatabaseClientImpl dbImpl = ((DatabaseClientImpl ) client );
3105
3113
int channelId = dbImpl .getSession ().getChannel ();
3106
3114
int dbId = dbImpl .dbId ;
3115
+ long NON_DETERMINISTIC = XGoogSpannerRequestIdTest .NON_DETERMINISTIC ;
3107
3116
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {
3108
3117
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3109
3118
"google.spanner.v1.Spanner/ExecuteStreamingSql" ,
3110
- new XGoogSpannerRequestId (dbId , channelId , 5 , 1 )),
3119
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 5 , 1 )),
3111
3120
};
3112
3121
3113
3122
xGoogReqIdInterceptor .checkExpectedStreamingXGoogRequestIds (wantStreamingValues );
3114
3123
3115
3124
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantUnaryValues = {
3116
3125
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3117
3126
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3118
- new XGoogSpannerRequestId (dbId , 0 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
3127
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 0 , NON_DETERMINISTIC , 1 )),
3119
3128
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3120
3129
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3121
- new XGoogSpannerRequestId (dbId , 1 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
3130
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , NON_DETERMINISTIC , 1 )),
3122
3131
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3123
3132
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3124
- new XGoogSpannerRequestId (dbId , 2 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
3133
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 2 , NON_DETERMINISTIC , 1 )),
3125
3134
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3126
3135
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3127
- new XGoogSpannerRequestId (dbId , 3 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
3136
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 3 , NON_DETERMINISTIC , 1 )),
3128
3137
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3129
3138
"google.spanner.v1.Spanner/BeginTransaction" ,
3130
- new XGoogSpannerRequestId (dbId , channelId , 6 , 1 )),
3139
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 6 , 1 )),
3131
3140
};
3132
3141
xGoogReqIdInterceptor .checkExpectedUnaryXGoogRequestIds (wantUnaryValues );
3133
3142
}
@@ -3540,8 +3549,11 @@ public void testNestedTransactionsUsingTwoDatabases() throws InterruptedExceptio
3540
3549
3541
3550
int channelId = client1 .getSession ().getChannel ();
3542
3551
int dbId = client1 .dbId ;
3543
- XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {};
3544
-
3552
+ XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {
3553
+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3554
+ "google.spanner.v1.Spanner/ExecuteStreamingSql" ,
3555
+ new XGoogSpannerRequestId (dbId , 1 , 5 , 1 )),
3556
+ };
3545
3557
xGoogReqIdInterceptor .checkExpectedStreamingXGoogRequestIds (wantStreamingValues );
3546
3558
3547
3559
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantUnaryValues = {
@@ -5375,20 +5387,66 @@ public void testRetryOnResourceExhausted() {
5375
5387
DatabaseClientImpl dbClient = (DatabaseClientImpl ) client ;
5376
5388
int channelId = dbClient .getSession ().getChannel ();
5377
5389
int dbId = dbClient .dbId ;
5390
+ long NON_DETERMINISTIC = XGoogSpannerRequestIdTest .NON_DETERMINISTIC ;
5378
5391
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {
5379
5392
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
5380
5393
"google.spanner.v1.Spanner/ExecuteStreamingSql" ,
5381
- new XGoogSpannerRequestId (dbId , 1 , 5 , 1 )),
5394
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , 5 , 1 )),
5395
+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
5396
+ "google.spanner.v1.Spanner/ExecuteStreamingSql" ,
5397
+ new XGoogSpannerRequestId (dbId , 1 , 6 , 1 )),
5398
+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
5399
+ "google.spanner.v1.Spanner/ExecuteStreamingSql" ,
5400
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , 7 , 1 )),
5401
+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
5402
+ "google.spanner.v1.Spanner/ExecuteStreamingSql" ,
5403
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , 8 , 1 )),
5404
+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
5405
+ "google.spanner.v1.Spanner/ExecuteStreamingSql" ,
5406
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , 5 , 1 )),
5407
+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
5408
+ "google.spanner.v1.Spanner/ExecuteStreamingSql" ,
5409
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , 5 , 1 )),
5410
+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
5411
+ "google.spanner.v1.Spanner/ExecuteStreamingSql" ,
5412
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , 5 , 1 )),
5413
+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
5414
+ "google.spanner.v1.Spanner/ExecuteStreamingSql" ,
5415
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , 5 , 1 )),
5416
+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
5417
+ "google.spanner.v1.Spanner/ExecuteStreamingSql" ,
5418
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , 5 , 1 )),
5419
+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
5420
+ "google.spanner.v1.Spanner/ExecuteStreamingSql" ,
5421
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , 5 , 1 )),
5422
+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
5423
+ "google.spanner.v1.Spanner/ExecuteStreamingSql" ,
5424
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , 5 , 1 )),
5425
+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
5426
+ "google.spanner.v1.Spanner/ExecuteStreamingSql" ,
5427
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , 5 , 1 )),
5428
+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
5429
+ "google.spanner.v1.Spanner/ExecuteStreamingSql" ,
5430
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , 5 , 1 )),
5431
+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
5432
+ "google.spanner.v1.Spanner/ExecuteStreamingSql" ,
5433
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , 5 , 1 )),
5434
+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
5435
+ "google.spanner.v1.Spanner/ExecuteStreamingSql" ,
5436
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , 5 , 1 )),
5437
+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
5438
+ "google.spanner.v1.Spanner/ExecuteStreamingSql" ,
5439
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , 5 , 1 )),
5382
5440
};
5383
5441
xGoogReqIdInterceptor .checkExpectedStreamingXGoogRequestIds (wantStreamingValues );
5384
5442
5385
5443
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantUnaryValues = {
5386
5444
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
5387
5445
"google.spanner.v1.Spanner/BatchCreateSessions" ,
5388
- new XGoogSpannerRequestId (dbId , 0 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
5446
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 0 , NON_DETERMINISTIC , 1 )),
5389
5447
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
5390
5448
"google.spanner.v1.Spanner/BatchCreateSessions" ,
5391
- new XGoogSpannerRequestId (dbId , 1 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
5449
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , NON_DETERMINISTIC , 1 )),
5392
5450
};
5393
5451
xGoogReqIdInterceptor .checkExpectedUnaryXGoogRequestIds (wantUnaryValues );
5394
5452
0 commit comments