Fixes the unit tests but is wrong?

jmchilton · Jan 22, 2024 · bcc01b8 · bcc01b8
1 parent 2238ee7
commit bcc01b8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/galaxy/quota/__init__.py b/lib/galaxy/quota/__init__.py
@@ -271,9 +271,9 @@ def relabel_quota_for_dataset(self, dataset, from_label: Optional[str], to_label
         bind = {"dataset_id": dataset.id, "adjust": int(adjust), "to_label": to_label, "from_label": from_label}
         engine = self.sa_session.get_bind()
         with engine.connect() as conn:
-            conn.execute(text(from_statement), bind)
-            conn.execute(text(to_statement), bind)
-            return None
+            with conn.begin():
+                conn.execute(text(from_statement), bind)
+                conn.execute(text(to_statement), bind)
 
     def _default_unregistered_quota(self, quota_source_label):
         return self._default_quota(self.model.DefaultQuotaAssociation.types.UNREGISTERED, quota_source_label)