From 57f46fa0996710ed1d8766ef8eb05ee0706fb8aa Mon Sep 17 00:00:00 2001 From: Anne Borcherding Date: Thu, 30 Apr 2020 14:22:13 +0200 Subject: [PATCH] moved addon for selenium authentication to new sub-folder of examples/complex. Co-authored-by: weichweich <14820950+weichweich@users.noreply.github.com> --- examples/complex/webscanner_helper/__init__.py | 0 .../complex/{ => webscanner_helper}/proxyauth_selenium.py | 0 test/examples/test_proxyauth_selenium.py | 8 ++++---- 3 files changed, 4 insertions(+), 4 deletions(-) create mode 100644 examples/complex/webscanner_helper/__init__.py rename examples/complex/{ => webscanner_helper}/proxyauth_selenium.py (100%) diff --git a/examples/complex/webscanner_helper/__init__.py b/examples/complex/webscanner_helper/__init__.py new file mode 100644 index 000000000..e69de29bb diff --git a/examples/complex/proxyauth_selenium.py b/examples/complex/webscanner_helper/proxyauth_selenium.py similarity index 100% rename from examples/complex/proxyauth_selenium.py rename to examples/complex/webscanner_helper/proxyauth_selenium.py diff --git a/test/examples/test_proxyauth_selenium.py b/test/examples/test_proxyauth_selenium.py index c0c4363bc..c552355c4 100644 --- a/test/examples/test_proxyauth_selenium.py +++ b/test/examples/test_proxyauth_selenium.py @@ -7,7 +7,7 @@ from mitmproxy.test import tflow from mitmproxy.test import tutils from mitmproxy.http import HTTPFlow -from examples.complex.proxyauth_selenium import logger, randomString, AuthorizationOracle, SeleniumAddon +from examples.complex.webscanner_helper.proxyauth_selenium import logger, randomString, AuthorizationOracle, SeleniumAddon class TestRandomString: @@ -79,7 +79,7 @@ class TestSeleniumAddon: selenium_addon.set_cookies = False assert not selenium_addon.set_cookies with mock.patch.object(logger, 'debug') as mock_debug: - with mock.patch('examples.complex.proxyauth_selenium.SeleniumAddon.login', + with mock.patch('examples.complex.webscanner_helper.proxyauth_selenium.SeleniumAddon.login', return_value=[{"name": "cookie", "value": "test"}]) as mock_login: selenium_addon.request(f) mock_debug.assert_called() @@ -104,7 +104,7 @@ class TestSeleniumAddon: f = tflow.tflow(resp=tutils.tresp()) f.request.url = "http://example.com/login.php" selenium_addon.set_cookies = False - with mock.patch('examples.complex.proxyauth_selenium.SeleniumAddon.login', return_value=[]) as mock_login: + with mock.patch('examples.complex.webscanner_helper.proxyauth_selenium.SeleniumAddon.login', return_value=[]) as mock_login: selenium_addon.response(f) mock_login.assert_called() @@ -112,7 +112,7 @@ class TestSeleniumAddon: f = tflow.tflow(resp=tutils.tresp()) f.request.url = "http://example.com/login.php" selenium_addon.set_cookies = False - with mock.patch('examples.complex.proxyauth_selenium.SeleniumAddon.login', + with mock.patch('examples.complex.webscanner_helper.proxyauth_selenium.SeleniumAddon.login', return_value=[{"name": "cookie", "value": "test"}]) as mock_login: selenium_addon.response(f) mock_login.assert_called()