Skip to content

Instantly share code, notes, and snippets.

@ProvoK
Last active March 10, 2018 17:48
Show Gist options
  • Save ProvoK/e664d1e468044eb01d356ad243b3fc49 to your computer and use it in GitHub Desktop.
Save ProvoK/e664d1e468044eb01d356ad243b3fc49 to your computer and use it in GitHub Desktop.
@pytest.fixture(scope='function')
def session(connection):
transaction = connection.begin()
session = Session(bind=connection)
UserFactory._meta.sqlalchemy_session = session # NB: This line added
yield session
session.close()
transaction.rollback()
def my_func_to_delete_user(session, user_id):
session.query(UserModel).filter(UserModel.id == user_id).delete()
def test_case(session):
user = UserFactory.create()
assert session.query(UserModel).one()
my_func_to_delete_user(session, user.id)
result = session.query(UserModel).one_or_none()
assert result is None
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment