mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-23 08:11:00 +00:00
2cf79b7912
Pathoc is an interactive tool, no need for a long leading timestamp. More generally, make timestamps optional in the logging mechanism so we can configure this with command-line flags or something down the track.
26 lines
482 B
Python
26 lines
482 B
Python
from pathod import log
|
|
from netlib.exceptions import TcpDisconnect
|
|
|
|
import six
|
|
|
|
|
|
class DummyIO(six.StringIO):
|
|
|
|
def start_log(self, *args, **kwargs):
|
|
pass
|
|
|
|
def get_log(self, *args, **kwargs):
|
|
return ""
|
|
|
|
|
|
def test_disconnect():
|
|
outf = DummyIO()
|
|
rw = DummyIO()
|
|
l = log.ConnectionLogger(outf, False, True, rw, rw)
|
|
try:
|
|
with l.ctx() as lg:
|
|
lg("Test")
|
|
except TcpDisconnect:
|
|
pass
|
|
assert "Test" in outf.getvalue()
|