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'])
