Last active
March 10, 2018 17:48
-
-
Save ProvoK/e664d1e468044eb01d356ad243b3fc49 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
@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