我正在尝试使用 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
是否具有我期望的值?我希望使用mock库。在其文档中,我有一个关于模拟链接调用的示例(http://www.voidspace.org.uk/python/mock/examples.html#mocking-chained-calls),但它并不清楚我如何将其适应到我的问题上。 Session.load(cookie)
方法设置Session
实例中的一些属性。我想为我的测试设置这些值,以便对于每个cookie值都是固定的。