def test_membership_form_choices(session):
people = ExtendedPersonCollection(session)
people.add(first_name="Nick", last_name="Rivera")
people.add(first_name="Nick", last_name="Rivera", phone="1234")
people.add(first_name="Nick", last_name="Rivera", phone="5555", email="x")
people.add(first_name="Nick", last_name="Rivera", phone_direct="4")
people.add(first_name="Nick", last_name="Rivera", address="Street")
people.add(first_name="Nick", last_name="Rivera", email="*****@*****.**")
agencies = ExtendedAgencyCollection(session)
agency = agencies.add_root(title="Hospital")
doc = people.add(first_name="Nick", last_name="Rivera")
agency.add_person(doc.id, "Doc")
request = DummyRequest(session)
form = MembershipForm()
form.request = request
form.on_request()
assert sorted([x[1] for x in form.person_id.choices]) == [
'Rivera Nick',
'Rivera Nick (1234)',
'Rivera Nick (4)',
'Rivera Nick (5555)',
'Rivera Nick (Hospital)',
'Rivera Nick (Street)',
'Rivera Nick ([email protected])'
]
def test_extended_people_used_agencies(session):
assert ExtendedPersonCollection(session).used_agencies == []
agencies = ExtendedAgencyCollection(session)
police = agencies.add_root(title="Police")
hospital = agencies.add_root(title="Ħospital")
agencies.add_root(title="Moe's Tavern")
assert ExtendedPersonCollection(session).used_agencies == []
people = ExtendedPersonCollection(session)
ned = people.add(first_name="Ned", last_name="Flanders")
police.add_person(ned.id, "Snitch")
hospital.add_person(ned.id, "Volunteer")
assert people.used_agencies == ['Ħospital', 'Police']
def test_agency_move(session):
# test URL template
move = AgencyMove(None, None, None, None).for_url_template()
assert move.direction == '{direction}'
assert move.subject_id == '{subject_id}'
assert move.target_id == '{target_id}'
# test execute
collection = ExtendedAgencyCollection(session)
collection.add_root(title='2', id=2, order=2)
collection.add_root(title='1', id=1, order=1)
parent = collection.add_root(title='3', id=3, order=3)
collection.add(parent=parent, title='5', id=5, order=2)
collection.add(parent=parent, title='4', id=4, order=1)
def tree():
return [
[o.title, [c.title for c in o.children]]
for o in collection.query().filter_by(parent_id=None)
]
assert tree() == [['1', []], ['2', []], ['3', ['4', '5']]]
AgencyMove(session, 1, 2, 'below').execute()
assert tree() == [['2', []], ['1', []], ['3', ['4', '5']]]
AgencyMove(session, 3, 1, 'above').execute()
assert tree() == [['2', []], ['3', ['4', '5']], ['1', []]]
AgencyMove(session, 5, 4, 'above').execute()
session.flush()
session.expire_all()
assert tree() == [['2', []], ['3', ['5', '4']], ['1', []]]
# invalid
AgencyMove(session, 8, 9, 'above').execute()
assert tree() == [['2', []], ['3', ['5', '4']], ['1', []]]
AgencyMove(session, 5, 2, 'above').execute()
session.expire_all()
assert tree() == [['2', []], ['3', ['5', '4']], ['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_move_agency_form(session):
agencies = ExtendedAgencyCollection(session)
agency_a = agencies.add_root(title="a")
agencies.add(title="a.1", parent=agency_a)
agency_a_2 = agencies.add(title="a.2", parent=agency_a)
agency_a_2_1 = agencies.add(title="a.2.1", parent=agency_a_2)
agency_f = agencies.add_root(title="f")
agencies.add_root(title="ç")
# request
form = MoveAgencyForm()
form.request = DummyRequest(session)
form.on_request()
assert form.parent_id.choices == [
('root', '- Root -'),
('1', 'a'), ('2', 'a.1'), ('3', 'a.2'), ('4', 'a.2.1'),
('6', 'ç'), ('5', 'f')
]
# apply
form.apply_model(agency_a)
assert form.parent_id.choices == [
('6', 'ç'), ('5', 'f')
]
form.on_request()
form.apply_model(agency_a_2)
assert form.parent_id.choices == [
('root', '- Root -'),
('2', 'a.1'),
('6', 'ç'), ('5', 'f')
]
form.on_request()
form.apply_model(agency_a_2_1)
assert form.parent_id.choices == [
('root', '- Root -'),
('1', 'a'), ('2', 'a.1'),
('6', 'ç'), ('5', 'f')
]
form.on_request()
form.apply_model(agency_f)
assert form.parent_id.choices == [
('1', 'a'), ('2', 'a.1'), ('3', 'a.2'), ('4', 'a.2.1'),
('6', 'ç')
]
# update
model = ExtendedAgency(title="Agency")
form = MoveAgencyForm()
form.request = DummyRequest(session)
form.update_model(model)
assert model.parent_id == None
form = MoveAgencyForm(DummyPostData({'parent_id': 'root'}))
form.request = DummyRequest(session)
form.update_model(model)
assert model.parent_id == None
form = MoveAgencyForm(DummyPostData({'parent_id': '10'}))
form.request = DummyRequest(session)
form.update_model(model)
assert model.parent_id == 10
# update with rename
agency_a_2_2 = agencies.add(title="a.2", parent=agency_a_2)
form = MoveAgencyForm(DummyPostData({'parent_id': agency_a_2.parent_id}))
form.request = DummyRequest(session)
form.update_model(agency_a_2_2)
session.flush()
def test_extended_agencies(session):
agencies = ExtendedAgencyCollection(session)
root = agencies.add_root(title="Agency")
assert isinstance(root, ExtendedAgency)
assert isinstance(agencies.query().one(), ExtendedAgency)