from turbogears import database
import unittest


class DBTest(unittest.TestCase):
	model = None

	def setUp(self):
		if not self.model:
			self.model = get_model()
			if not self.model:
				raise "Unable to run database tests without a model"

		database.metadata.create_all(database.get_engine())
		database.session.clear()

	def tearDown(self):
		database.rollback_all()
		database.metadata.drop_all(database.get_engine())