mirror of
https://github.com/rjNemo/villafleurie
synced 2026-06-06 02:16:47 +00:00
40 lines
1 KiB
Python
40 lines
1 KiB
Python
from datetime import datetime
|
|
from django.shortcuts import get_object_or_404
|
|
from django.test import TestCase
|
|
from rental.models.place import Place
|
|
from rental.models.booking import Booking
|
|
from rental.models.guest import Guest
|
|
|
|
place_name = 'T2'
|
|
place = get_object_or_404(Place, name=place_name)
|
|
|
|
|
|
x = get_reservation_price(place_name, start, end)
|
|
print(x)
|
|
|
|
|
|
class TestBooking(TestCase):
|
|
def setUp(self):
|
|
self.place = Place.objects.create(
|
|
name='TX',
|
|
price=100
|
|
)
|
|
self.guest = Guest.objects.create(
|
|
name="Ruidy",
|
|
email="r@mail.com",
|
|
phone="1092198YE8939798"
|
|
)
|
|
|
|
self.start = datetime(2019, 11, 14)
|
|
self.end = datetime(2019, 11, 20)
|
|
|
|
def test_bookingPrice(self):
|
|
# place = Place.objects.get(name='TX')
|
|
booking = Booking.objects.create(
|
|
place=self.place,
|
|
start=self.start,
|
|
end=self.end,
|
|
guest=self.guest
|
|
)
|
|
|
|
self.assertEqual(booking.get_reservation_price())
|