diff --git a/tests/videodinges/factories.py b/tests/videodinges/factories.py index 4a09346..d9a7ead 100644 --- a/tests/videodinges/factories.py +++ b/tests/videodinges/factories.py @@ -28,6 +28,8 @@ def create(model: Type[T], **kwargs) -> T: if model is models.Upload: return _create_with_defaults(models.Upload, kwargs, file=SimpleUploadedFile('some_file.txt', b'some contents')) + raise NotImplementedError('Factory for %s not implemented' % model) + def _create_with_defaults(model: Type[T], kwargs: dict, **defaults) -> T: """ diff --git a/tests/videodinges/test_factories/test_create.py b/tests/videodinges/test_factories/test_create.py new file mode 100644 index 0000000..53b9ace --- /dev/null +++ b/tests/videodinges/test_factories/test_create.py @@ -0,0 +1,16 @@ +from django.db import models + +from tests.videodinges import factories +from django.test import TestCase + +class CreateTestCase(TestCase): + + def test_factory_returns_model(self): + + class NotImplementedModel(models.Model): + class Meta: + app_label = 'some_test_label' + + with self.assertRaises(NotImplementedError): + factories.create(NotImplementedModel) +