def test_apply__no_ids(self): from ebpub.db.schemafilters import IdFilter fil = IdFilter(request=None, context={}, queryset=None, ids=[]) self.assertEqual(list(fil.apply()), []) from ebpub.db.models import NewsItem self.assert_(NewsItem.objects.all().count() >= 1, "we should have some items")
def test_apply__some_ids(self): from ebpub.db.schemafilters import IdFilter from ebpub.db.models import NewsItem items = NewsItem.objects.all()[:2] self.assert_(len(items), "we should have some items") expected_ids = [item.id for item in items] fil = IdFilter(request=None, context={}, queryset=None, ids=expected_ids) got_ids = [item.id for item in fil.apply()] self.assertEqual(sorted(expected_ids), sorted(got_ids))