|
5 | 5 | from modernomad.core.models import Payment, Use, Booking, UserProfile, LocationEmailTemplate
|
6 | 6 | from modernomad.core.emails.messages import new_booking_notify, send_booking_receipt, updated_booking_notify, admin_daily_update
|
7 | 7 | from modernomad.core.tasks import send_departure_email, send_guest_welcome, guests_residents_daily_update
|
8 |
| -from django.utils import timezone |
9 | 8 | from datetime import datetime, timedelta, date
|
| 9 | +from freezegun import freeze_time |
10 | 10 |
|
| 11 | +TODAY = date(2019, 3, 14) |
11 | 12 |
|
| 13 | +@freeze_time(TODAY) |
12 | 14 | class EmailsTestCase(TestCase):
|
13 | 15 |
|
14 | 16 | def create_booking(self, user, rate=100, status=Use.PENDING, arrive=date(4016, 1, 13), depart=date(4016, 1, 23)):
|
@@ -44,14 +46,13 @@ class MockResponse():
|
44 | 46 | self. admin = self. create_user( 'admin1', admin=True, email='[email protected]')
|
45 | 47 | self.booking = self.create_booking(user=self.guest1)
|
46 | 48 |
|
47 |
| - today = timezone.localtime(timezone.now()) |
48 |
| - yesterday = today + timedelta(days=-1) |
49 |
| - in_two_days = today + timedelta(days=2) |
| 49 | + yesterday = TODAY + timedelta(days=-1) |
| 50 | + in_two_days = TODAY + timedelta(days=2) |
50 | 51 | after_that = in_two_days + timedelta(days=1)
|
51 | 52 |
|
52 |
| - self.departing_today = self.create_booking(arrive=yesterday, depart=today, user=self.guest1, status="confirmed") |
| 53 | + self.departing_today = self.create_booking(arrive=yesterday, depart=TODAY, user=self.guest1, status="confirmed") |
53 | 54 | self.arriving_in_two_days = self.create_booking(arrive=in_two_days, depart=after_that, user=self.guest2, status="confirmed")
|
54 |
| - self.arriving_today = self.create_booking(arrive=today, depart=after_that, user=self.guest3, status="confirmed") |
| 55 | + self.arriving_today = self.create_booking(arrive=TODAY, depart=after_that, user=self.guest3, status="confirmed") |
55 | 56 |
|
56 | 57 | def tearDown(self):
|
57 | 58 | self.mock_mailgun_send.stop()
|
|
0 commit comments