aboutsummaryrefslogtreecommitdiffstats
path: root/test
diff options
context:
space:
mode:
authorMaximilian Hils <git@maximilianhils.com>2015-08-30 01:21:58 +0200
committerMaximilian Hils <git@maximilianhils.com>2015-08-30 01:21:58 +0200
commitdd7f50d64bef38fa67b4cace91913d03691dde26 (patch)
treec183aa1f286ec191a094713d1f697f2fa54ab385 /test
parent100ea27c30d89b895a02a1b128edc5472ab84b3e (diff)
downloadmitmproxy-dd7f50d64bef38fa67b4cace91913d03691dde26.tar.gz
mitmproxy-dd7f50d64bef38fa67b4cace91913d03691dde26.tar.bz2
mitmproxy-dd7f50d64bef38fa67b4cace91913d03691dde26.zip
restructure code, remove cruft
Diffstat (limited to 'test')
-rw-r--r--test/test_dump.py4
-rw-r--r--test/test_proxy.py22
-rw-r--r--test/tservers.py1
3 files changed, 10 insertions, 17 deletions
diff --git a/test/test_dump.py b/test/test_dump.py
index b3d724a5..b05f6a0f 100644
--- a/test/test_dump.py
+++ b/test/test_dump.py
@@ -5,8 +5,8 @@ import netlib.tutils
from netlib.http.semantics import CONTENT_MISSING
from libmproxy import dump, flow
-from libmproxy.protocol import http, http_wrappers
-from libmproxy.proxy.primitives import Log
+from libmproxy.protocol import http_wrappers
+from libmproxy.proxy import Log
import tutils
import mock
diff --git a/test/test_proxy.py b/test/test_proxy.py
index fac4a4f4..301ce2ca 100644
--- a/test/test_proxy.py
+++ b/test/test_proxy.py
@@ -1,8 +1,8 @@
from libmproxy import cmdline
-from libmproxy.proxy import ProxyConfig, process_proxy_options
+from libmproxy.proxy import ProxyConfig
+from libmproxy.proxy.config import process_proxy_options
from libmproxy.proxy.connection import ServerConnection
-from libmproxy.proxy.primitives import ProxyError
-from libmproxy.proxy.server import DummyServer, ProxyServer, ConnectionHandler2
+from libmproxy.proxy.server import DummyServer, ProxyServer, ConnectionHandler
import tutils
from libpathod import test
from netlib import http, tcp
@@ -11,11 +11,6 @@ import mock
from OpenSSL import SSL
-def test_proxy_error():
- p = ProxyError(111, "msg")
- assert str(p)
-
-
class TestServerConnection:
def setUp(self):
self.d = test.Daemon()
@@ -177,12 +172,11 @@ class TestConnectionHandler:
root_layer = mock.Mock()
root_layer.side_effect = RuntimeError
config.mode.return_value = root_layer
- c = ConnectionHandler2(
- config,
+ c = ConnectionHandler(
mock.MagicMock(),
- ("127.0.0.1",
- 8080),
- None,
- mock.MagicMock())
+ ("127.0.0.1", 8080),
+ config,
+ mock.MagicMock()
+ )
with tutils.capture_stderr(c.handle) as output:
assert "mitmproxy has crashed" in output
diff --git a/test/tservers.py b/test/tservers.py
index dfd3f627..c5256e53 100644
--- a/test/tservers.py
+++ b/test/tservers.py
@@ -7,7 +7,6 @@ import mock
from libmproxy.proxy.config import ProxyConfig
from libmproxy.proxy.server import ProxyServer
-from libmproxy.proxy.primitives import TransparentProxyMode
import libpathod.test
import libpathod.pathoc
from libmproxy import flow, controller