mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-23 08:11:00 +00:00
26 lines
494 B
Python
26 lines
494 B
Python
from pathod import log
|
|
from netlib.exceptions import TcpDisconnect
|
|
|
|
from six.moves import cStringIO
|
|
|
|
|
|
class DummyIO(cStringIO):
|
|
|
|
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, rw, rw)
|
|
try:
|
|
with l.ctx() as lg:
|
|
lg("Test")
|
|
except TcpDisconnect:
|
|
pass
|
|
assert "Test" in outf.getvalue()
|