嘲諷鏈式調用在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)