[CALCITE-2115] Populate sales_fact_1997, sales_fact_1998 for MongoAdapterTest

Author: tuichenchuxin

mongodb/src/test/java/org/apache/calcite/adapter/mongodb/MongoAdapterTest.java

@@ -104,6 +104,12 @@ public static void setUp() throws Exception {
     populate(database.getCollection("warehouse"),
         requireNonNull(FoodmartJson.class.getResource("/warehouse.json"),
             "url"));
+    populate(database.getCollection("sales_fact_1997"),
+            requireNonNull(FoodmartJson.class.getResource("/sales_fact_1997.json"),
+             "url"));
+    populate(database.getCollection("sales_fact_1998"),
+            requireNonNull(FoodmartJson.class.getResource("/sales_fact_1998.json"),
+             "url"));
 
     // Manually insert data for data-time test.
     MongoCollection<BsonDocument> datatypes = database.getCollection("datatypes")
@@ -284,7 +290,6 @@ private CalciteAssert.AssertThat assertModel(URL url) {
             "CITY=LAWTON; LONGITUDE=null; LATITUDE=null; POP=45542; STATE=OK; ID=73505");
   }
 
-  @Disabled("broken; [CALCITE-2115] is logged to fix it")
   @Test void testUnionPlan() {
     assertModel(MODEL)
         .query("select * from \"sales_fact_1997\"\n"
@@ -303,8 +308,6 @@ private CalciteAssert.AssertThat assertModel(URL url) {
                 "product_id=337", "product_id=1512"));
   }
 
-  @Disabled(
-      "java.lang.ClassCastException: java.lang.Integer cannot be cast to java.lang.Double")
   @Test void testFilterUnionPlan() {
     assertModel(MODEL)
         .query("select * from (\n"
@@ -584,13 +587,19 @@ private CalciteAssert.AssertThat assertModel(URL url) {
                 "{$project: {C: 1, STATE: 1, CITY: 1}}"));
   }
 
-  @Disabled("broken; [CALCITE-2115] is logged to fix it")
   @Test void testDistinctCount() {
     assertModel(MODEL)
         .query("select state, count(distinct city) as cdc from zips\n"
             + "where state in ('CA', 'TX') group by state order by state")
-        .returns("STATE=CA; CDC=1072\n"
-            + "STATE=TX; CDC=1233\n")
+        .returns("STATE=CA; CDC=3\n"
+            + "STATE=TX; CDC=3\n")
+        .explainContains("PLAN=EnumerableSort(sort0=[$0], dir0=[ASC])\n"
+            + "  EnumerableAggregate(group=[{0}], CDC=[COUNT($1)])\n"
+            + "    EnumerableAggregate(group=[{0, 1}])\n"
+            + "      MongoToEnumerableConverter\n"
+            + "        MongoProject(STATE=[CAST(ITEM($0, 'state')):VARCHAR(2)], CITY=[CAST(ITEM($0, 'city')):VARCHAR(20)])\n"
+            + "          MongoFilter(condition=[SEARCH(CAST(CAST(ITEM($0, 'state')):VARCHAR(2)):CHAR(2), Sarg['CA', 'TX']:CHAR(2))])\n"
+            + "            MongoTableScan(table=[[mongo_raw, zips]])\n\n")
         .queryContains(
             mongoChecker(
                 "{\n"
@@ -605,12 +614,7 @@ private CalciteAssert.AssertThat assertModel(URL url) {
                     + "    ]\n"
                     + "  }\n"
                     + "}",
-                "{$project: {CITY: '$city', STATE: '$state'}}",
-                "{$group: {_id: {CITY: '$CITY', STATE: '$STATE'}}}",
-                "{$project: {_id: 0, CITY: '$_id.CITY', STATE: '$_id.STATE'}}",
-                "{$group: {_id: '$STATE', CDC: {$sum: {$cond: [ {$eq: ['CITY', null]}, 0, 1]}}}}",
-                "{$project: {STATE: '$_id', CDC: '$CDC'}}",
-                "{$sort: {STATE: 1}}"));
+                "{$project: {STATE: '$state', CITY: '$city'}}"));
   }
 
   @Test void testDistinctCountOrderBy() {
@@ -636,18 +640,16 @@ private CalciteAssert.AssertThat assertModel(URL url) {
                 "{$limit: 5}"));
   }
 
-  @Disabled("broken; [CALCITE-2115] is logged to fix it")
   @Test void testProject() {
     assertModel(MODEL)
         .query("select state, city, 0 as zero from zips order by state, city")
         .limit(2)
-        .returns("STATE=AK; CITY=AKHIOK; ZERO=0\n"
-            + "STATE=AK; CITY=AKIACHAK; ZERO=0\n")
+        .returns("STATE=AK; CITY=ANCHORAGE; ZERO=0\n"
+            + "STATE=AK; CITY=FAIRBANKS; ZERO=0\n")
         .queryContains(
             mongoChecker(
-                "{$project: {CITY: '$city', STATE: '$state'}}",
-                "{$sort: {STATE: 1, CITY: 1}}",
-                "{$project: {STATE: 1, CITY: 1, ZERO: {$literal: 0}}}"));
+                "{$project: {STATE: '$state', CITY: '$city', 'ZERO': {$literal: 0}}}",
+                "{$sort: {STATE: 1, CITY: 1}}"));
   }
 
   @Test void testFilter() {