Skip to content

Latest commit

 

History

History
333 lines (223 loc) · 10 KB

helpers.rst

File metadata and controls

333 lines (223 loc) · 10 KB

Django helpers

Markers

pytest-django registers and uses markers. See the pytest documentation on what marks are and for notes on using them.

pytest.mark.django_db(transaction=False) - request database access

This is used to mark a test function as requiring the database. It will ensure the database is set up correctly for the test. Each test will run in its own transaction which will be rolled back at the end of the test. This behavior is the same as Django's standard django.test.TestCase class.

In order for a test to have access to the database it must either be marked using the django_db mark or request one of the db or transactional_db fixtures. Otherwise the test will fail when trying to access the database.

type transaction:bool
param transaction:The transaction argument will allow the test to use real transactions. With transaction=False (the default when not specified), transaction operations are noops during the test. This is the same behavior that django.test.TestCase uses. When transaction=True, the behavior will be the same as django.test.TransactionTestCase

Note

If you want access to the Django database inside a fixture this marker will not help even if the function requesting your fixture has this marker applied. To access the database in a fixture, the fixture itself will have to request the db or transactional_db fixture. See below for a description of them.

Note

Automatic usage with django.test.TestCase.

Test classes that subclass django.test.TestCase will have access to the database always to make them compatible with existing Django tests. Test classes that subclass Python's unittest.TestCase need to have the marker applied in order to access the database.

pytest.mark.urls - override the urlconf

.. py:function:: pytest.mark.urls(urls)

   Specify a different ``settings.ROOT_URLCONF`` module for the marked tests.

   :type urls: string
   :param urls:
     The urlconf module to use for the test, e.g. ``myapp.test_urls``.  This is
     similar to Django's ``TestCase.urls`` attribute.

   Example usage::

     @pytest.mark.urls('myapp.test_urls')
     def test_something(client):
         assert 'Success!' in client.get('/some_url_defined_in_test_urls/').content


pytest.mark.ignore_template_errors - ignore invalid template variables

.. py:function:: pytest.mark.ignore_template_errors

  If you run pytest using the ``--fail-on-template-vars`` option,
  tests will fail should your templates contain any invalid variables.
  This marker will disable this feature by setting ``settings.TEMPLATE_STRING_IF_INVALID=None``
  or the ``string_if_invalid`` template option

  Example usage::

     @pytest.mark.ignore_template_errors
     def test_something(client):
         client('some-url-with-invalid-template-vars')


pytest.mark.django_use_model - force model creation for unmanaged models

.. py:function:: pytest.mark.django_use_model(model=ModelClass)

   :type model: django model or list of django models, as kwarg
   :param model:
     Model or models to be created, should be used only with models that
     have ``Meta.managed = False``

   This will create requested model(s) for the scope of the marker.
   Allows testing of unmanaged models that are normally not created.

   .. note::

      To access database you still have to request access by using
      ``pytest.mark.django_db``

  Example usage::

     @pytest.mark.django_db
     @pytest.mark.django_use_model(model=Unmanaged)
     def test_unmanaged():
        assert Unmanaged.objects.count() >= 0


Fixtures

pytest-django provides some pytest fixtures to provide dependencies for tests. More information on fixtures is available in the pytest documentation.

rf - RequestFactory

An instance of a django.test.RequestFactory

Example

from myapp.views import my_view

def test_details(rf):
    request = rf.get('/customer/details')
    response = my_view(request)
    assert response.status_code == 200

client - django.test.Client

An instance of a django.test.Client

Example

def test_with_client(client):
    response = client.get('/')
    assert response.content == 'Foobar'

To use client as an authenticated standard user, call its login() method before accessing a URL:

def test_with_authenticated_client(client, django_user_model):
    username = "user1"
    password = "bar"
    django_user_model.objects.create(username=username, password=password)
    client.login(username=username, password=password)
    response = client.get('/private')
    assert response.content == 'Protected Area'

admin_client - django.test.Client logged in as admin

An instance of a django.test.Client, logged in as an admin user.

Example

def test_an_admin_view(admin_client):
    response = admin_client.get('/admin/')
    assert response.status_code == 200

Using the admin_client fixture will cause the test to automatically be marked for database use (no need to specify the django_db mark).

admin_user - a admin user (superuser)

An instance of a superuser, with username "admin" and password "password" (in case there is no "admin" user yet).

Using the admin_user fixture will cause the test to automatically be marked for database use (no need to specify the django_db mark).

django_user_model

A shortcut to the User model configured for use by the current Django project (aka the model referenced by settings.AUTH_USER_MODEL). Use this fixture to make pluggable apps testable regardless what User model is configured in the containing Django project.

Example

def test_new_user(django_user_model):
    django_user_model.objects.create(username="someone", password="something")

django_username_field

This fixture extracts the field name used for the username on the user model, i.e. resolves to the current settings.USERNAME_FIELD. Use this fixture to make pluggable apps testable regardless what the username field is configured to be in the containing Django project.

db

.. fixture:: db

This fixture will ensure the Django database is set up. Only required for fixtures that want to use the database themselves. A test function should normally use the :py:func:`~pytest.mark.django_db` mark to signal it needs the database.

transactional_db

This fixture can be used to request access to the database including transaction support. This is only required for fixtures which need database access themselves. A test function would normally use the :py:func:`~pytest.mark.django_db` mark to signal it needs the database.

live_server

This fixture runs a live Django server in a background thread. The server's URL can be retrieved using the live_server.url attribute or by requesting it's string value: unicode(live_server). You can also directly concatenate a string to form a URL: live_server + '/foo.

settings

This fixture will provide a handle on the Django settings module, and automatically revert any changes made to the settings (modifications, additions and deletions).

Example

def test_with_specific_settings(settings):
    settings.USE_TZ = True
    assert settings.USE_TZ

django_assert_num_queries

This fixture allows to check for an expected number of DB queries. It currently only supports the default database.

Example

def test_queries(django_assert_num_queries):
    with django_assert_num_queries(3):
        Item.objects.create('foo')
        Item.objects.create('bar')
        Item.objects.create('baz')

mailoutbox

A clean email outbox to which Django-generated emails are sent.

Example

from django.core import mail

def test_mail(mailoutbox):
    mail.send_mail('subject', 'body', '[email protected]', ['[email protected]'])
    assert len(mailoutbox) == 1
    m = mailoutbox[0]
    assert m.subject == 'subject'
    assert m.body == 'body'
    assert m.from_email == '[email protected]'
    assert list(m.to) == ['[email protected]']

Automatic cleanup

pytest-django provides some functionality to assure a clean and consistent environment during tests.

Clearing of site cache

If django.contrib.sites is in your INSTALLED_APPS, Site cache will be cleared for each test to avoid hitting the cache and causing the wrong Site object to be returned by Site.objects.get_current().

Clearing of mail.outbox

mail.outbox will be cleared for each pytest, to give each new test an empty mailbox to work with. However, it's more "pytestic" to use the mailoutbox fixture described above than to access mail.outbox.