Python-为具有上下文管理器的类方法创建模拟测试

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了Python-为具有上下文管理器的类方法创建模拟测试，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1248字，纯文字阅读大概需要2分钟。

内容图文

我正在尝试为具有上下文管理器和许多调用的类函数的方法编写单元测试.我很难理解如何正确模拟该函数,以便可以测试返回值.我要模拟的类是db.正如您在下面看到的那样,我正在使用补丁程序,但是我不知道如何获取它以返回正确的方法调用.我正在得到一个通用的模拟函数,而不是我期望的返回值.

db_class.py

import db

class Foo():
    def __init__(self):
        pass
    def method(self):
        with db.a() as a:
            b = a.b
            return b.fetch()

unit_db.py

 from mock import Mock, patch, MagicMock
 from db_class import Foo

 @patch('db_class.db')
 def test(db_mock):
     expected_result = [5,10]
     db_mock.return_value = Mock(__enter__ = db_mock,
                                 __exit___ = Mock(),
                                 b = Mock(fetch=expected_result))

     foo = Foo()
     result = foo.method()
     assert result == expected_result

解决方法:

感谢评论者,我找到了适合我的解决方案.技巧是修补正确的类,在这种情况下,我想修补db_class.db.a而不是db_class.db.之后,重要的是要确保fetch()调用是一个方法(我认为我得到了正确的).对我来说,这个问题最棘手的部分是修补正确的东西,以及与上下文管理器打交道,这需要一些额外的修补工作.

@patch('db_class.db.a')
def test(db_a):
    expected_result = [5,10]
    b_fetch = MagicMock()
    b_fetch.fetch.return_value = expected_result 
    db_a.return_value = Mock(b = b_fetch,
                         __enter__= db_a,
                         __exit__ =Mock())
    foo = Foo()
    result = foo.method()
    assert result == expected_result

if __name__ == "__main__":
    test()

内容总结

以上是互联网集市为您收集整理的Python-为具有上下文管理器的类方法创建模拟测试全部内容，希望文章能够帮你解决Python-为具有上下文管理器的类方法创建模拟测试所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端