python 电话呼叫_在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是否具有我期望的值,该如何测试？

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