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")