Add create_tables test option

This commit is contained in:
Konsta Vesterinen
2014-04-22 13:02:58 +03:00
parent 5e42d32a84
commit ac6cc592d2
3 changed files with 14 additions and 3 deletions

View File

@@ -4,6 +4,12 @@ Changelog
Here you can see the full list of changes between each SQLAlchemy-Utils release.
0.25.4 (2014-04-22)
^^^^^^^^^^^^^^^^^^^
- Added ExpressionParser
0.25.3 (2014-04-21)
^^^^^^^^^^^^^^^^^^^

View File

@@ -34,16 +34,18 @@ def get_locale():
class TestCase(object):
dns = 'sqlite:///:memory:'
create_tables = True
def setup_method(self, method):
self.engine = create_engine(self.dns)
# self.engine.echo = True
self.engine.echo = True
self.connection = self.engine.connect()
self.Base = declarative_base()
self.create_models()
sa.orm.configure_mappers()
self.Base.metadata.create_all(self.connection)
if self.create_tables:
self.Base.metadata.create_all(self.connection)
Session = sessionmaker(bind=self.connection)
self.session = Session()
@@ -53,7 +55,8 @@ class TestCase(object):
def teardown_method(self, method):
aggregates.manager.reset()
self.session.close_all()
self.Base.metadata.drop_all(self.connection)
if self.create_tables:
self.Base.metadata.drop_all(self.connection)
self.connection.close()
self.engine.dispose()

View File

@@ -14,6 +14,8 @@ class MyExpressionParser(ExpressionParser):
class TestExpressionParser(TestCase):
create_tables = False
def setup_method(self, method):
TestCase.setup_method(self, method)
self.parser = MyExpressionParser(self.Category)