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