def test_get_parish_council_election_none(self, view_obj): """ Test if there is no parish council related to views ballots that None is returned """ post_election = PostElectionFactory() post_election.num_parish_councils = 0 view_obj.ballots = PostElection.objects.annotate( num_parish_councils=Count("parish_councils")) result = view_obj.get_parish_council_election() assert result is None assert view_obj.parish_council_election is None
def test_get_parish_council_election_object_returned(self, view_obj): """ Test if there is a parish council related to views ballots that it is returned """ post_election = PostElectionFactory() post_election.num_parish_councils = 0 parish_council_election = ParishCouncilElection.objects.create() parish_council_election.ballots.add(post_election) view_obj.ballots = PostElection.objects.annotate( num_parish_councils=Count("parish_councils")) result = view_obj.get_parish_council_election() assert result == parish_council_election assert view_obj.parish_council_election == parish_council_election