def setUp(self): self.test_user = ConpecUser() self.test_user.user = User.objects.create_user(username='******', email='*****@*****.**') self.test_user.user.set_password('teste') self.test_user.user.save() self.test_user.ra = '123456' self.test_user.save()
def test_saving_and_retrieving_users(self): first_user = ConpecUser() first_user.user = User.objects.create_user(username='******', email='*****@*****.**') first_user.ra = '123456' first_user.save() second_user = ConpecUser() second_user.user = User.objects.create_user(username='******', email='*****@*****.**') second_user.ra = '654321' second_user.save() saved_users = ConpecUser.objects.all() self.assertEqual(saved_users.count(), 3) first_saved_user = saved_users[1] second_saved_user = saved_users[2] self.assertEqual(str(first_saved_user), 'first-123456') self.assertEqual(str(second_saved_user), 'second-654321')