How to mock an attribute of a callable instance in Python?

Question

I have the following code, which I need to test:

uploader = AjaxFileUploader(backend=XLSXFileUploadBackend, obj=obj)
response = uploader(request)
...
response.content

I'd like to mock AjaxFileUploader and replace the response.content with some value.

I do the following:

@patch('guinness.apps.home.views.AjaxFileUploader')
def test_import_asset_to_bucket(self, mock_file_uploader):
    test_content = 'a:"1"'
    mock_file_uploader.return_value = MagicMock(return_value=MagicMock(content=test_content))

Is there more elegant way to do it? Thanks!

Filipe Felisbino · Accepted Answer · 2022-07-13 21:59:22Z

2

You should be able to get away with it just doing

mock_file_uploader.return_value.return_value.content = test_content

because the attribute return_value returns a new Mock object which is created on first access, as described in the documentation.

I'm not sure this is much more elegant, though.

edited Jul 13, 2022 at 21:59

Filipe Felisbino

2,95228 silver badges26 bronze badges

answered Jun 24, 2015 at 10:58

poros

4064 silver badges12 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to mock an attribute of a callable instance in Python?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related