ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

在Python中模拟链接的呼叫

2019-12-08 16:55:35  阅读:289  来源: 互联网

标签:python-mock unit-testing mocking python


我正在尝试使用unittest和模拟库测试以下类:

class Connection(object):
    def __init__(self, cookie):
    self.connect = None
    self.session = Session()
    self.session.load(cookie)
    # do some stuff with self.session
    self.some_info = self.session.data['the_info']

在创建Connection实例时,如何根据Session实例的返回值断言self.some_info是否具有我期望的值,该如何测试?

我希望使用模拟库.在其文档中,我有一个模拟链接调用(http://www.voidspace.org.uk/python/mock/examples.html#mocking-chained-calls)的示例,但尚不清楚如何使它适应我的问题.

Session.load(cookie)方法在Session实例中设置一些属性.我想为cookie的每个值设置固定的测试值.

解决方法:

假设连接位于模块package.module.connection中

以下代码应该是测试会话的方式:

import mock


class TestConnection(unittest.TestCase):

    @mock.patch('package.module.connection.Session')
    def test_some_info_on_session_is_set(self, fake_session):
        fake_session.data = {'the_info': 'blahblah'}
        cookie = Cookie()
        connection = Connection(cookie)
        self.assertEqual(connection.some_info, 'blahblah')
        fake_session.load.assert_called_once_with(cookie)

标签:python-mock,unit-testing,mocking,python
来源: https://codeday.me/bug/20191208/2092820.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有