模拟一系列相互依赖的调用

我有一个方法,它可以抓取网页并将数据保存到文件中(参见下面的示例代码)。我需要测试结果数据的格式是否正确。

问题是,数据是从一系列调用中接收到的,而进一步的调用使用以前调用的结果。更糟糕的是,涉及的许多调用都是在同一对象上进行的(a Webdriver A WebDriverWait 以及 expected_conditions 模块),参数不同。

我明白了 unittest.mock.Mock 可以模拟一个简单调用或一系列简单调用的结果,但看不到如何实现这样的纠缠。我看到的唯一方法是手动重新实现方法所做的每个调用,并将方法中传递的参数复制到这些实现中,以便它们知道每次调用返回什么。对于其他的测试用例也要这样做。这听起来像是编写和维护的一场噩梦:代码数量是测试本身的几倍,与代码的重复率接近1:1。所以我拒绝继续,直到有人告诉我有更好的方法或者证明没有,并且每个人都这样做(我不相信),例如每次页面上的标签更改时重写所有测试(这是一个实现细节,所以通常不应该影响测试代码)。

样本代码(适用于 http://example.com ):

import selenium.webdriver
from selenium.webdriver.common.by import By as by
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.ui import WebDriverWait


def dump_accreditation_data(d, w, i, path):
    f = codecs.open(os.path.join(path, "%d.txt" % i), "w", encoding="utf-8")

    u = u'http://example.com/%s/accreditation' % i
    d.get(u)

    # page load
    w.until(EC.visibility_of_element_located((by.XPATH,"//p")))    #the real code has a more complex expression here with national characters
    w.until_not(EC.visibility_of_element_located((by.CSS_SELECTOR, '.waiter')))
    print >> f, u

    # organization name
    e = w.until(EC.visibility_of_element_located((
        by.CSS_SELECTOR, 'h1'
    )))
    org_name = e.text
    print >> f, org_name
    del e

    #etc
    e = d.find_element_by_xpath(u'//a[text()="More information..."')
    print >> f, e.get_attribute('href')

#How it's supposed to be used:
d = selenium.webdriver.Firefox()
w = WebDriverWait(d, 10)
dump_accreditation_data(d, w, 123, "<output_path>")