A Simple Way to Test your Django Models
This plugin allows you to simply test your Django models, by writing your tests as you would write your models. On the other hand, pytest-django-model tests only fields, constants, and the Meta inner class. You will have to write tests of methods and properties.
The principle here is not really to test the behavior of your models but rather to make sure that the settings are the right ones.
You can install "pytest-django-model" via pip from PyPI:
$ pip install pytest-django-model
The plugin is very easy to use, there are only three things to remember:
- The
PytestDjangoModel
metaclass. - The
model
attribute. - The
parents
attribute (optional).
Let's take this model:
class Foo(Bar, Baz):
class Meta:
unique_together = ("name", "email")
FAVORITE_COLOR_CHOICES = (
("BL", "blue"),
("YE", "yellow"),
("GR", "green"),
("RE", "red")
)
name = models.CharField(max_length=256)
email = models.EmailField(blank=True)
favorite_color = models.CharField(
max_length=2, choices=FAVORITE_COLOR_CHOICES, default="BL"
)
is_awesome = models.BooleanField(default=True)
To test it, we just have to write this:
from pytest_django_model import PytestDjangoModel
# Name of the test class doesn't matter.
class TestFoo(metaclass=PytestDjangoModel):
class Meta:
model = Foo
# Parents can be a model or a list/tuple of models.
parents = (Bar, Baz)
unique_together = ("name", "email")
FAVORITE_COLOR_CHOICES = (
("BL", "blue"),
("YE", "yellow"),
("GR", "green"),
("RE", "red")
)
name = models.CharField(max_length=256)
email = models.EmailField(blank=True)
favorite_color = models.CharField(
max_length=2, choices=FAVORITE_COLOR_CHOICES, default="BL"
)
is_awesome = models.BooleanField(default=True)
And voila! We can now launch tests with the command pytest
.
From there, the class PytestDjangoModel
will generate a fake Django model
from constants, fields and options of the Meta class. This model will inherit
all the models of the parents
attribute.
The data of Foo
model and the model created from the TestFoo
class will
be extracted and compared. If any constant differs or isn't found, pytest will
raise a error, same for fields and Meta options.
Contributions are very welcome. Development Environment can be setup with
make setup
. Tests can be run with make test
, please ensure the coverage
at least stays the same before you submit a pull request.
Distributed under the terms of the GNU GPL v3.0 license, "pytest-django-model" is free and open source software.
If you encounter any problems, please file an issue along with a detailed description.