1 files changed, 33 insertions, 0 deletions
diff --git a/test/mitmproxy/mastertest.py b/test/mitmproxy/mastertest.py
new file mode 100644
index 00000000..9bb8826d
--- /dev/null
+++ b/test/mitmproxy/mastertest.py
@@ -0,0 +1,33 @@
+import tutils
+import netlib.tutils
+import mock
+
+from mitmproxy import flow, proxy, models
+
+
+class MasterTest:
+    def cycle(self, master, content):
+        f = tutils.tflow(req=netlib.tutils.treq(content=content))
+        l = proxy.Log("connect")
+        l.reply = mock.MagicMock()
+        master.log(l)
+        master.clientconnect(f.client_conn)
+        master.serverconnect(f.server_conn)
+        master.request(f)
+        if not f.error:
+            f.response = models.HTTPResponse.wrap(netlib.tutils.tresp(content=content))
+            f = master.response(f)
+        master.clientdisconnect(f.client_conn)
+        return f
+
+    def dummy_cycle(self, master, n, content):
+        for i in range(n):
+            self.cycle(master, content)
+        master.shutdown()
+
+    def flowfile(self, path):
+        f = open(path, "wb")
+        fw = flow.FlowWriter(f)
+        t = tutils.tflow(resp=True)
+        fw.add(t)
+        f.close()