Logo Questions Linux Laravel Mysql Ubuntu Git Menu

Mock a class with tedious __init__

I have a class that actually connects to a service and does authentication and stuff, but all of this is well tested somewhere else in my code, I just want to mock in the following test:

Object with tedious __init__

class B(object):
    def __init__(self, username, password):
        con = connect_to_service() # 
        auth = con.authenticate(username, password)

    def upload(self)
        return "Uploaded"

class A(object):
    def send():
       b = B('user', 'pass')


# Now, I want here to test A, and since it uses B, I need to mock it, but I can't get it to work.
def test_A():
    # Here I need to mock B and B.upload, I tried this:
    a = A()
    b = Mock()

But this test is failling because it reached auth() on B.init, which I want to be a Mock model.

like image 646
PepperoniPizza Avatar asked Mar 18 '23 04:03


1 Answers

I think that is a typical use case for mock.patch. Take care that patch decorator need the full path of the module where the main module become __main__

from mock import patch, Mock
from abmodule import A,  B #Where A and B are

def test_A(bmock):
    # Set B costructor to return a Mock object
    bmock.return_value = Mock()
    a = A()
    #Check if upload was called!

After that you could use use again the original B in the other part of the code: the patched version live just in scope of the function.

like image 186
Michele d'Amico Avatar answered Mar 26 '23 03:03

Michele d'Amico