def test_entity_manager_save(self, manager):
"""Test that we can save all of a manager's dirty entities."""
mock1 = Mock(is_savable=True, is_dirty=False)
SomeEntity._instances["mock1"] = mock1
mock2 = Mock(is_savable=True, is_dirty=False)
SomeEntity._instances["mock2"] = mock2
# First test with no dirty entities.
manager.save()
assert not mock1.save.called
assert not mock1.save.called
# Then test with a dirty entity.
mock1.is_dirty = True
manager.save()
assert mock1.save.called
assert not mock2.save.called
# Clean up the mock entities.
del SomeEntity._instances["mock1"]
del SomeEntity._instances["mock2"]
