aboutsummaryrefslogtreecommitdiffstats
path: root/mitmproxy/test/tflow.py
diff options
context:
space:
mode:
Diffstat (limited to 'mitmproxy/test/tflow.py')
-rw-r--r--mitmproxy/test/tflow.py20
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