def test_public_method_pub_date(self):
qs = PublicQuerySet(PublicDefault,
pub_date_attr='pub_date')
self.assertEqual(
set(PublicDefault.objects.all()),
set(qs.all()))
self.assertEqual(4, len(qs.all()))
self.assertEqual(
set(PublicDefault.objects.filter(pub_date__lte=datetime.utcnow())),
set(qs.public()))
self.assertEqual(2, len(qs.public()))
def test_public_method_is_public(self):
qs = PublicQuerySet(PublicDefault,
is_public_attr='is_public')
self.assertEqual(
set(PublicDefault.objects.all()),
set(qs.all()))
self.assertEqual(4, len(qs.all()))
self.assertEqual(
set(PublicDefault.objects.filter(is_public=True)),
set(qs.public()))
self.assertEqual(2, len(qs.public()))
def test_public_method_status(self):
qs = PublicQuerySet(PublicStatus,
status_attr='status',
status_values=PublicStatus.PUBLIC_STATUS)
self.assertEqual(
set(PublicStatus.objects.all()),
set(qs.all()))
self.assertEqual(4, len(qs.all()))
self.assertEqual(
set(PublicStatus.objects.filter(status__in=PublicStatus.PUBLIC_STATUS)),
set(qs.public()))
self.assertEqual(2, len(qs.public()))
def test_public_method_both(self):
qs = PublicQuerySet(PublicDefault,
is_public_attr='is_public',
pub_date_attr='pub_date')
self.assertEqual(
set(PublicDefault.objects.all()),
set(qs.all()))
self.assertEqual(4, len(qs.all()))
self.assertEqual(
set(PublicDefault.objects.filter(is_public=True, pub_date__lte=datetime.utcnow())),
set(qs.public()))
self.assertEqual(1, len(qs.public()))