add validation of broken transactions

Author: timgraham

django_mongodb_backend/compiler.py

@@ -685,6 +685,7 @@ def execute_sql(self, returning_fields=None):
     @wrap_database_errors
     def insert(self, docs, returning_fields=None):
         """Store a list of documents using field columns as element names."""
+        self.connection.validate_no_broken_transaction()
         inserted_ids = self.collection.insert_many(
             docs, session=self.connection.session
         ).inserted_ids
@@ -770,6 +771,7 @@ def execute_sql(self, result_type):
 
     @wrap_database_errors
     def update(self, criteria, pipeline):
+        self.connection.validate_no_broken_transaction()
         return self.collection.update_many(
             criteria, pipeline, session=self.connection.session
         ).matched_count

django_mongodb_backend/features.py

@@ -84,8 +84,6 @@ class DatabaseFeatures(BaseDatabaseFeatures):
         # Value.as_mql() doesn't call output_field.get_db_prep_save():
         # https://github.com/mongodb/django-mongodb-backend/issues/282
         "model_fields.test_jsonfield.TestSaveLoad.test_bulk_update_custom_get_prep_value",
-        # to debug
-        "transactions.tests.AtomicMiscTests.test_mark_for_rollback_on_error_in_transaction",
     }
     # $bitAnd, #bitOr, and $bitXor are new in MongoDB 6.3.
     _django_test_expected_failures_bitwise = {

django_mongodb_backend/query.py

@@ -61,6 +61,7 @@ def __repr__(self):
     @wrap_database_errors
     def delete(self):
         """Execute a delete query."""
+        self.compiler.connection.validate_no_broken_transaction()
         if self.compiler.subqueries:
             raise NotSupportedError("Cannot use QuerySet.delete() when a subquery is required.")
         return self.compiler.collection.delete_many(
@@ -73,6 +74,7 @@ def get_cursor(self):
         Return a pymongo CommandCursor that can be iterated on to give the
         results of the query.
         """
+        self.compiler.connection.validate_no_broken_transaction()
         return self.compiler.collection.aggregate(
             self.get_pipeline(), session=self.compiler.connection.session
         )