我正在尝试使用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
具有我期望的值,我会断言?< / p>
我希望使用模拟库。在我的文档中,我有一个模拟链式调用的例子(http://www.voidspace.org.uk/python/mock/examples.html#mocking-chained-calls),但我不清楚如何使它适应我的问题。
Session.load(cookie)
方法在Session
实例中设置了一些属性。我想为我的测试中的每个cookie值设置固定值。
答案 0 :(得分:3)
假设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)