openstack
/
neutron
Code Issues Proposed changes

Merge "Delete DB records instead of tables to speedup UT"

This commit is contained in:
Jenkins 2014-09-23 22:47:40 +00:00 committed by Gerrit Code Review
parent d3dfbf3e50 34cf04a0ea
commit 7dc54f67d2
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
neutron/tests/unit/testlib_api.py
View File

@ -55,17 +55,24 @@ def create_request(path, body, content_type, method='GET',
class SqlTestCase(base.BaseTestCase):
# flag to indicate that the models have been loaded
_TABLES_ESTABLISHED = False
def setUp(self):
super(SqlTestCase, self).setUp()
# Register all data models
engine = db_api.get_engine()
model_base.BASEV2.metadata.create_all(engine)
if not SqlTestCase._TABLES_ESTABLISHED:
model_base.BASEV2.metadata.create_all(engine)
SqlTestCase._TABLES_ESTABLISHED = True
def unregister_models():
"""Unregister all data models."""
model_base.BASEV2.metadata.drop_all(engine)
def clear_tables():
with engine.begin() as conn:
for table in reversed(
model_base.BASEV2.metadata.sorted_tables):
conn.execute(table.delete())
self.addCleanup(unregister_models)
self.addCleanup(clear_tables)
class WebTestCase(SqlTestCase):