Django Mock isn't working as expected in Unit test

Question

I'm having a view in which there is a method called can_upload_file that makes external calls. This method is called from view's post (create) method. So, in my unit test I'm trying to test the post method and at the same time mock can_upload_file method, as follows:

DatasetViewSet._can_upload_file = MagicMock()
DatasetViewSet._can_upload_file = "valid"
response = self.client.post(self.url, self.dataset, format="multipart")

However I see that the actual can_upload_file method in the view is still executed.

How can I fix this?

Answer 1

You can specify the return value of a mocked method like this:

DatasetViewSet._can_upload_file = MagicMock(return_value="valid")

• Neo: thanks a ton @blhsing I was struggling with this for hours together and was really losing it. Much thanks again..!!