def create_election(self, election_name, date=None): kwargs = {"name": election_name} if date: kwargs["election_date"] = date else: kwargs["election_date"] = date_in_near_future return ElectionFactory(**kwargs)
def test_person_update_election_future_not_current(self): future_election = ElectionFactory(election_date="2050-01-01", slug="parl.2050-01-01", current=False) ballot = BallotPaperFactory( election=future_election, post=self.dulwich_post, ballot_paper_id="parl.foo.2050-01-01", ) self.person.memberships.create(ballot=ballot, post=self.dulwich_post, party=self.green_party) response = self.app.get("/person/{}/update/".format(self.person.pk), user=self.user) self.assertEqual(response.status_code, 200) self.assertEqual( set(response.context["form"].initial.keys()), { "honorific_prefix", "name", "honorific_suffix", "gender", "birth_date", "death_date", "biography", "favourite_biscuit", "standing_parl.2015-05-07", "constituency_parl.2015-05-07", "party_GB_parl.2015-05-07", "standing_parl.2050-01-01", "constituency_parl.2050-01-01", "party_GB_parl.2050-01-01", "person", }, )