diff options
author | Maximilian Hils <git@maximilianhils.com> | 2017-06-01 23:35:18 +0200 |
---|---|---|
committer | Maximilian Hils <git@maximilianhils.com> | 2017-06-02 00:01:37 +0200 |
commit | 5e17b5f62e87970d4cdb7ef1c5176f3e295eb16e (patch) | |
tree | afc276f1e5bae9b38b84c4833351fe2086b89893 | |
parent | 2ba0791810cc7f843c49249c1107a9886da66f0f (diff) | |
download | mitmproxy-5e17b5f62e87970d4cdb7ef1c5176f3e295eb16e.tar.gz mitmproxy-5e17b5f62e87970d4cdb7ef1c5176f3e295eb16e.tar.bz2 mitmproxy-5e17b5f62e87970d4cdb7ef1c5176f3e295eb16e.zip |
minor improvements
-rw-r--r-- | mitmproxy/log.py | 3 | ||||
-rw-r--r-- | mitmproxy/test/taddons.py | 2 | ||||
-rw-r--r-- | test/mitmproxy/proxy/test_server.py | 21 | ||||
-rw-r--r-- | test/mitmproxy/test_log.py | 7 |
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)" |