def test_mark_annotated(self): queryset = Project.objects.annotate( admin_vote_count=Count('vote', distinct=True) ) mark_as_plan_new(None, None, queryset) for project in Project.objects.all(): self.assertEqual(project.status.slug, 'plan-new')
def test_mark_as_plan_new(self): mark_as_plan_new(None, None, Project.objects) for project in Project.objects.all(): self.assertEqual(project.status.slug, 'plan-new')