Add support to clear DB

Fixes bug 1171740

The method cleanup() can be called when a test is complete,
for example in the tearDown method.

Change-Id: Id30bcb0d7ef167f6cb3b1bdea06f6421e1234a86

1 files changed, 11 insertions, 0 deletions

diff --git a/openstack/common/db/sqlalchemy/session.py b/openstack/common/db/sqlalchemy/session.py
index 99439f4..a19fc01 100644
--- a/openstack/common/db/sqlalchemy/session.py
+++ b/openstack/common/db/sqlalchemy/session.py
@@ -318,6 +318,17 @@ def set_defaults(sql_connection, sqlite_db):
                      sqlite_db=sqlite_db)
 
 
+def cleanup():
+    global _ENGINE, _MAKER
+
+    if _MAKER:
+        _MAKER.close_all()
+        _MAKER = None
+    if _ENGINE:
+        _ENGINE.dispose()
+        _ENGINE = None
+
+
 class SqliteForeignKeysListener(PoolListener):
     """
     Ensures that the foreign key constraints are enforced in SQLite.