mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-27 02:24:18 +00:00
23 lines
479 B
Python
23 lines
479 B
Python
import StringIO
|
|
from libpathod import log
|
|
import netlib.tcp
|
|
|
|
|
|
class DummyIO(StringIO.StringIO):
|
|
def start_log(self, *args, **kwargs):
|
|
pass
|
|
|
|
def get_log(self, *args, **kwargs):
|
|
return ""
|
|
|
|
|
|
def test_disconnect():
|
|
outf = DummyIO()
|
|
rw = DummyIO()
|
|
try:
|
|
with log.Log(outf, False, rw, rw) as lg:
|
|
raise netlib.tcp.NetLibDisconnect()
|
|
except netlib.tcp.NetLibDisconnect:
|
|
pass
|
|
assert "Disconnected" in outf.getvalue()
|