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)