aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--mitmproxy/log.py3
-rw-r--r--mitmproxy/test/taddons.py2
-rw-r--r--test/mitmproxy/proxy/test_server.py21
-rw-r--r--test/mitmproxy/test_log.py7
4 files changed, 21 insertions, 12 deletions
diff --git a/mitmproxy/log.py b/mitmproxy/log.py
index a2e7ea99..886b1449 100644
--- a/mitmproxy/log.py
+++ b/mitmproxy/log.py
@@ -4,6 +4,9 @@ class LogEntry:
self.msg = msg
self.level = level
+ def __repr__(self):
+ return "LogEntry({}, {})".format(self.msg, self.level)
+
class Log:
"""
diff --git a/mitmproxy/test/taddons.py b/mitmproxy/test/taddons.py
index 5680e847..49142871 100644
--- a/mitmproxy/test/taddons.py
+++ b/mitmproxy/test/taddons.py
@@ -17,6 +17,8 @@ class TestAddons(addonmanager.AddonManager):
def trigger(self, event, *args, **kwargs):
if event == "log":
self.master.logs.append(args[0])
+ elif event == "tick" and not args and not kwargs:
+ pass
else:
self.master.events.append((event, args, kwargs))
super().trigger(event, *args, **kwargs)
diff --git a/test/mitmproxy/proxy/test_server.py b/test/mitmproxy/proxy/test_server.py
index 8381e511..bd61f600 100644
--- a/test/mitmproxy/proxy/test_server.py
+++ b/test/mitmproxy/proxy/test_server.py
@@ -1,28 +1,27 @@
import os
import socket
import time
-import pytest
from unittest import mock
-from mitmproxy.test import tutils
-from mitmproxy import options
-from mitmproxy.addons import script
-from mitmproxy.addons import proxyauth
-from mitmproxy import http
-from mitmproxy.proxy.config import HostMatcher
+import pytest
+
import mitmproxy.net.http
-from mitmproxy.net import tcp
-from mitmproxy.net import socks
from mitmproxy import certs
from mitmproxy import exceptions
+from mitmproxy import http
+from mitmproxy import options
+from mitmproxy.addons import proxyauth
+from mitmproxy.addons import script
+from mitmproxy.net import socks
+from mitmproxy.net import tcp
from mitmproxy.net.http import http1
+from mitmproxy.proxy.config import HostMatcher
+from mitmproxy.test import tutils
from pathod import pathoc
from pathod import pathod
-
from .. import tservers
from ...conftest import skip_appveyor
-
"""
Note that the choice of response code in these tests matters more than you
might think. libcurl treats a 304 response code differently from, say, a
diff --git a/test/mitmproxy/test_log.py b/test/mitmproxy/test_log.py
index 777ab4dd..cde679ed 100644
--- a/test/mitmproxy/test_log.py
+++ b/test/mitmproxy/test_log.py
@@ -1 +1,6 @@
-# TODO: write tests
+from mitmproxy import log
+
+
+def test_logentry():
+ e = log.LogEntry("foo", "info")
+ assert repr(e) == "LogEntry(foo, info)"