嘲讽链式调用在Python

Question

我试图用unittest测试下面的类和mock库：

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的值。

Answer 1

假设Connection位于模块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)