Mock Django database calls

gistfile1.py

"""
This is for a case where I was accessing the database connection/cursors directly,
and had need to mock up a result from a database call. The real trick here is
returning the cursor MagicMock object as a side effect of the initial patch.
"""

from unittest import mock


cursor = mock.MagicMock(
    description=(('foo',), ('bar',)),
    **{'fetchall.return_value': [('baz', 'frob')]}
)
with mock.patch('django.db.backends.utils.CursorWrapper', mock.MagicMock(side_effect=lambda x, y: cursor)):
    # Insert test code here.
    pass