def test_get_queryset_no_dates_provided(self):
"""Test what happens if neither start or end date/time are provided."""
# Using invite because it's one of the easier models in Connect to
# quickly create a timestamped instance with.
timezone = get_current_timezone()
invite1 = mommy.make(Invite)
invite1.created_at = make_aware(datetime(2012, 12, 31, 23, 59),
timezone)
invite1.save()
invite2 = mommy.make(Invite)
invite2.created_at = make_aware(datetime(2013, 06, 01, 00, 01),
timezone)
invite2.save()
invite3 = mommy.make(Invite)
invite3.created_at = make_aware(datetime(2014, 01, 01, 00, 01),
timezone)
invite3.save()
# Get the queryset
mixin = DateTimeRangeListMixin()
mixin.model = Invite
mixin.request = self.request_factory.get('')
queryset = mixin.get_queryset()
self.assertIn(invite1, queryset)
self.assertIn(invite2, queryset)
self.assertIn(invite3, queryset)
def test_get_queryset_no_dates_provided(self):
"""Test what happens if neither start or end date/time are provided."""
# Using invite because it's one of the easier models in Connect to
# quickly create a timestamped instance with.
timezone = get_current_timezone()
invite1 = mommy.make(Invite)
invite1.created_at = make_aware(
datetime(2012, 12, 31, 23, 59), timezone)
invite1.save()
invite2 = mommy.make(Invite)
invite2.created_at = make_aware(
datetime(2013, 06, 01, 00, 01), timezone)
invite2.save()
invite3 = mommy.make(Invite)
invite3.created_at = make_aware(
datetime(2014, 01, 01, 00, 01), timezone)
invite3.save()
# Get the queryset
mixin = DateTimeRangeListMixin()
mixin.model = Invite
mixin.request = self.request_factory.get('')
queryset = mixin.get_queryset()
self.assertIn(invite1, queryset)
self.assertIn(invite2, queryset)
self.assertIn(invite3, queryset)
def test_get_queryset_only_end_date(self):
"""Test updating queryset when only end date is provided."""
# Using invite because it's one of the easier models in Connect to
# quickly create a timestamped instance with.
timezone = get_current_timezone()
invite1 = mommy.make(Invite)
invite1.created_at = make_aware(datetime(2012, 12, 31, 23, 59),
timezone)
invite1.save()
invite2 = mommy.make(Invite)
invite2.created_at = make_aware(datetime(2013, 06, 01, 00, 01),
timezone)
invite2.save()
invite3 = mommy.make(Invite)
invite3.created_at = make_aware(datetime(2014, 01, 01, 00, 01),
timezone)
invite3.save()
# Get the queryset
mixin = DateTimeRangeListMixin()
mixin.model = Invite
mixin.request = self.request_factory.get(
'/?end_datetime=2013-12-31 11:59 pm')
queryset = mixin.get_queryset()
self.assertIn(invite1, queryset)
self.assertIn(invite2, queryset)
self.assertNotIn(invite3, queryset)
def test_get_queryset_only_end_date(self):
"""Test updating queryset when only end date is provided."""
# Using invite because it's one of the easier models in Connect to
# quickly create a timestamped instance with.
timezone = get_current_timezone()
invite1 = mommy.make(Invite)
invite1.created_at = make_aware(
datetime(2012, 12, 31, 23, 59), timezone)
invite1.save()
invite2 = mommy.make(Invite)
invite2.created_at = make_aware(
datetime(2013, 06, 01, 00, 01), timezone)
invite2.save()
invite3 = mommy.make(Invite)
invite3.created_at = make_aware(
datetime(2014, 01, 01, 00, 01), timezone)
invite3.save()
# Get the queryset
mixin = DateTimeRangeListMixin()
mixin.model = Invite
mixin.request = self.request_factory.get(
'/?end_datetime=2013-12-31 11:59 pm'
)
queryset = mixin.get_queryset()
self.assertIn(invite1, queryset)
self.assertIn(invite2, queryset)
self.assertNotIn(invite3, queryset)
def test_get_date_range_form_initial(self):
"""Test getting form initial values."""
mixin = DateTimeRangeListMixin()
mixin.request = self.request_factory.get(
'/?start_datetime=2013-01-01 12:01 am'
'&end_datetime=2013-12-31 11:59 pm')
initial = mixin.get_date_range_form_initial()
self.assertEqual(initial['start_datetime'], u'2013-01-01 12:01 am')
self.assertEqual(initial['end_datetime'], u'2013-12-31 11:59 pm')
def test_get_date_range_form_initial(self):
"""Test getting form initial values."""
mixin = DateTimeRangeListMixin()
mixin.request = self.request_factory.get(
'/?start_datetime=2013-01-01 12:01 am'
'&end_datetime=2013-12-31 11:59 pm'
)
initial = mixin.get_date_range_form_initial()
self.assertEqual(initial['start_datetime'], u'2013-01-01 12:01 am')
self.assertEqual(initial['end_datetime'], u'2013-12-31 11:59 pm')
def test_get_date_range_form_kwargs(self):
"""Test getting form kwargs."""
mixin = DateTimeRangeListMixin()
mixin.request = self.request_factory.get(
'/?start_datetime=2013-01-01 12:01 am'
'&end_datetime=2013-12-31 11:59 pm')
kwargs = mixin.get_date_range_form_kwargs()
self.assertEqual(kwargs['initial']['start_datetime'],
u'2013-01-01 12:01 am')
self.assertEqual(kwargs['initial']['end_datetime'],
u'2013-12-31 11:59 pm')
def test_get_date_range_form(self):
"""Test form creation."""
mixin = DateTimeRangeListMixin()
mixin.request = self.request_factory.get(
'/?start_datetime=2013-01-01 12:01 am'
'&end_datetime=2013-12-31 11:59 pm')
form = mixin.get_date_range_form()
self.assertIsInstance(form, DateTimeRangeForm)
self.assertEqual(form.initial['start_datetime'],
u'2013-01-01 12:01 am')
self.assertEqual(form.initial['end_datetime'], u'2013-12-31 11:59 pm')
def test_get_date_range_form_kwargs(self):
"""Test getting form kwargs."""
mixin = DateTimeRangeListMixin()
mixin.request = self.request_factory.get(
'/?start_datetime=2013-01-01 12:01 am'
'&end_datetime=2013-12-31 11:59 pm'
)
kwargs = mixin.get_date_range_form_kwargs()
self.assertEqual(
kwargs['initial']['start_datetime'],
u'2013-01-01 12:01 am'
)
self.assertEqual(
kwargs['initial']['end_datetime'],
u'2013-12-31 11:59 pm'
)
def test_get_date_range_form(self):
"""Test form creation."""
mixin = DateTimeRangeListMixin()
mixin.request = self.request_factory.get(
'/?start_datetime=2013-01-01 12:01 am'
'&end_datetime=2013-12-31 11:59 pm'
)
form = mixin.get_date_range_form()
self.assertIsInstance(form, DateTimeRangeForm)
self.assertEqual(
form.initial['start_datetime'],
u'2013-01-01 12:01 am'
)
self.assertEqual(
form.initial['end_datetime'],
u'2013-12-31 11:59 pm'
)
def test_get_context_data(self):
"""Test adding form to context."""
mixin = DateTimeRangeListMixin()
mixin.request = self.request_factory.get('/')
self.assertIsInstance(mixin.get_context_data()['date_range_form'],
DateTimeRangeForm)
def test_get_context_data(self):
"""Test adding form to context."""
mixin = DateTimeRangeListMixin()
mixin.request = self.request_factory.get('/')
self.assertIsInstance(
mixin.get_context_data()['date_range_form'], DateTimeRangeForm)