fixup! Merge pull request #1 from gurneyalex/15.0-database_cleanup

adhoc-cicd · May 3, 2022 · c012c31 · c012c31
1 parent fb5e7df
commit c012c31
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/database_cleanup/models/purge_tables.py b/database_cleanup/models/purge_tables.py
@@ -72,7 +72,9 @@ def purge(self):
                     )
 
             self.logger.info("Dropping table %s", line.name)
-            self.env.cr.execute("DROP TABLE %s CASCADE", (IdentifierAdapter(line.name),))
+            self.env.cr.execute(
+                "DROP TABLE %s CASCADE", (IdentifierAdapter(line.name),)
+            )
             line.write({"purged": True})
         return True
 

diff --git a/database_cleanup/tests/common.py b/database_cleanup/tests/common.py
@@ -3,8 +3,8 @@
 from contextlib import contextmanager
 
 import odoo
-from odoo.modules.registry import Registry
-from odoo.tests.common import tagged, TransactionCase
+from odoo.tests import common
+from odoo.tests.common import TransactionCase, tagged
 
 
 @contextmanager
@@ -13,13 +13,12 @@ def new_rollbacked_env():
     uid = odoo.SUPERUSER_ID
     cr = registry.cursor()
     try:
-        yield api.Environment(cr, uid, {})
+        yield odoo.api.Environment(cr, uid, {})
     finally:
         cr.rollback()  # we shouldn't have to commit anything
         cr.close()
 
 
-
 # Use post_install to get all models loaded more info: odoo/odoo#13458
 @tagged("post_install", "-at_install")
 class Common(TransactionCase):