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)
