diff options
Diffstat (limited to 'test')
-rw-r--r-- | test/test_flow.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/test/test_flow.py b/test/test_flow.py index 947d0d59..91e3f5f4 100644 --- a/test/test_flow.py +++ b/test/test_flow.py @@ -168,18 +168,20 @@ class uFlow(libpry.AutoTree): assert f == f2 def test_kill(self): + s = flow.State() + fm = flow.FlowMaster(None, s) f = tutils.tflow() f.request = tutils.treq() f.intercept() assert not f.request.acked - f.kill() + f.kill(fm) assert f.request.acked f.intercept() f.response = tutils.tresp() f.request = f.response.request f.request.ack() assert not f.response.acked - f.kill() + f.kill(fm) assert f.response.acked def test_accept_intercept(self): |