def test_only_a_user_who_created_a_class_can_delete_it(self): user_1 = UserFactory() user_2 = UserFactory() class_1 = ClassroomFactory(created_by=user_1) class_2 = ClassroomFactory(created_by=user_2) self.assertTrue(class_1.can_delete(user_1)) self.assertFalse(class_1.can_delete(user_2)) self.assertFalse(class_2.can_delete(user_1)) self.assertTrue(class_2.can_delete(user_2))
def test_superuser_can_delete_any_class(self): super_user = get_user_model().objects.create_superuser( username='******', email='*****@*****.**', password='******') user_1 = UserFactory() user_2 = UserFactory() class_1 = ClassroomFactory(created_by=user_1) class_2 = ClassroomFactory(created_by=user_2) self.assertTrue(class_1.can_delete(super_user)) self.assertTrue(class_2.can_delete(super_user))