def test_render_mock_ddl(self): with mock_engine('self.engine') as stream: self.User.__table__.create(self.engine) text = stream.getvalue() assert 'CREATE TABLE user' in text assert 'PRIMARY KEY' in text
def test_render_mock_ddl(self, engine, User): # TODO: mock_engine doesn't seem to work with locally scoped variables. self.engine = engine with mock_engine('self.engine') as stream: User.__table__.create(self.engine) text = stream.getvalue() assert 'CREATE TABLE user' in text assert 'PRIMARY KEY' in text