def test_add_country_preferences(self): """ Tests that adding country preferences works """ # Initialization params = self.valid_params.copy() for i in xrange(1,11): params['country_pref' + str(i)] = 0 # No preferences at all params['school_name'] = 'CountryTest1' form = RegistrationForm(params) self.assertTrue(form.is_valid()) school = form.create_school() self.assertTrue(form.add_country_preferences(school)) prefs = CountryPreference.objects.filter(school=school) self.assertEqual(len(prefs), 0) # Couple of random preferences scattered around params['school_name'] = 'CountryTest2' params['country_pref1'] = 2 params['country_pref3'] = 10 params['country_pref7'] = 7 form = RegistrationForm(params) self.assertTrue(form.is_valid()) school = form.create_school() self.assertTrue(form.add_country_preferences(school)) prefs = CountryPreference.objects.filter(school=school) self.assertEqual(len(prefs), 3) self.assertEqual(prefs[0].country.id, params['country_pref1']) self.assertEqual(prefs[1].country.id, params['country_pref3']) self.assertEqual(prefs[2].country.id, params['country_pref7'])
def test_add_country_preferences(self): """ Tests that adding country preferences works """ # Initialization params = self.valid_params.copy() for i in xrange(1, 11): params['country_pref' + str(i)] = 0 # No preferences at all params['school_name'] = 'CountryTest1' form = RegistrationForm(params) self.assertTrue(form.is_valid()) school = form.create_school() self.assertTrue(form.add_country_preferences(school)) prefs = CountryPreference.objects.filter(school=school) self.assertEqual(len(prefs), 0) # Couple of random preferences scattered around params['school_name'] = 'CountryTest2' params['country_pref1'] = 2 params['country_pref3'] = 10 params['country_pref7'] = 7 form = RegistrationForm(params) self.assertTrue(form.is_valid()) school = form.create_school() self.assertTrue(form.add_country_preferences(school)) prefs = CountryPreference.objects.filter(school=school) self.assertEqual(len(prefs), 3) self.assertEqual(prefs[0].country.id, params['country_pref1']) self.assertEqual(prefs[1].country.id, params['country_pref3']) self.assertEqual(prefs[2].country.id, params['country_pref7'])