def test_save_with_deposit(self):
transaction = model_creation.create_test_client_transaction(
self.admin, timezone.now())
adults = 2
children = 1
start_date = timezone.now()
end_date = timezone.now() + timezone.timedelta(days=7)
aptrental = model_creation.create_test_aptrental(
transaction,
timezone.now(),
adults=adults,
children=children,
start_date=start_date,
end_date=end_date,
)
form_data = {
'adults': adults,
'children': children,
'start_date': start_date,
'end_date': end_date,
'add_deposit': True,
}
form = AptRentalForm(data=form_data, instance=aptrental)
self.assertTrue(form.is_valid())
self.assertTrue(form.save())
def test_form_ok(self):
form_data = {
'adults': 2,
'children': 0,
'start_date': timezone.now(),
'end_date': timezone.now() + timezone.timedelta(days=7),
}
form = AptRentalForm(data=form_data)
self.assertTrue(form.is_valid())
def test_form_no_start(self):
form_data = {
'adults': 2,
'children': 0,
'end_date': timezone.now(),
}
form = AptRentalForm(data=form_data)
print(form.errors)
self.assertIn('This field is required.', form.errors['start_date'])
self.assertFalse(form.is_valid())
def test_form_many_people(self):
form_data = {
'adults': 3,
'children': 3,
'start_date': timezone.now(),
'end_date': timezone.now() + timezone.timedelta(days=7),
}
form = AptRentalForm(data=form_data)
self.assertIn('Five people maximum.', form.non_field_errors())
self.assertFalse(form.is_valid())
def test_form_many_children(self):
form_data = {
'adults': 2,
'children': 5,
'start_date': timezone.now(),
'end_date': timezone.now() + timezone.timedelta(days=7),
}
form = AptRentalForm(data=form_data)
self.assertFalse(form.is_valid())
self.assertIn('Up to four children only.', form.errors['children'])
def test_form_departure_same(self):
form_data = {
'adults': 2,
'children': 0,
'start_date': '{}'.format(timezone.now().date()),
'end_date': '{}'.format(timezone.now().date()),
}
form = AptRentalForm(data=form_data)
self.assertIn('Departure date is the same as arrival.',
form.non_field_errors())
self.assertFalse(form.is_valid())
def test_form_missing_prices(self):
form_data = {
'adults': 2,
'children': 0,
'start_date': timezone.now() + timezone.timedelta(days=1000),
'end_date': timezone.now() + timezone.timedelta(days=1007),
}
form = AptRentalForm(data=form_data)
self.assertIn('Missing prices in selected dates.',
form.non_field_errors())
self.assertFalse(form.is_valid())
def test_form_old_start_date(self):
form_data = {
'adults': 2,
'children': 0,
'start_date': timezone.now() - timezone.timedelta(days=50),
'end_date': timezone.now() - timezone.timedelta(days=45),
}
form = AptRentalForm(data=form_data)
self.assertIn('Arrival date is too far back.',
form.errors['start_date'])
self.assertFalse(form.is_valid())