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()