python mock unittest：'self'参数缺少默认值(python mock unittest: 'self' parameter lackin

python mock unittest：'self'参数缺少默认值(python mock unittest: 'self' parameter lacking default value)

我试图测试一个调用另一个类方法的类方法。我已经看到了这个问题，但我认为这有点不同。我尝试了这个问题的答案，但它没有用。我继续得到这个错误：

AssertionError: Expected call: run(autocommit=False, parameters=None, sql='SELECT') Actual call: run(<plugins.bigsql.hooks.bigsql_hook.BigSqlHook object at 0x10b201050>, 'SELECT', False, parameters=None) 'self' parameter lacking default value

这是测试的代码：

import mock from plugins.bigsql.hooks import BigSqlHook from plugins.bigsql.operators import BigSqlOperator @mock.patch.object(BigSqlHook, 'run', autospec=True) def test_bigsql_hook(mock_hook_object): """Test BigSqlOperator connection is formed correctly.""" operator = BigSqlOperator(task_id='test', jdbc_conn_id='sqlite_default', sql='SELECT') operator.execute(context={}) mock_hook_object.assert_called_with(sql="SELECT", autocommit=False, parameters=None)

这是我尝试测试的类方法的代码：

import logging from airflow.operators.jdbc_operator import JdbcOperator from plugins.bigsql.hooks import BigSqlHook class BigSqlOperator(JdbcOperator): """Extends py:class:`airflow.hooks.jdbc_operator.JdbcOperator` to connect to IBM BigSQL. Notes: Works the same as JdbcOperator, just uses the BigSqlHook instead """ def execute(self, context): """Execute SQL statement on IBM BigSQL.""" logging.info('Executing: {0}'.format(str(self.sql))) hook = BigSqlHook(jdbc_conn_id=self.jdbc_conn_id) hook.run(self.sql, self.autocommit, parameters=self.parameters)

提前致谢！

I am trying to test a class method that calls another class method. I have seen this question, but I think this is slightly different. I tried the answer to that question, but it was not working. I continue to get this error:

Here is the code for the test:

Here is the code for the class method I am trying to test:

Thanks in advance!

最满意答案

我解决了我不得不删除auto_spec=True而不传入关键字参数。

这工作：

import mock from plugins.bigsql.hooks import BigSqlHook from plugins.bigsql.operators import BigSqlOperator @mock.patch.object(BigSqlHook, 'run') def test_bigsql_hook(mock_hook_object): """Test BigSqlOperator connection is formed correctly.""" operator = BigSqlOperator(task_id='test', jdbc_conn_id='sqlite_default', sql='SELECT') operator.execute(context={}) mock_hook_object.assert_called_with("SELECT", False, parameters=None)

I solved it. I had to remove the auto_spec=True and not pass in key word arguments.

This worked:

更多推荐