def tearDown(self):
"""Undo all the state we have."""
rmtree(self.testdata_dir)
rmtree(self.virtualenv_dir)
DB.drop_tables(ORM_OBJECTS, cascade=True)
DB.drop_tables([MetadataSystem])
DB.close()
def tearDownClass(cls):
"""Close the database connection."""
DB.close()
def setUp(self):
"""Setup the database."""
self._models = self.dependent_cls()
DB.create_tables(self._models)
def tearDown(self):
"""Tear down the database."""
DB.drop_tables(self._models)
def setUpClass(cls):
"""Connect to the database."""
if not DB.is_closed(): # pragma: no cover this is just in case
DB.close()
DB.connect()