def __init__(self, *args, **kwargs): super(SearchBookingForm, self).__init__(*args, **kwargs) self.fields['booking_name'] = forms.ChoiceField( choices=Booking.get_bookings(), label=_('Booking:'), required=False, widget=forms.widgets.Select(attrs={'class': 'select2 optional'}))
def test_get_bookings(self): self.booking = Booking.objects.order_by('name').first() result = Booking.get_bookings() self.assertEqual(result[0][0], self.booking.name)