Как я могу проверить свой менеджер моделей в 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)

это правильно??? или другой метод... Как лучше всего протестировать пользовательский менеджер моделей

0 ответов

Другие вопросы по тегам