def test_seats_contested_local_election(self): builder = ElectionBuilder('local', '2017-06-08')\ .with_organisation(self.org1)\ .with_division(self.org_div_1) election = builder.build_ballot(None) election.save() self.assertIsNone(election.seats_contested) self.assertEqual(3, election.seats_total)
def test_with_metadata(self): snooper = SnoopedElection.objects.create() builder = ElectionBuilder('local', '2017-06-08')\ .with_organisation(self.org1)\ .with_division(self.org_div_1)\ .with_source('foo/bar.baz')\ .with_snooped_election(snooper.id) election = builder.build_ballot(None) election.save() self.assertEqual('foo/bar.baz', election.source) assert isinstance(election.snooped_election, SnoopedElection)