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