Django ORM - mock values().filter() chain

Question

I am trying to mock a chained call on the Djangos model.Manager() class. For now I want to mock the values() and filter() method.

Answer 1

Try this:

import mock
from mocktest.mockme.models import MyModel

class SimpleTest(TestCase):
    def test_chained_query(self):
        my_model_value_mock = mock.patch(MyModel.objects, 'value')
        my_model_value_mock.return_value.filter.return_value.count.return_value = 10000
        self.assertTrue(my_model_value_mock.return_value.filter.return_value.count.called)