20
20
import static com .google .firebase .firestore .testutil .IntegrationTestUtil .testCollectionWithDocs ;
21
21
import static com .google .firebase .firestore .testutil .IntegrationTestUtil .testFirestore ;
22
22
import static com .google .firebase .firestore .testutil .IntegrationTestUtil .waitFor ;
23
- import static com .google .firebase .firestore .testutil .TestUtil .assertSetEquals ;
24
23
import static com .google .firebase .firestore .testutil .TestUtil .expectError ;
25
24
import static com .google .firebase .firestore .testutil .TestUtil .map ;
26
25
import static java .util .Arrays .asList ;
32
31
33
32
import androidx .test .ext .junit .runners .AndroidJUnit4 ;
34
33
import com .google .android .gms .tasks .Task ;
34
+ import com .google .common .collect .Lists ;
35
35
import com .google .common .collect .Maps ;
36
36
import com .google .firebase .firestore .Query .Direction ;
37
37
import com .google .firebase .firestore .testutil .EventAccumulator ;
@@ -514,30 +514,30 @@ public void testQueriesCanUseNotEqualFilters() {
514
514
expectedDocsMap .remove ("i" );
515
515
516
516
QuerySnapshot snapshot = waitFor (collection .whereNotEqualTo ("zip" , 98101L ).get ());
517
- assertSetEquals ( expectedDocsMap .values (), querySnapshotToValues (snapshot ));
517
+ assertEquals ( Lists . newArrayList ( expectedDocsMap .values () ), querySnapshotToValues (snapshot ));
518
518
519
519
// With objects.
520
520
expectedDocsMap = Maps .newHashMap (allDocs );
521
521
expectedDocsMap .remove ("f" );
522
522
expectedDocsMap .remove ("h" );
523
523
expectedDocsMap .remove ("i" );
524
524
snapshot = waitFor (collection .whereNotEqualTo ("zip" , map ("code" , 500 )).get ());
525
- assertSetEquals ( expectedDocsMap .values (), querySnapshotToValues (snapshot ));
525
+ assertEquals ( Lists . newArrayList ( expectedDocsMap .values () ), querySnapshotToValues (snapshot ));
526
526
527
527
// With Null.
528
528
expectedDocsMap = Maps .newHashMap (allDocs );
529
529
expectedDocsMap .remove ("h" );
530
530
expectedDocsMap .remove ("i" );
531
531
snapshot = waitFor (collection .whereNotEqualTo ("zip" , null ).get ());
532
- assertSetEquals ( expectedDocsMap .values (), querySnapshotToValues (snapshot ));
532
+ assertEquals ( Lists . newArrayList ( expectedDocsMap .values () ), querySnapshotToValues (snapshot ));
533
533
534
534
// With NaN.
535
535
expectedDocsMap = Maps .newHashMap (allDocs );
536
536
expectedDocsMap .remove ("h" );
537
537
expectedDocsMap .remove ("i" );
538
538
expectedDocsMap .remove ("j" );
539
539
snapshot = waitFor (collection .whereNotEqualTo ("zip" , Double .NaN ).get ());
540
- assertSetEquals ( expectedDocsMap .values (), querySnapshotToValues (snapshot ));
540
+ assertEquals ( Lists . newArrayList ( expectedDocsMap .values () ), querySnapshotToValues (snapshot ));
541
541
}
542
542
543
543
// TODO(ne-queries): Re-enable once emulator support is added to CI.
@@ -647,14 +647,14 @@ public void testQueriesCanUseNotInFilters() {
647
647
648
648
QuerySnapshot snapshot =
649
649
waitFor (collection .whereNotIn ("zip" , asList (98101L , 98103L , asList (98101L , 98102L ))).get ());
650
- assertSetEquals ( expectedDocsMap .values (), querySnapshotToValues (snapshot ));
650
+ assertEquals ( Lists . newArrayList ( expectedDocsMap .values () ), querySnapshotToValues (snapshot ));
651
651
652
652
// With objects.
653
653
expectedDocsMap = Maps .newHashMap (allDocs );
654
654
expectedDocsMap .remove ("f" );
655
655
expectedDocsMap .remove ("h" );
656
656
snapshot = waitFor (collection .whereNotIn ("zip" , asList (map ("code" , 500L ))).get ());
657
- assertSetEquals ( expectedDocsMap .values (), querySnapshotToValues (snapshot ));
657
+ assertEquals ( Lists . newArrayList ( expectedDocsMap .values () ), querySnapshotToValues (snapshot ));
658
658
659
659
// With Null.
660
660
List <Object > nullArray = new ArrayList <>();
@@ -667,15 +667,15 @@ public void testQueriesCanUseNotInFilters() {
667
667
expectedDocsMap .remove ("h" );
668
668
expectedDocsMap .remove ("j" );
669
669
snapshot = waitFor (collection .whereNotIn ("zip" , asList (Double .NaN )).get ());
670
- assertSetEquals ( expectedDocsMap .values (), querySnapshotToValues (snapshot ));
670
+ assertEquals ( Lists . newArrayList ( expectedDocsMap .values () ), querySnapshotToValues (snapshot ));
671
671
672
672
// With NaN and a number.
673
673
expectedDocsMap = Maps .newHashMap (allDocs );
674
674
expectedDocsMap .remove ("a" );
675
675
expectedDocsMap .remove ("h" );
676
676
expectedDocsMap .remove ("j" );
677
677
snapshot = waitFor (collection .whereNotIn ("zip" , asList (Float .NaN , 98101L )).get ());
678
- assertSetEquals ( expectedDocsMap .values (), querySnapshotToValues (snapshot ));
678
+ assertEquals ( Lists . newArrayList ( expectedDocsMap .values () ), querySnapshotToValues (snapshot ));
679
679
}
680
680
681
681
// TODO(ne-queries): Re-enable once emulator support is added to CI.
0 commit comments