我正在测试一些处理tweets的类。我使用sixohsix twitter来处理twitter API。
我有一个类作为Twitter类的门面,我的想法是模拟实际的sixohix类,通过随机生成新的tweet或从数据库中检索它们来模拟tweet的到达。在
我的门面看起来像:from twitter import TwitterStream
class TwitterFacade(object):
def __init__(self, dev='soom'):
self._auth = OAuth(dev_keys["ACCESS_TOKEN"], dev_keys["ACCESS_SECRET"], dev_keys["CONSUMER_KEY"], dev_keys["CONSUMER_SECRET"])
def tweets(self, callback=None, users=[], terms=[], locations=[], count=5):
t = TwitterStream(auth=self._auth)
args = {}
if users: args['follow'] = ",".join(users)
if terms: args['track'] = ",".join(terms)
if locations: args['locations'] = ",".join(str(l) for l in locations)
# this controls the general loop, it re-enters if there was an exception,
# otherwise the for loop should take care of looping trough the tweets
cant = count
while cant > 0:
try:
iterator = t.statuses.filter(**args)
for twit in iterator:
if twit.get('text'):
callback(twit)
cant -= 1
if cant == 0:
iterator.close()
break
except Exception as e:
print e
#some error handling code
所以,如果在一个单元测试中,我想测试一些对tweets做一些事情的模块,我该如何模拟TwitterStream类呢?
我试过使用Mock:
^{pr2}$
这不起作用,仍在调用twitter api。我本以为,自从没有向mock类添加任何代码之后,我会得到一个错误或其他什么,而是调用了sixhoh6个twitter类。在