def test_tracking_aggregate_on_add(test_domain):
uow = UnitOfWork()
uow.start()
test_domain.repository_for(User).add(User(name="John Doe"))
assert len(uow._seen) == 1
def test_session_is_destroyed_after_rollback(self, test_domain):
uow = UnitOfWork()
uow.start()
uow.rollback()
assert uow._sessions == {}
assert uow.in_progress is False
def test_uow_can_be_initiated_explicitly(self, test_domain):
uow = UnitOfWork()
assert uow is not None
assert uow.in_progress is False
uow.start()
assert uow.in_progress is True
def test_that_uow_can_be_started_manually(self, test_domain):
uow = UnitOfWork()
uow.start()
uow.commit() # `commit` should not raise exception
uow.start()
uow.rollback() # `rollback` should not raise exception
def test_tracking_aggregate_on_filtering(test_domain):
test_domain.repository_for(User).add(User(id=12, name="John Doe"))
test_domain.repository_for(User).add(User(id=13, name="Jane Doe"))
uow = UnitOfWork()
uow.start()
test_domain.repository_for(User)._dao.query.filter(name__contains="Doe").all()
assert len(uow._seen) == 2
assert all(isinstance(item, User) for item in uow._seen)
def test_tracking_aggregate_on_get(test_domain):
test_domain.repository_for(User).add(User(id=12, name="John Doe"))
uow = UnitOfWork()
uow.start()
test_domain.repository_for(User).get(12)
assert len(uow._seen) == 1
seen_item = next(iter(uow._seen))
assert isinstance(seen_item, User)
def test_tracking_aggregate_on_update(test_domain):
test_domain.repository_for(User).add(User(id=12, name="John Doe"))
user = test_domain.repository_for(User).get(12)
uow = UnitOfWork()
uow.start()
user.name = "Name Changed"
test_domain.repository_for(User).add(user)
assert len(uow._seen) == 1
seen_item = next(iter(uow._seen))
assert seen_item.name == "Name Changed"
def test_changed_objects_are_committed_as_part_of_one_transaction_on_explict_commit(
self, test_domain):
# Add a Person the database
repo = test_domain.repository_for(Person)
person_to_be_updated = self.persisted_person(test_domain)
person_to_be_deleted = self.persisted_person(test_domain)
repo.add(person_to_be_updated)
repo.add(person_to_be_deleted)
person_dao = test_domain.get_dao(Person)
# Initiate a UnitOfWork Session
uow = UnitOfWork()
uow.start()
repo_with_uow = test_domain.repository_for(Person)
# Create a new person object to be added
person_to_be_added = Person(first_name="John", last_name="Doe")
repo_with_uow.add(person_to_be_added)
# Update an existing Person record
person_to_be_updated.last_name = "FooBar"
repo_with_uow.add(person_to_be_updated)
# Remove an existing Person record
repo_with_uow.remove(person_to_be_deleted)
# Test that the underlying database is untouched
assert len(person_dao.outside_uow().query.all().items) == 2
assert (person_dao.outside_uow().get(person_to_be_updated.id).last_name
!= "FooBar")
assert person_dao.outside_uow().get(
person_to_be_deleted.id) is not None
uow.commit()
assert uow.in_progress is False
assert len(person_dao.query.all().items) == 2
assert person_dao.get(person_to_be_added.id) is not None
assert person_dao.get(person_to_be_updated.id).last_name == "FooBar"
with pytest.raises(ObjectNotFoundError):
person_dao.get(person_to_be_deleted.id)
def test_all_changes_are_discard_on_rollback(self, test_domain):
repo = test_domain.repository_for(Person)
person_to_be_updated = self.persisted_person(test_domain)
person_to_be_deleted = self.persisted_person(test_domain)
repo.add(person_to_be_updated)
repo.add(person_to_be_deleted)
person_dao = test_domain.get_dao(Person)
# Initiate a UnitOfWork Session
uow = UnitOfWork()
uow.start()
repo_with_uow = test_domain.repository_for(Person)
# Create a new person object to be added
person_to_be_added = Person(first_name="John", last_name="Doe")
repo_with_uow.add(person_to_be_added)
# Update an existing Person record
person_to_be_updated.last_name = "FooBar"
repo_with_uow.add(person_to_be_updated)
# Remove an existing Person record
repo_with_uow.remove(person_to_be_deleted)
# Test that the underlying database is untouched
assert len(person_dao.outside_uow().query.all().items) == 2
assert (person_dao.outside_uow().get(person_to_be_updated.id).last_name
!= "FooBar")
assert person_dao.outside_uow().get(
person_to_be_deleted.id) is not None
uow.rollback()
assert uow.in_progress is False
assert len(person_dao.query.all().items) == 2
assert person_dao.get(person_to_be_updated.id).last_name != "FooBar"
assert person_dao.get(person_to_be_deleted.id) is not None
