Skip to content

Commit 1db3353

Browse files
committed
Fixes linting.
1 parent 2501f61 commit 1db3353

1 file changed

Lines changed: 7 additions & 1 deletion

File tree

tests/onegov/org/test_views_resources.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
from onegov.file import FileCollection
2020
from onegov.form import FormSubmission
2121
from onegov.org.models import ResourceRecipientCollection
22+
from onegov.org.models.ticket import ReservationHandler
2223
from onegov.pay import Payment, PaymentCollection, InvoiceCollection
2324
from onegov.pdf.utils import extract_pdf_info
2425
from onegov.reservation import Resource, ResourceCollection
@@ -5501,9 +5502,14 @@ def _push_reservations_into_past(
55015502
session = client.app.session()
55025503
ticket_id = UUID(cancel_url.split('/ticket/RSV/')[1].split('/')[0])
55035504
ticket = session.query(Ticket).filter(Ticket.id == ticket_id).one()
5504-
reservations = sorted(ticket.handler.reservations, key=lambda r: r.start)
5505+
handler = ticket.handler
5506+
assert isinstance(handler, ReservationHandler)
5507+
reservations = sorted(
5508+
handler.reservations, key=lambda r: r.start or datetime.min
5509+
)
55055510
past_ids = []
55065511
for r in reservations[:count]:
5512+
assert r.start is not None and r.end is not None
55075513
r.start = r.start - timedelta(days=40)
55085514
r.end = r.end - timedelta(days=40)
55095515
past_ids.append(r.id)

0 commit comments

Comments
 (0)