@@ -56,6 +56,9 @@ public void testListenersAreExecuted() {
56
56
AtomicInteger preQuery = new AtomicInteger ();
57
57
AtomicInteger failedQuery = new AtomicInteger ();
58
58
AtomicInteger onQuery = new AtomicInteger ();
59
+ AtomicInteger preSlice = new AtomicInteger ();
60
+ AtomicInteger failedSlice = new AtomicInteger ();
61
+ AtomicInteger onSlice = new AtomicInteger ();
59
62
AtomicInteger onFetch = new AtomicInteger ();
60
63
AtomicInteger preFetch = new AtomicInteger ();
61
64
AtomicInteger failedFetch = new AtomicInteger ();
@@ -86,6 +89,24 @@ public void onQueryPhase(SearchContext searchContext, long tookInNanos) {
86
89
onQuery .incrementAndGet ();
87
90
}
88
91
92
+ @ Override
93
+ public void onPreSliceExecution (SearchContext searchContext ) {
94
+ assertNotNull (searchContext );
95
+ preSlice .incrementAndGet ();
96
+ }
97
+
98
+ @ Override
99
+ public void onFailedSliceExecution (SearchContext searchContext ) {
100
+ assertNotNull (searchContext );
101
+ failedSlice .incrementAndGet ();
102
+ }
103
+
104
+ @ Override
105
+ public void onSliceExecution (SearchContext searchContext ) {
106
+ assertNotNull (searchContext );
107
+ onSlice .incrementAndGet ();
108
+ }
109
+
89
110
@ Override
90
111
public void onPreFetchPhase (SearchContext searchContext ) {
91
112
assertNotNull (searchContext );
@@ -167,10 +188,30 @@ public void onSearchIdleReactivation() {
167
188
compositeListener .onQueryPhase (ctx , timeInNanos .get ());
168
189
assertEquals (0 , preFetch .get ());
169
190
assertEquals (0 , preQuery .get ());
191
+ assertEquals (0 , preSlice .get ());
192
+ assertEquals (0 , failedFetch .get ());
193
+ assertEquals (0 , failedQuery .get ());
194
+ assertEquals (0 , failedSlice .get ());
195
+ assertEquals (2 , onQuery .get ());
196
+ assertEquals (0 , onFetch .get ());
197
+ assertEquals (0 , onSlice .get ());
198
+ assertEquals (0 , newContext .get ());
199
+ assertEquals (0 , newScrollContext .get ());
200
+ assertEquals (0 , freeContext .get ());
201
+ assertEquals (0 , freeScrollContext .get ());
202
+ assertEquals (0 , searchIdleReactivateCount .get ());
203
+ assertEquals (0 , validateSearchContext .get ());
204
+
205
+ compositeListener .onSliceExecution (ctx );
206
+ assertEquals (0 , preFetch .get ());
207
+ assertEquals (0 , preQuery .get ());
208
+ assertEquals (0 , preSlice .get ());
170
209
assertEquals (0 , failedFetch .get ());
171
210
assertEquals (0 , failedQuery .get ());
211
+ assertEquals (0 , failedSlice .get ());
172
212
assertEquals (2 , onQuery .get ());
173
213
assertEquals (0 , onFetch .get ());
214
+ assertEquals (2 , onSlice .get ());
174
215
assertEquals (0 , newContext .get ());
175
216
assertEquals (0 , newScrollContext .get ());
176
217
assertEquals (0 , freeContext .get ());
@@ -181,10 +222,13 @@ public void onSearchIdleReactivation() {
181
222
compositeListener .onFetchPhase (ctx , timeInNanos .get ());
182
223
assertEquals (0 , preFetch .get ());
183
224
assertEquals (0 , preQuery .get ());
225
+ assertEquals (0 , preSlice .get ());
184
226
assertEquals (0 , failedFetch .get ());
185
227
assertEquals (0 , failedQuery .get ());
228
+ assertEquals (0 , failedSlice .get ());
186
229
assertEquals (2 , onQuery .get ());
187
230
assertEquals (2 , onFetch .get ());
231
+ assertEquals (2 , onSlice .get ());
188
232
assertEquals (0 , newContext .get ());
189
233
assertEquals (0 , newScrollContext .get ());
190
234
assertEquals (0 , freeContext .get ());
@@ -195,10 +239,30 @@ public void onSearchIdleReactivation() {
195
239
compositeListener .onPreQueryPhase (ctx );
196
240
assertEquals (0 , preFetch .get ());
197
241
assertEquals (2 , preQuery .get ());
242
+ assertEquals (0 , preSlice .get ());
198
243
assertEquals (0 , failedFetch .get ());
199
244
assertEquals (0 , failedQuery .get ());
245
+ assertEquals (0 , failedSlice .get ());
200
246
assertEquals (2 , onQuery .get ());
201
247
assertEquals (2 , onFetch .get ());
248
+ assertEquals (2 , onSlice .get ());
249
+ assertEquals (0 , newContext .get ());
250
+ assertEquals (0 , newScrollContext .get ());
251
+ assertEquals (0 , freeContext .get ());
252
+ assertEquals (0 , freeScrollContext .get ());
253
+ assertEquals (0 , searchIdleReactivateCount .get ());
254
+ assertEquals (0 , validateSearchContext .get ());
255
+
256
+ compositeListener .onPreSliceExecution (ctx );
257
+ assertEquals (0 , preFetch .get ());
258
+ assertEquals (2 , preQuery .get ());
259
+ assertEquals (2 , preSlice .get ());
260
+ assertEquals (0 , failedFetch .get ());
261
+ assertEquals (0 , failedQuery .get ());
262
+ assertEquals (0 , failedSlice .get ());
263
+ assertEquals (2 , onQuery .get ());
264
+ assertEquals (2 , onFetch .get ());
265
+ assertEquals (2 , onSlice .get ());
202
266
assertEquals (0 , newContext .get ());
203
267
assertEquals (0 , newScrollContext .get ());
204
268
assertEquals (0 , freeContext .get ());
@@ -209,10 +273,13 @@ public void onSearchIdleReactivation() {
209
273
compositeListener .onPreFetchPhase (ctx );
210
274
assertEquals (2 , preFetch .get ());
211
275
assertEquals (2 , preQuery .get ());
276
+ assertEquals (2 , preSlice .get ());
212
277
assertEquals (0 , failedFetch .get ());
213
278
assertEquals (0 , failedQuery .get ());
279
+ assertEquals (0 , failedSlice .get ());
214
280
assertEquals (2 , onQuery .get ());
215
281
assertEquals (2 , onFetch .get ());
282
+ assertEquals (2 , onSlice .get ());
216
283
assertEquals (0 , newContext .get ());
217
284
assertEquals (0 , newScrollContext .get ());
218
285
assertEquals (0 , freeContext .get ());
@@ -223,10 +290,13 @@ public void onSearchIdleReactivation() {
223
290
compositeListener .onFailedFetchPhase (ctx );
224
291
assertEquals (2 , preFetch .get ());
225
292
assertEquals (2 , preQuery .get ());
293
+ assertEquals (2 , preSlice .get ());
226
294
assertEquals (2 , failedFetch .get ());
227
295
assertEquals (0 , failedQuery .get ());
296
+ assertEquals (0 , failedSlice .get ());
228
297
assertEquals (2 , onQuery .get ());
229
298
assertEquals (2 , onFetch .get ());
299
+ assertEquals (2 , onSlice .get ());
230
300
assertEquals (0 , newContext .get ());
231
301
assertEquals (0 , newScrollContext .get ());
232
302
assertEquals (0 , freeContext .get ());
@@ -237,10 +307,30 @@ public void onSearchIdleReactivation() {
237
307
compositeListener .onFailedQueryPhase (ctx );
238
308
assertEquals (2 , preFetch .get ());
239
309
assertEquals (2 , preQuery .get ());
310
+ assertEquals (2 , preSlice .get ());
311
+ assertEquals (2 , failedFetch .get ());
312
+ assertEquals (2 , failedQuery .get ());
313
+ assertEquals (0 , failedSlice .get ());
314
+ assertEquals (2 , onQuery .get ());
315
+ assertEquals (2 , onFetch .get ());
316
+ assertEquals (2 , onSlice .get ());
317
+ assertEquals (0 , newContext .get ());
318
+ assertEquals (0 , newScrollContext .get ());
319
+ assertEquals (0 , freeContext .get ());
320
+ assertEquals (0 , freeScrollContext .get ());
321
+ assertEquals (0 , searchIdleReactivateCount .get ());
322
+ assertEquals (0 , validateSearchContext .get ());
323
+
324
+ compositeListener .onFailedSliceExecution (ctx );
325
+ assertEquals (2 , preFetch .get ());
326
+ assertEquals (2 , preQuery .get ());
327
+ assertEquals (2 , preSlice .get ());
240
328
assertEquals (2 , failedFetch .get ());
241
329
assertEquals (2 , failedQuery .get ());
330
+ assertEquals (2 , failedSlice .get ());
242
331
assertEquals (2 , onQuery .get ());
243
332
assertEquals (2 , onFetch .get ());
333
+ assertEquals (2 , onSlice .get ());
244
334
assertEquals (0 , newContext .get ());
245
335
assertEquals (0 , newScrollContext .get ());
246
336
assertEquals (0 , freeContext .get ());
@@ -251,10 +341,13 @@ public void onSearchIdleReactivation() {
251
341
compositeListener .onNewReaderContext (mock (ReaderContext .class ));
252
342
assertEquals (2 , preFetch .get ());
253
343
assertEquals (2 , preQuery .get ());
344
+ assertEquals (2 , preSlice .get ());
254
345
assertEquals (2 , failedFetch .get ());
255
346
assertEquals (2 , failedQuery .get ());
347
+ assertEquals (2 , failedSlice .get ());
256
348
assertEquals (2 , onQuery .get ());
257
349
assertEquals (2 , onFetch .get ());
350
+ assertEquals (2 , onSlice .get ());
258
351
assertEquals (2 , newContext .get ());
259
352
assertEquals (0 , newScrollContext .get ());
260
353
assertEquals (0 , freeContext .get ());
@@ -265,10 +358,13 @@ public void onSearchIdleReactivation() {
265
358
compositeListener .onNewScrollContext (mock (ReaderContext .class ));
266
359
assertEquals (2 , preFetch .get ());
267
360
assertEquals (2 , preQuery .get ());
361
+ assertEquals (2 , preSlice .get ());
268
362
assertEquals (2 , failedFetch .get ());
269
363
assertEquals (2 , failedQuery .get ());
364
+ assertEquals (2 , failedSlice .get ());
270
365
assertEquals (2 , onQuery .get ());
271
366
assertEquals (2 , onFetch .get ());
367
+ assertEquals (2 , onSlice .get ());
272
368
assertEquals (2 , newContext .get ());
273
369
assertEquals (2 , newScrollContext .get ());
274
370
assertEquals (0 , freeContext .get ());
@@ -279,10 +375,13 @@ public void onSearchIdleReactivation() {
279
375
compositeListener .onFreeReaderContext (mock (ReaderContext .class ));
280
376
assertEquals (2 , preFetch .get ());
281
377
assertEquals (2 , preQuery .get ());
378
+ assertEquals (2 , preSlice .get ());
282
379
assertEquals (2 , failedFetch .get ());
283
380
assertEquals (2 , failedQuery .get ());
381
+ assertEquals (2 , failedSlice .get ());
284
382
assertEquals (2 , onQuery .get ());
285
383
assertEquals (2 , onFetch .get ());
384
+ assertEquals (2 , onSlice .get ());
286
385
assertEquals (2 , newContext .get ());
287
386
assertEquals (2 , newScrollContext .get ());
288
387
assertEquals (2 , freeContext .get ());
@@ -293,10 +392,13 @@ public void onSearchIdleReactivation() {
293
392
compositeListener .onFreeScrollContext (mock (ReaderContext .class ));
294
393
assertEquals (2 , preFetch .get ());
295
394
assertEquals (2 , preQuery .get ());
395
+ assertEquals (2 , preSlice .get ());
296
396
assertEquals (2 , failedFetch .get ());
297
397
assertEquals (2 , failedQuery .get ());
398
+ assertEquals (2 , failedSlice .get ());
298
399
assertEquals (2 , onQuery .get ());
299
400
assertEquals (2 , onFetch .get ());
401
+ assertEquals (2 , onSlice .get ());
300
402
assertEquals (2 , newContext .get ());
301
403
assertEquals (2 , newScrollContext .get ());
302
404
assertEquals (2 , freeContext .get ());
@@ -307,10 +409,13 @@ public void onSearchIdleReactivation() {
307
409
compositeListener .onSearchIdleReactivation ();
308
410
assertEquals (2 , preFetch .get ());
309
411
assertEquals (2 , preQuery .get ());
412
+ assertEquals (2 , preSlice .get ());
310
413
assertEquals (2 , failedFetch .get ());
311
414
assertEquals (2 , failedQuery .get ());
415
+ assertEquals (2 , failedSlice .get ());
312
416
assertEquals (2 , onQuery .get ());
313
417
assertEquals (2 , onFetch .get ());
418
+ assertEquals (2 , onSlice .get ());
314
419
assertEquals (2 , newContext .get ());
315
420
assertEquals (2 , newScrollContext .get ());
316
421
assertEquals (2 , freeContext .get ());
0 commit comments