def test_add_same_slug(self, class_entity_repo, monkeypatch):
monkeypatch.setattr(MyEntity, 'slug', None)
session = class_entity_repo.session_factory()
ent0 = MyEntity(id=0)
ent0.slug = str(ent0.id)
ent1 = MyEntity(id=1)
ent1.slug = ent0.slug
session.add(MyEntity, ent0)
session.add(MyEntity , ent1)
ents = session.get_by_slug(MyEntity, '0')
assert len(ents) == 2
def test_find_added_with_none_slug_by_slug(self, class_entity_repo,
monkeypatch):
monkeypatch.setattr(MyEntity, 'slug', None)
session = class_entity_repo.session_factory()
ent1 = MyEntity()
session.add(MyEntity, ent1)
ent1.slug = 'testslug'
ents = session.get_by_slug(MyEntity, ent1.slug)
assert not ents is None
assert ent1.id == ents[0].id
