Make render_statement SA 1.0 compliant
This commit is contained in:
@@ -58,7 +58,7 @@ def render_statement(statement, bind=None):
|
||||
|
||||
if isinstance(statement, sa.orm.query.Query):
|
||||
if bind is None:
|
||||
bind = statement.session.get_bind(statement._mapper_zero_or_none())
|
||||
bind = statement.session.get_bind(statement._mapper_zero())
|
||||
|
||||
statement = statement.statement
|
||||
|
||||
|
@@ -17,7 +17,7 @@ class TestRender(TestCase):
|
||||
|
||||
self.User = User
|
||||
|
||||
def test_render_statement_query(self):
|
||||
def test_render_orm_query(self):
|
||||
query = self.session.query(self.User).filter_by(id=3)
|
||||
text = render_statement(query)
|
||||
|
||||
@@ -33,6 +33,12 @@ class TestRender(TestCase):
|
||||
assert 'FROM user' in text
|
||||
assert 'WHERE user.id = 3' in text
|
||||
|
||||
def test_render_statement_without_mapper(self):
|
||||
statement = sa.select([sa.text('1')])
|
||||
text = render_statement(statement, bind=self.session.bind)
|
||||
|
||||
assert 'SELECT 1' in text
|
||||
|
||||
def test_render_ddl(self):
|
||||
expression = 'self.User.__table__.create(engine)'
|
||||
stream = render_expression(expression, self.engine)
|
||||
|
Reference in New Issue
Block a user