У меня есть функция, которая возвращает объект smb.SMBConnection.
В my_module.py
from smb.SMBConnection import SMBConnection
def get_connection(user, pwd, server):
return SMBConnection(user, pwd, socket.gethostname(), server, use_ntlm_v2=True)
Я пытаюсь использовать mock.patch
@mock.patch(‘smb.SMBConnection.SMBConnection’)
def test_get_connection(self, mock_class):
self.assertEquals(mock_class, my_module.get_connection)
Но это не работает