Skip to content

eralumin/pytest-django-model

Repository files navigation

pytest-django-model

PyPI version Python versions See Build Status on Travis CI

A Simple Way to Test your Django Models


Description

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.

Installation

You can install "pytest-django-model" via pip from PyPI:

$ pip install pytest-django-model

Usage

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.

Contributing

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.

License

Distributed under the terms of the GNU GPL v3.0 license, "pytest-django-model" is free and open source software.

Issues

If you encounter any problems, please file an issue along with a detailed description.

Releases

No releases published

Packages

No packages published