Утверждая положение with..as в Python

Question

Итак, имея unittest.mock и код

with pysftp.Connection(host, username, password, port) as sftp: 
      sftp.get("filename") 

... можно утверждать Connection.assert_called_with(host, username, password, port). Но утверждение Connection.assert_called_with(host, username, password, port).get("filename") не удается.

Как это утверждать?

Answer 1

Я предполагаю, что вы хотели бы patchpysftp.Connection ссылка.

Вот пример того, как вы можете сделать свой тест:

>>> import pysftp 
>>> from mock import * 

>>> with patch("pysftp.Connection") as mock_connection: 
...  with pysftp.Connection("1.2.3.4", "user", "pwd", 12345) as sftp: 
...   sftp.get("filename") 
...  mock_connection.assert_called_with("1.2.3.4", "user", "pwd", 12345) 
...  sftp.get.assert_called_with("filename") 
... 
<MagicMock name='Connection().__enter__().get()' id='139907730030992'> 

Поскольку sftp объект является MagicMock вы можете использовать его непосредственно. Как вы можете видеть в журнале sftp точно так же из

mock_connection.return_value.__enter__.return_value 

• Connection ->mock_connection
• () ->return_value