From ca3e98640c2344da356700635c8d84be1606b411 Mon Sep 17 00:00:00 2001
From: Maximilian Hils <git@maximilianhils.com>
Date: Mon, 21 Aug 2017 18:49:51 +0200
Subject: fix mitmweb master shutdown, tests++

---
 test/mitmproxy/data/addonscripts/shutdown.py |  5 +++++
 test/mitmproxy/tools/test_main.py            | 19 +++++++++++++++++++
 2 files changed, 24 insertions(+)
 create mode 100644 test/mitmproxy/data/addonscripts/shutdown.py
 create mode 100644 test/mitmproxy/tools/test_main.py

(limited to 'test')

diff --git a/test/mitmproxy/data/addonscripts/shutdown.py b/test/mitmproxy/data/addonscripts/shutdown.py
new file mode 100644
index 00000000..51a99b5c
--- /dev/null
+++ b/test/mitmproxy/data/addonscripts/shutdown.py
@@ -0,0 +1,5 @@
+from mitmproxy import ctx
+
+
+def running():
+    ctx.master.shutdown()
diff --git a/test/mitmproxy/tools/test_main.py b/test/mitmproxy/tools/test_main.py
new file mode 100644
index 00000000..88e2fe86
--- /dev/null
+++ b/test/mitmproxy/tools/test_main.py
@@ -0,0 +1,19 @@
+from mitmproxy.test import tutils
+from mitmproxy.tools import main
+
+shutdown_script = tutils.test_data.path("mitmproxy/data/addonscripts/shutdown.py")
+
+
+def test_mitmweb():
+    main.mitmweb([
+        "--no-web-open-browser",
+        "-q",
+        "-s", shutdown_script
+    ])
+
+
+def test_mitmdump():
+    main.mitmdump([
+        "-q",
+        "-s", shutdown_script
+    ])
-- 
cgit v1.2.3