Sort features.py tests.

Author: WaVEV

django_mongodb/compiler.py

@@ -21,6 +21,10 @@ class SQLCompiler(compiler.SQLCompiler):
     query_class = MongoQuery
     SEPARATOR = "10__MESSI__3"
 
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self._group_pipeline = None
+
     def _get_group_alias_column(self, col, annotation_group_idx):
         """Generate alias and replacement for group columns."""
         replacement = None
@@ -182,8 +186,6 @@ def pre_sql_setup(self, with_col_aliases=False):
                     }
                 )
             self._group_pipeline = pipeline
-        else:
-            self._group_pipeline = None
 
         self.annotations = {
             target: expr.replace_expressions(all_replacements)
@@ -503,7 +505,6 @@ def insert(self, docs, returning_fields=None):
 
 class SQLDeleteCompiler(compiler.SQLDeleteCompiler, SQLCompiler):
     def execute_sql(self, result_type=MULTI):
-        self.pre_sql_setup()
         cursor = Cursor()
         cursor.rowcount = self.build_query().delete()
         return cursor

django_mongodb/features.py

@@ -142,6 +142,7 @@ def django_test_expected_failures(self):
             "expressions.tests.BasicExpressionsTests.test_object_create_with_aggregate",
             "expressions.tests.BasicExpressionsTests.test_object_create_with_f_expression_in_subquery",
             # PI()
+            "aggregation.tests.AggregateTestCase.test_aggregation_default_using_decimal_from_database",
             "db_functions.math.test_round.RoundTests.test_decimal_with_precision",
             "db_functions.math.test_round.RoundTests.test_float_with_precision",
         },
@@ -289,8 +290,13 @@ def django_test_expected_failures(self):
             "timezones.tests.NewDatabaseTests.test_query_annotation",
         },
         "Exists is not supported on MongoDB.": {
+            "aggregation.test_filter_argument.FilteredAggregateTests.test_filtered_aggregate_on_exists",
             "aggregation.test_filter_argument.FilteredAggregateTests.test_filtered_aggregate_ref_multiple_subquery_annotation",
             "annotations.tests.NonAggregateAnnotationTestCase.test_annotation_exists_none_query",
+            "aggregation.tests.AggregateTestCase.test_aggregation_exists_multivalued_outeref",
+            "aggregation.tests.AggregateTestCase.test_group_by_exists_annotation",
+            "aggregation.tests.AggregateTestCase.test_exists_none_with_aggregate",
+            "aggregation.tests.AggregateTestCase.test_exists_extra_where_with_aggregate",
             "delete_regress.tests.DeleteTests.test_self_reference_with_through_m2m_at_second_level",
             "expressions.tests.BasicExpressionsTests.test_annotation_with_deeply_nested_outerref",
             "expressions.tests.BasicExpressionsTests.test_boolean_expression_combined",
@@ -370,6 +376,9 @@ def django_test_expected_failures(self):
             "aggregation.test_filter_argument.FilteredAggregateTests.test_filtered_aggregate_ref_subquery_annotation",
         },
         "Using a QuerySet in annotate() is not supported on MongoDB.": {
+            "aggregation.tests.AggregateTestCase.test_filter_in_subquery_or_aggregation",
+            "aggregation.tests.AggregateTestCase.test_group_by_subquery_annotation",
+            "aggregation.tests.AggregateTestCase.test_group_by_reference_subquery",
             "annotations.tests.NonAggregateAnnotationTestCase.test_annotation_and_alias_filter_in_subquery",
             "annotations.tests.NonAggregateAnnotationTestCase.test_annotation_and_alias_filter_related_in_subquery",
             "annotations.tests.NonAggregateAnnotationTestCase.test_empty_expression_annotation",
@@ -443,6 +452,7 @@ def django_test_expected_failures(self):
             "one_to_one.tests.OneToOneTests.test_o2o_primary_key_delete",
         },
         "QuerySet.dates() is not supported on MongoDB.": {
+            "aggregation.tests.AggregateTestCase.test_dates_with_aggregation",
             "annotations.tests.AliasTests.test_dates_alias",
             "dates.tests.DatesTests.test_dates_trunc_datetime_fields",
             "dates.tests.DatesTests.test_related_model_traverse",
@@ -514,6 +524,9 @@ def django_test_expected_failures(self):
             "update.tests.AdvancedTests.test_update_annotated_multi_table_queryset",
         },
         "Test inspects query for SQL": {
+            "aggregation.tests.AggregateAnnotationPruningTests.test_unreferenced_aggregate_annotation_pruned",
+            "aggregation.tests.AggregateAnnotationPruningTests.test_unused_aliased_aggregate_pruned",
+            "aggregation.tests.AggregateAnnotationPruningTests.test_referenced_aggregate_annotation_kept",
             "aggregation.tests.AggregateTestCase.test_count_star",
             "delete.tests.DeletionTests.test_only_referenced_fields_selected",
             "lookup.tests.LookupTests.test_in_ignore_none",
@@ -623,35 +636,12 @@ def django_test_expected_failures(self):
         },
         "Test not applicable for MongoDB's SQLCompiler.": {
             "queries.test_iterator.QuerySetIteratorTests",
-        },
-        "skip agg": {
-            "aggregation.test_filter_argument.FilteredAggregateTests.test_filtered_aggregate_on_exists",
             # Custom aggregations:
             "aggregation.tests.AggregateAnnotationPruningTests.test_non_aggregate_annotation_pruned",
             "aggregation.tests.AggregateTestCase.test_add_implementation",
             "aggregation.tests.AggregateTestCase.test_aggregation_default_using_datetime_from_database",
             "aggregation.tests.AggregateTestCase.test_multi_arg_aggregate",
-            # SQL custom values.
             "aggregation.tests.AggregateTestCase.test_aggregation_default_using_date_from_database",
-            # No sql generate
-            "aggregation.tests.AggregateAnnotationPruningTests.test_unreferenced_aggregate_annotation_pruned",
-            "aggregation.tests.AggregateAnnotationPruningTests.test_unused_aliased_aggregate_pruned",
-            # PI expression:
-            "aggregation.tests.AggregateTestCase.test_aggregation_default_using_decimal_from_database",
-            # check sql query performance
-            "aggregation.tests.AggregateAnnotationPruningTests.test_referenced_aggregate_annotation_kept",
-            # Using a QuerySet in annotate() is not supported on MongoDB
-            "aggregation.tests.AggregateTestCase.test_group_by_subquery_annotation",
-            # Query set with annotate not supported.
-            "aggregation.tests.AggregateTestCase.test_group_by_reference_subquery",
-            "aggregation.tests.AggregateTestCase.test_filter_in_subquery_or_aggregation",
-            # Exists is not supported in mongodb
-            "aggregation.tests.AggregateTestCase.test_aggregation_exists_multivalued_outeref",
-            "aggregation.tests.AggregateTestCase.test_group_by_exists_annotation",
-            "aggregation.tests.AggregateTestCase.test_exists_none_with_aggregate",
-            "aggregation.tests.AggregateTestCase.test_exists_extra_where_with_aggregate",
-            # Queryset dates not supported
-            "aggregation.tests.AggregateTestCase.test_dates_with_aggregation",
         },
     }