def test_create_and_save_candidate(self):
centre = Centre()
centre.name = "Test School"
centre.centre_id = "60114"
centre.create_date = datetime.datetime(2012, 03, 04)
centre.save()
# Create new user who belongs to centre
user = User.objects.create_user("john", "*****@*****.**", "johnpassword")
user.first_name = "John"
user.last_name = "Lennon"
user.save()
u = User.objects.get(username__exact="john")
self.assertEquals(u.first_name, "John")
self.assertEquals(u.last_name, "Lennon")
candidate = Candidate()
candidate.gender = "M"
candidate.dob = datetime.date(1969, 05, 01)
candidate.centre = centre
candidate.user = user
candidate.save()
# check we can retrieve it
all_objs = Candidate.objects.all()
self.assertEquals(len(all_objs), 1)
my_obj = all_objs[0]
self.assertEquals(my_obj.gender, "M")
self.assertEquals(my_obj.dob, datetime.date(1969, 05, 01))
self.assertEquals(unicode(my_obj.centre), "60114 Test School")
self.assertEquals(unicode(my_obj.user.first_name), "John")
self.assertEquals(unicode(my_obj), "John Lennon")
def test_create_and_save_centre(self):
centre = Centre()
centre.name = "Test School"
centre.centre_id = "60114"
centre.create_date = datetime.datetime(2012, 03, 04)
centre.save()
# check we can retrieve it
all_objs = Centre.objects.all()
self.assertEquals(len(all_objs), 1)
my_obj = all_objs[0]
self.assertEquals(my_obj.name, "Test School")
self.assertEquals(my_obj.centre_id, "60114")
self.assertEquals(my_obj.create_date, datetime.datetime(2012, 03, 04))
# check it returns its own name
self.assertEquals(unicode(my_obj), "60114 Test School")
