mirror of
https://github.com/rjNemo/villafleurie
synced 2026-06-12 05:16:50 +00:00
34 lines
920 B
Python
34 lines
920 B
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
|
|
|
|
|
|
class BookingTestCase(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_booking(
|
|
place=self.place,
|
|
start=self.start,
|
|
end=self.end,
|
|
guest=self.guest
|
|
)
|
|
|
|
self.assertEqual(booking.price, 600)
|