def test_add_committee_preferences(self): """ Tests that adding committee preferences works """ params = self.valid_params.copy() params['school_name'] = 'CommitteeTest1' form = RegistrationForm(params) # No committee preferences self.assertTrue(form.is_valid()) school = form.create_school() self.assertTrue(form.add_committee_preferences(school)) self.assertEqual(len(school.committeepreferences.all()), 0) # Some preferences params['committee_prefs'] = [13, 14, 19] params['school_name'] = 'CommitteeTest2' form = RegistrationForm(params) self.assertTrue(form.is_valid()) school = form.create_school() self.assertTrue(form.add_committee_preferences(school)) self.assertEqual(len(school.committeepreferences.all()), 3) for committee in school.committeepreferences.all(): self.assertIn(committee.id, params['committee_prefs'])