Simple library for creating fake models in the unit tests.
This simple library allows to create fake models in your test without migrations, test apps and test tables in your base. All tables that you need will created/removed during the test.
Install django-fake-model:
pip install django-fake-model
Just create a model in any file (Ex: in your test) and add decorator @YourModel.fake_me to test method or test class.
from django_fake_model import models as f
from django.db import models
from django.test import TestCase, TransactionTestCase
class MyFakeModel(f.FakeModel):
name = models.CharField(max_length=100)
@MyFakeModel.fake_me
class MyFakeModelTests(TestCase):
def test_create_model(self):
MyFakeModel.objects.create(name='123')
model = MyFakeModel.objects.get(name='123')
self.assertEqual(model.name, '123')
class MyFakeModelFunctionTest(TestCase):
@MyFakeModel.fake_me
def test_create_model(self):
MyFakeModel.objects.create(name='123')
model = MyFakeModel.objects.get(name='123')
self.assertEqual(model.name, '123')
class RelatedModel(f.FakeModel):
text = models.CharField(max_length=400)
class NyModel(f.FakeModel):
text = models.CharField(max_length=400)
related_model = models.ForeignKey(RelatedModel)
@NyModel.fake_me
@RelatedModel.fake_me
class TestRelatedModelsClassDecorator(TransactionTestCase):
def test_create_models(self):
related_model = RelatedModel()
related_model.text = 'qwerty'
related_model.save()
my_model = NyModel()
my_model.test = 'qwerty'
my_model.related_model = related_model
my_model.save()
self.assertIsNotNone(my_model)
self.assertIsNotNone(related_model)
To develop on this locally with Docker, install the Docker Engine and Docker Compose. Then you can build the Docker image and run the tests on all tox activities(this also uses a shared pip cache to reduce download times):
docker-compose up -d pg mysql docker-compose run --rm test
If you wanna run just one Tox activity you can specify that as well:
docker-compose run --rm test tox -e py35-dj19-mysql-unittest
If you add any dependencies or change the tox configuration, you have to rebuild the image:
docker-compose build
It will share this folder with the Docker containers, so that
- cookiecutter
- cookiecutter-djangopackage