diff options
Diffstat (limited to 'mitmproxy/test/tflow.py')
| -rw-r--r-- | mitmproxy/test/tflow.py | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/mitmproxy/test/tflow.py b/mitmproxy/test/tflow.py index edf4d7a7..5749bd52 100644 --- a/mitmproxy/test/tflow.py +++ b/mitmproxy/test/tflow.py @@ -115,6 +115,26 @@ def tflow(client_conn=True, server_conn=True, req=True, resp=None, err=None): return f +def tdummyflow(client_conn=True, server_conn=True, err=None): + class DummyFlow(flow.Flow): + """A flow that is neither HTTP nor TCP.""" + + def __init__(self, client_conn, server_conn, live=None): + super().__init__("dummy", client_conn, server_conn, live) + + if client_conn is True: + client_conn = tclient_conn() + if server_conn is True: + server_conn = tserver_conn() + if err is True: + err = terr() + + f = DummyFlow(client_conn, server_conn) + f.error = err + f.reply = controller.DummyReply() + return f + + def tclient_conn(): """ @return: mitmproxy.proxy.connection.ClientConnection |
