Skip to main content

Replaces with something you can patch globally.

Project description


Patch the storage of your models' FileFields effortlessly.
Best served with pytest-django and your own fixtures.

Install and enable the plugin:

pip install pytest-django-filefield

Roll your fixture:

from django.core.files import storage
from django.db import models

def fake_storage(monkeypatch, tmp_path) 
    storage = storage.FileSystemStorage(location=tmp_path)
    monkeypatch.setattr(models.FileField, "storage", storage)
    return storage

Have it scoped however you want, or just mark as autouse=True and forget about it.

This simple hack is packaged as a pytest plugin to make it work with pytest-django, because there is no simpler way to inject custom code before it calls django.setup().

Do I even need this?

There's a good chance that you don't! But let's first recap on some of the ways to patch the storage, given a model somewhat like below:

class Contract(models.Model):
    document = models.FileField(storage=DocumentStorage())

Just patch the field

storage is an instance attribute of each FileField, so the most straightforward and robust way would be to just find the field instance and patch the attribute:

def test_document_upload(monkeypatch, my_fake_storage):
    document_field = Contract._meta.get_field_by_name("document")[0]
    monkeypatch.setattr(document_field, "storage", my_fake_storage)



The requirement here is that you know which fields of which models should be patched, and make sure to get it right for corresponding tests. This could become quite noisy and/or tedious to maintain. However, the approach is beneficial if there is quite a number of different storages, especially different storages used within the same model.

If the project uses a single storage for everything, or just a handful of them, such flexibility might be overkill.

Define a different storage for tests

Extract the name of the storage class into, use a different settings file for tests (which you should anyway) with a different value there. This way the storage is replaced before Django is even initialized.

DOCUMENT_STORAGE = 'myapp.storages.DocumentStorage'

DOCUMENT_STORAGE = 'myapp.tests.storages.FakeDocumentStorage'
class Contract(models.Model):
    document = models.FileField(

# tests/
def test_document_upload():
    document_field = Contract._meta.get_field_by_name("document")[0]



Patching not required, each field still has its own instance of a test storage, but all the objects using that field share the same storage for the whole test session. Which is already good enough and you may just stop here!

Patch all the storages, with configurable scope

This is what the plugin enables. It replaces FileField().storage, which is normally an instance attribute, with a property of the same name. A property is defined in a class, so this forces all the "storage" attribute lookups to be done on a class. You now don't have to hunt down every FileField instance and can just patch the class property instead.

The drawback is that any and all FileFields are patched for duration of the fixture's scope. Since the narrowest scope is function, this may (only) be a problem if a single test needs to check the behaviours of multiple storages in isolation.


As always, choose what works best for your specific project!

Manual patch Storage from settings This plugin
function-scoped storage x x
field-scoped storage x
no model changes required x x
low/no maintenance x x

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

pytest-django-filefield-0.1.2.tar.gz (3.8 kB view hashes)

Uploaded Source

Built Distribution

pytest_django_filefield-0.1.2-py2.py3-none-any.whl (3.8 kB view hashes)

Uploaded Python 2 Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page