Fix flow formatting unit tests.

This commit is contained in:
Aldo Cortesi 2011-03-14 13:49:30 +13:00
parent 897bd5c2b8
commit e35fd7126b

View File

@ -101,11 +101,9 @@ class uformat_flow(libpry.AutoTree):
f.request = f.response.request
f.backup()
assert ('method', '[edited] ') in console.format_flow(f, True)
assert ('method', '[edited] ') in console.format_flow(f, True, True)
f.request.set_replay()
assert ('method', '[replay] ') in console.format_flow(f, True)
assert ('method', '[replay] ') in console.format_flow(f, True, True)
assert ('method', '[replay]') in console.format_flow(f, True)
assert ('method', '[replay]') in console.format_flow(f, True, True)
f.response.code = 404
assert ('error', '404') in console.format_flow(f, True, True)