1 files changed, 37 insertions, 0 deletions
diff --git a/test/mitmproxy/addons/test_clientplayback.py b/test/mitmproxy/addons/test_clientplayback.py
new file mode 100644
index 00000000..c5407d18
--- /dev/null
+++ b/test/mitmproxy/addons/test_clientplayback.py
@@ -0,0 +1,37 @@
+import mock
+
+from mitmproxy.addons import clientplayback
+from mitmproxy import options
+
+from .. import tutils, mastertest
+
+
+class TestClientPlayback:
+    def test_playback(self):
+        cp = clientplayback.ClientPlayback()
+        cp.configure(options.Options(), [])
+        assert cp.count() == 0
+        f = tutils.tflow(resp=True)
+        cp.load([f])
+        assert cp.count() == 1
+        RP = "mitmproxy.protocol.http_replay.RequestReplayThread"
+        with mock.patch(RP) as rp:
+            assert not cp.current
+            with mastertest.mockctx():
+                cp.tick()
+            rp.assert_called()
+            assert cp.current
+
+        cp.keepserving = False
+        cp.flows = None
+        cp.current = None
+        with mock.patch("mitmproxy.master.Master.shutdown") as sd:
+            with mastertest.mockctx():
+                cp.tick()
+            sd.assert_called()
+
+    def test_configure(self):
+        cp = clientplayback.ClientPlayback()
+        cp.configure(
+            options.Options(), []
+        )