def test_if_all_conditions_are_false(self, datetime_mock):
booking = Mock(cancelled=False,
start=datetime(year=2020, month=4, day=16))
booking.is_fully_paid.return_value = False
datetime_mock.now.return_value = datetime(year=2020, month=4, day=17)
result = get_booking_status(booking)
self.assertEqual(result, 'Waiting taxes')
def test_booking_status_wtih_cancelled_mock(self):
booking_mock = Mock(cancelled=True)
self.assertEqual(get_booking_status(booking_mock), 'Cancelled')
def test_booking_status_wtih_upcoming_mock(self, datetime_mock):
booking_mock = Mock(cancelled=False,
start=datetime(year=2020, month=1, day=1))
booking_mock.is_fully_paid.return_value = False
datetime_mock.now.return_value = datetime(2000, 4, 1)
self.assertEqual(get_booking_status(booking_mock), 'Upcoming')
def test_booking_status_wtih_paid_mock(self):
booking_mock = Mock(cancelled=False)
booking_mock.is_fully_paid.return_value = True
self.assertEqual(get_booking_status(booking_mock), 'Paid')
def test_if_it_booking_is_cancelled(self):
booking = Mock(cancelled=True)
result = get_booking_status(booking)
self.assertEqual(result, 'Cancelled')
def test_if_booking_is_paid(self):
booking = Mock(cancelled=False)
result = get_booking_status(booking)
self.assertEqual(result, 'Paid')