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¶
-
pytest.mark.
urls
(urls)¶ Specify a different
settings.ROOT_URLCONF
module for the marked tests.Parameters: urls (string) – The urlconf module to use for the test, e.g. myapp.test_urls
. This is similar to Django’sTestCase.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¶
-
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 settingsettings.TEMPLATE_STRING_IF_INVALID=None
or thestring_if_invalid
template optionExample usage:
@pytest.mark.ignore_template_errors def test_something(client): client('some-url-with-invalid-template-vars')
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
¶
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 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
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', 'from@example.com', ['to@example.com'])
assert len(mailoutbox) == 1
m = mailoutbox[0]
assert m.subject == 'subject'
assert m.body == 'body'
assert m.from_email == 'from@example.com'
assert list(m.to) == ['to@example.com']
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
.