56
56
import com .google .firestore .v1 .ListenResponse ;
57
57
import com .google .firestore .v1 .PartitionQueryRequest ;
58
58
import com .google .firestore .v1 .PartitionQueryResponse ;
59
+ import com .google .firestore .v1 .QueryMode ;
60
+ import com .google .firestore .v1 .ResultSetStats ;
59
61
import com .google .firestore .v1 .RollbackRequest ;
60
62
import com .google .firestore .v1 .RunAggregationQueryRequest ;
61
63
import com .google .firestore .v1 .RunAggregationQueryResponse ;
@@ -505,9 +507,14 @@ public void runQueryTest() throws Exception {
505
507
.setDocument (Document .newBuilder ().build ())
506
508
.setReadTime (Timestamp .newBuilder ().build ())
507
509
.setSkippedResults (880286183 )
510
+ .setStats (ResultSetStats .newBuilder ().build ())
508
511
.build ();
509
512
mockFirestore .addResponse (expectedResponse );
510
- RunQueryRequest request = RunQueryRequest .newBuilder ().setParent ("parent-995424086" ).build ();
513
+ RunQueryRequest request =
514
+ RunQueryRequest .newBuilder ()
515
+ .setParent ("parent-995424086" )
516
+ .setMode (QueryMode .forNumber (0 ))
517
+ .build ();
511
518
512
519
MockStreamObserver <RunQueryResponse > responseObserver = new MockStreamObserver <>();
513
520
@@ -523,7 +530,11 @@ public void runQueryTest() throws Exception {
523
530
public void runQueryExceptionTest () throws Exception {
524
531
StatusRuntimeException exception = new StatusRuntimeException (io .grpc .Status .INVALID_ARGUMENT );
525
532
mockFirestore .addException (exception );
526
- RunQueryRequest request = RunQueryRequest .newBuilder ().setParent ("parent-995424086" ).build ();
533
+ RunQueryRequest request =
534
+ RunQueryRequest .newBuilder ()
535
+ .setParent ("parent-995424086" )
536
+ .setMode (QueryMode .forNumber (0 ))
537
+ .build ();
527
538
528
539
MockStreamObserver <RunQueryResponse > responseObserver = new MockStreamObserver <>();
529
540
@@ -547,10 +558,14 @@ public void runAggregationQueryTest() throws Exception {
547
558
.setResult (AggregationResult .newBuilder ().build ())
548
559
.setTransaction (ByteString .EMPTY )
549
560
.setReadTime (Timestamp .newBuilder ().build ())
561
+ .setStats (ResultSetStats .newBuilder ().build ())
550
562
.build ();
551
563
mockFirestore .addResponse (expectedResponse );
552
564
RunAggregationQueryRequest request =
553
- RunAggregationQueryRequest .newBuilder ().setParent ("parent-995424086" ).build ();
565
+ RunAggregationQueryRequest .newBuilder ()
566
+ .setParent ("parent-995424086" )
567
+ .setMode (QueryMode .forNumber (0 ))
568
+ .build ();
554
569
555
570
MockStreamObserver <RunAggregationQueryResponse > responseObserver = new MockStreamObserver <>();
556
571
@@ -568,7 +583,10 @@ public void runAggregationQueryExceptionTest() throws Exception {
568
583
StatusRuntimeException exception = new StatusRuntimeException (io .grpc .Status .INVALID_ARGUMENT );
569
584
mockFirestore .addException (exception );
570
585
RunAggregationQueryRequest request =
571
- RunAggregationQueryRequest .newBuilder ().setParent ("parent-995424086" ).build ();
586
+ RunAggregationQueryRequest .newBuilder ()
587
+ .setParent ("parent-995424086" )
588
+ .setMode (QueryMode .forNumber (0 ))
589
+ .build ();
572
590
573
591
MockStreamObserver <RunAggregationQueryResponse > responseObserver = new MockStreamObserver <>();
574
592
0 commit comments