def test_search_in_one_cluster(self):
search = SingleSubjectManagerReservationSearch(
subject_manager=self.hotel1.rooms, reservation_model=HotelRoomReservation
)
results = search.search("2050-01-05", "2050-01-22")
length = len(results)
self.assertEqual(length, 1, "There should be 1 results returned")
def reservations(request, pool_id):
pool = get_object_or_404(Pool, pk=pool_id)
Form = PoolReservationsSearchForm
form = Form(request.GET) if request.GET else Form()
searcher = SingleSubjectManagerReservationSearch(
reservation_model=LaneReservation, subject_manager=pool.subjects)
reservations = (
searcher.search(**form.cleaned_data)
if form.is_valid() else
LaneReservation.objects.filter(subject__cluster=pool)
if not request.GET else
[]
)
return render(
request,
'pools/reservations.html',
{
'reservations': reservations,
'pool': pool,
'form': form,
}
)
def search(self, subject_manager):
search = SingleSubjectManagerReservationSearch(reservation_model=LaneReservation,
subject_manager=subject_manager)
return search.search(self.cleaned_data['start'], self.cleaned_data['end'])
