def setUp(self):
self.user = User.objects.create(username='******')
self.calendar = Calendar.objects.create(name='Basic', slug='basic')
self.event = Event.objects.create(
name='Event', slug='event', creator=self.user
)
self.start = datetime.now() + timedelta(minutes=30)
self.finish = self.start + timedelta(hours=2)
self.occurrence = self.event.add_occurrences(
self.calendar, self.start, self.finish)[0]
self.timezone = 'America/Chicago'
self.localized = LocalizedOccurrenceProxy(
self.occurrence, timezone=self.timezone
)
deactivate_default_occurrence_validators()
class TestLocalizedOccurrenceProxy(TestCase):
def setUp(self):
self.user = User.objects.create(username='******')
self.calendar = Calendar.objects.create(name='Basic', slug='basic')
self.event = Event.objects.create(
name='Event', slug='event', creator=self.user
)
self.start = datetime.now() + timedelta(minutes=30)
self.finish = self.start + timedelta(hours=2)
self.occurrence = self.event.add_occurrences(
self.calendar, self.start, self.finish)[0]
self.timezone = 'America/Chicago'
self.localized = LocalizedOccurrenceProxy(
self.occurrence, timezone=self.timezone
)
deactivate_default_occurrence_validators()
def tearDown(self):
activate_default_occurrence_validators()
def test_timezone_property(self):
expected = pytz.timezone('America/Chicago')
assert_equal(self.localized.timezone, expected)
# Should handle actual timezone object as well as string name:
localized2 = LocalizedOccurrenceProxy(
self.occurrence, timezone=expected
)
assert_equal(localized2.timezone, expected)
# Garbage input results in a timezone of None:
localized3 = LocalizedOccurrenceProxy(
self.occurrence, timezone='Non-Existent Timezone'
)
assert localized3.timezone is None
def test_default_timezone_property(self):
original_timezone = settings.TIME_ZONE
try:
settings.TIME_ZONE = 'UTC'
expected = pytz.timezone('UTC')
settings.TIME_ZONE = self.timezone
expected = pytz.timezone(self.timezone)
assert_equal(self.localized.default_timezone, expected)
finally:
settings.TIME_ZONE = original_timezone
def test_localized_occurrence_proxy_wrapping_proxy_works_properly(self):
timezone = 'Antarctica/McMurdo'
localized = LocalizedOccurrenceProxy(
self.localized, timezone=timezone
)
assert_equal(localized.real_start, self.occurrence.start)
assert_equal(localized.real_finish, self.occurrence.finish)
for attr in ('start', 'finish'):
expected = adjust_datetime_to_timezone(
getattr(self.occurrence, attr),
settings.TIME_ZONE,
timezone
)
assert_equal(getattr(localized, attr), expected)
def test_real_datetime_property_populated(self):
assert hasattr(self.localized, 'real_start')
assert hasattr(self.localized, 'real_finish')
assert_equal(self.localized.real_start, self.occurrence.start)
assert_equal(self.localized.real_finish, self.occurrence.finish)
def test_datetime_properties_localized_to_specified_timezone(self):
expected = localtime_for_timezone(self.start, self.timezone)
assert_equal(self.localized.start, expected)
expected = localtime_for_timezone(self.finish, self.timezone)
assert_equal(self.localized.finish, expected)
def test_assign_to_datetime_properties_updates_real_datetime_properties(self):
new_date = self.start + timedelta(days=5)
dts = [
localtime_for_timezone(new_date, 'UTC'),
localtime_for_timezone(new_date, 'America/Chicago'),
localtime_for_timezone(new_date, 'Asia/Shanghai'),
localtime_for_timezone(new_date, 'Europe/London'),
localtime_for_timezone(new_date, 'Australia/Sydney'),
]
expected = new_date
for attr in ('start', 'finish'):
for dt in dts:
setattr(self.localized, attr, dt)
assert_equal(getattr(self.localized, 'real_%s' % attr), expected)
self.localized.save()
occurrence = Occurrence.objects.get(pk=self.occurrence.pk)
assert_equal(getattr(occurrence, attr), expected)
