def test_extended_people_exclude_hidden(session):
people = ExtendedPersonCollection(session)
assert people.exclude_hidden is False
person = people.add(first_name="Hans", last_name="Maulwurf")
assert people.query().count() == 1
people.exclude_hidden = True
assert people.query().count() == 1
person.is_hidden_from_public = True
assert people.query().count() == 0
person.is_hidden_from_public = False
assert people.query().count() == 1
def test_extended_people_filter(session):
people = ExtendedPersonCollection(session)
people.add(first_name="Hans", last_name="Maulwurf")
people.add(first_name="Waylon", last_name="Śmithers")
people.add(first_name="Lenny", last_name="leonard")
people.add(first_name="Carl", last_name="Çarlson")
ned = people.add(first_name="Ned", last_name="Flanders")
agencies = ExtendedAgencyCollection(session)
agencies.add_root(title="Police").add_person(ned.id, "Snitch")
agencies.add_root(title="Ħospital").add_person(ned.id, "Volunteer")
agencies.add_root(title="Moe's Tavern")
assert [p.last_name for p in people.query()
] == ["Çarlson", "Flanders", "leonard", "Maulwurf", "Śmithers"]
people = people.for_filter(letter="C")
assert [p.last_name for p in people.query()] == ['Çarlson']
people = people.for_filter(letter="c")
assert [p.last_name for p in people.query()] == ['Çarlson']
people = people.for_filter(letter="L")
assert [p.last_name for p in people.query()] == ['leonard']
people = people.for_filter(agency="Police")
assert [p.last_name for p in people.query()] == []
people = people.for_filter(letter=None)
assert [p.last_name for p in people.query()] == ['Flanders']
def test_extended_people_pagination(session):
people = ExtendedPersonCollection(session)
assert people.pages_count == 0
assert people.batch == []
for letter in ascii_uppercase:
people.add(first_name=letter.lower(), last_name=letter)
assert people.query().count() == 26
people = ExtendedPersonCollection(session)
assert people.subset_count == 26
assert people.pages_count == 2
assert len(people.batch) == 20
assert len(people.next.batch) == 6
assert len(people.page_by_index(1).batch) == 6
def test_extended_people(session):
people = ExtendedPersonCollection(session)
person = people.add(first_name="Hans", last_name="Maulwurf")
assert isinstance(person, ExtendedPerson)
assert isinstance(people.query().one(), ExtendedPerson)