CODEWITHSUNDEEP
pythonunit-testingpython-2.7mocking
Brian Bolton
Brian Bolton

Reputation: 3642

How to mock a function inside of another function?

I have been trying to mock this function call in another function with no success. How do I successfully mock this?

from mock import patch
from path.to.a import function_a

@patch("class_b.function_c")
def test_method(self, method_to_mock):
    method_to_mock.return_value = 7890
    result = function_a() #error - type object 'class_b' has no attribute 'function_c'

#another module -- "path.to.a module"
def function_a():
    return class_b.function_c()

#another module
class class_b(class_c):
    pass

#another module
class class_c():
    @classmethod
    def function_c():
        return 123

Upvotes: 5

Views: 3199

Answers (1)

Anthony Kong
Anthony Kong

Reputation: 40624

There are two issues with your code:

1) The class method is not declared correctly

class class_c():
    @classmethod
    def function_c(cls):
        return 123

2) The @patch is used incorrectly. You need to change it to 

def mock_method(cls):
    return 7890

# asssume the module name of class_b is modb
@patch("modb.class_b.function_c", new=classmethod(mock_method))
def test_method():
    result = function_a() 
    print result # check the result

Upvotes: 1

Related Questions