diff options
Diffstat (limited to 'test')
-rw-r--r-- | test/test_console.py | 2 | ||||
-rw-r--r-- | test/test_flow.py | 4 |
2 files changed, 1 insertions, 5 deletions
diff --git a/test/test_console.py b/test/test_console.py index 4fd9bb9f..20534177 100644 --- a/test/test_console.py +++ b/test/test_console.py @@ -16,7 +16,7 @@ class TestConsoleState: """ c = console.ConsoleState() f = self._add_request(c) - assert f.request in c._flow_map + assert f in c._flow_list assert c.get_focus() == (f, 0) def test_focus(self): diff --git a/test/test_flow.py b/test/test_flow.py index aec04152..42118e36 100644 --- a/test/test_flow.py +++ b/test/test_flow.py @@ -355,23 +355,19 @@ class TestState: f = c.add_request(req) assert f assert c.flow_count() == 1 - assert c._flow_map.get(req) assert c.active_flow_count() == 1 newreq = tutils.treq() assert c.add_request(newreq) - assert c._flow_map.get(newreq) assert c.active_flow_count() == 2 resp = tutils.tresp(req) assert c.add_response(resp) assert c.flow_count() == 2 - assert c._flow_map.get(resp.request) assert c.active_flow_count() == 1 unseen_resp = tutils.tresp() assert not c.add_response(unseen_resp) - assert not c._flow_map.get(unseen_resp.request) assert c.active_flow_count() == 1 resp = tutils.tresp(newreq) |