Factor out flow unit tests into speparate file.

1 files changed, 22 insertions, 0 deletions

diff --git a/test/utils.py b/test/utils.py
new file mode 100644
index 00000000..d07cb658
--- /dev/null
+++ b/test/utils.py
@@ -0,0 +1,22 @@
+from libmproxy import proxy, utils, filt, flow
+
+def treq(conn=None):
+    if not conn:
+        conn = proxy.BrowserConnection("address", 22)
+    headers = utils.Headers()
+    headers["header"] = ["qvalue"]
+    return proxy.Request(conn, "host", 80, "http", "GET", "/path", headers, "content")
+
+
+def tresp(req=None):
+    if not req:
+        req = treq()
+    headers = utils.Headers()
+    headers["header_response"] = ["svalue"]
+    return proxy.Response(req, 200, "HTTP/1.1", "message", headers, "content_response")
+
+
+def tflow():
+    bc = proxy.BrowserConnection("address", 22)
+    return flow.Flow(bc)
+