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