26 lines
480 B
Python
26 lines
480 B
Python
import io
|
|
|
|
from pathod import log
|
|
from mitmproxy import exceptions
|
|
|
|
|
|
class DummyIO(io.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 exceptions.TcpDisconnect:
|
|
pass
|
|
assert "Test" in outf.getvalue()
|