模拟类与方法

内容来源于 Stack Overflow,并遵循CC BY-SA 3.0许可协议进行翻译与使用

  • 回答 (1)
  • 关注 (0)
  • 查看 (27)

我想要模拟python对象-心理学2-来测试我的类。

当我打印物体时-dir(conn),我看到连接,附加,等等,都是被列出来的。

conn = mock.MagicMock(psycopg2.connect)

结果:

['BINARY', 'Binary', 'DATETIME', 'DataError', 'DatabaseError', 'Date',
'DateFromTicks', 'Error', 'IntegrityError', 'InterfaceError',
'InternalError', 'NUMBER', 'NotSupportedError', 'OperationalError',
'ProgrammingError', 'ROWID', 'STRING', 'Time', 'TimeFromTicks',
'Timestamp', 'TimestampFromTicks', 'Warning', '__builtins__',
'__doc__', '__file__', '__libpq_version__', '__name__', '__package__',
'__path__', '__version__', '__warningregistry__', '_connect', '_ext',
'_ipaddress', '_json', '_psycopg', '_range', 'apilevel',
'assert_any_call', 'assert_called', 'assert_called_once',
'assert_called_once_with', 'assert_called_with', 'assert_has_calls',
'assert_not_called', 'attach_mock', 'call_args', 'call_args_list',
'call_count', 'called', 'configure_mock', 'connect', 'extensions',
'extras', 'method_calls', 'mock_add_spec', 'mock_calls', 'paramstyle',
'reset_mock', 'return_value', 'side_effect', 'threadsafety', 'tz',
'warn']

现在,当我想要站在_执行方法返回值的效果。

conn.cursor().execute().side_effect = psycopg2.DatabaseError

我知道这个错误:

AttributeError: Mock object has no attribute 'cursor'

新代码:

pg = mock.MagicMock(psycopg2)
print dir(pg())
print dir(pg.connect)
pg.connect()

输出量

AttributeError: Mock object has no attribute 'cursor'
提问于
用户回答回答于

属性的返回值。conn,不是conn本身,因为正在模拟该函数。connect

conn().cursor().execute().side_effect = psycopg2.DatabaseError

扫码关注云+社区

领取腾讯云代金券