minor improvements

This commit is contained in:
Maximilian Hils 2017-06-01 23:35:18 +02:00
parent 2ba0791810
commit 5e17b5f62e
4 changed files with 21 additions and 12 deletions

View File

@ -4,6 +4,9 @@ class LogEntry:
self.msg = msg self.msg = msg
self.level = level self.level = level
def __repr__(self):
return "LogEntry({}, {})".format(self.msg, self.level)
class Log: class Log:
""" """

View File

@ -17,6 +17,8 @@ class TestAddons(addonmanager.AddonManager):
def trigger(self, event, *args, **kwargs): def trigger(self, event, *args, **kwargs):
if event == "log": if event == "log":
self.master.logs.append(args[0]) self.master.logs.append(args[0])
elif event == "tick" and not args and not kwargs:
pass
else: else:
self.master.events.append((event, args, kwargs)) self.master.events.append((event, args, kwargs))
super().trigger(event, *args, **kwargs) super().trigger(event, *args, **kwargs)

View File

@ -1,28 +1,27 @@
import os import os
import socket import socket
import time import time
import pytest
from unittest import mock from unittest import mock
from mitmproxy.test import tutils import pytest
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 mitmproxy.net.http import mitmproxy.net.http
from mitmproxy.net import tcp
from mitmproxy.net import socks
from mitmproxy import certs from mitmproxy import certs
from mitmproxy import exceptions 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.net.http import http1
from mitmproxy.proxy.config import HostMatcher
from mitmproxy.test import tutils
from pathod import pathoc from pathod import pathoc
from pathod import pathod from pathod import pathod
from .. import tservers from .. import tservers
from ...conftest import skip_appveyor from ...conftest import skip_appveyor
""" """
Note that the choice of response code in these tests matters more than you 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 might think. libcurl treats a 304 response code differently from, say, a

View File

@ -1 +1,6 @@
# TODO: write tests from mitmproxy import log
def test_logentry():
e = log.LogEntry("foo", "info")
assert repr(e) == "LogEntry(foo, info)"