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')