def test_country_form_country_choices():
form = CountryForm(data={'csrf': '', 'country': 'PL'})
assert form.fields['country'].choices == []
zone = ShippingZone.objects.create(countries=['PL', 'DE'], name='Europe')
form = CountryForm(data={'csrf': '', 'country': 'PL'})
expected_choices = [(country.code, country.name)
for country in zone.countries]
expected_choices = sorted(expected_choices, key=lambda choice: choice[1])
assert form.fields['country'].choices == expected_choices
def test_country_form_country_choices():
form = CountryForm(data={"csrf": "", "country": "PL"})
assert form.fields["country"].choices == []
zone = ShippingZone.objects.create(countries=["PL", "DE"], name="Europe")
form = CountryForm(data={"csrf": "", "country": "PL"})
expected_choices = [(country.code, country.name)
for country in zone.countries]
expected_choices = sorted(expected_choices, key=lambda choice: choice[1])
assert form.fields["country"].choices == expected_choices
def reserve_picture(request, slug, product_id):
products = products_with_details(user=request.user)
product = get_object_or_404(products, id=product_id)
default_country = get_user_shipping_country(request)
print(default_country)
country_form = CountryForm(initial={'country': default_country})
reservations = None
if request.user.is_authenticated:
reservations = Reservation.objects.filter(user=request.user)
is_reserved = False
reservation_got = None
if request.user.is_authenticated:
for res in reservations:
if res.product_name == product.name:
is_reserved = True
reservation_got = Reservation.objects.get(product_name=product.name)
break
ctx = {
'product': product,
'country_form': country_form,
'is_reserved': is_reserved,
'reservation': reservation_got
}
return TemplateResponse(request, 'order/reservation.html', ctx)
