Как я могу проверить свой менеджер моделей в Django
Ниже мой таможенный менеджер
class DefaultModelManager(models.Manager):
def get_or_none(self, **kwargs):
try:
return self.get(**kwargs)
except self.model.DoesNotExist:
return None
модель
class TestModel(models.Model):
objects=DefaultModelManager()
как я могу проверить этот код? Ниже мой код
class ModelManagerTest(TestCase):
def test_can_get_or_none(self):
TestModel.create('...')
test = TestModel.objects.get_or_none('...')
test2 = TestModel.objects.get('...')
self.assertEqual(test, test2)
def test_cant_get_or_none(self):
test = TestModel.objects.get_or_none('...')
self.assertEqual(test, None)
это правильно??? или другой метод... Как лучше всего протестировать пользовательский менеджер моделей