Python Mock Not Asserting Calls

I'm using the mock library to patch a class in a program that connects to a external resource and sends a dictioanry. The structure goes a litle like this... code.py def make_conn

Solution 1:

The connect() and send() methods are called on the return value of the first call; adjust your test accordingly:

mocked_conn.return_value.connect.assert_called_once()
mocked_conn.return_value.send.assert_called_with(param)

I usually store a reference to the 'instance' first:

@mock.path('code.OriginalClass')deftest_connection(self, mocked_conn):
    code.make_connection()
    mocked_conn.assert_called_with(host, port)
    mocked_instance = mocked_conn.return_value
    mocked_instance.connect.assert_called_once()
    mocked_instance.send.assert_called_with(param)