def test_normal(self, load_flow, tmpdir, data):
rf = readfile.ReadFileStdin()
with taddons.context():
tfile = tmpdir.join("tfile")
tfile.write(data.getvalue())
rf.load_flows_from_path(str(tfile))
assert load_flow.called
async def test_normal(self, tmpdir, data):
rf = readfile.ReadFileStdin()
with taddons.context(rf) as tctx:
tf = tmpdir.join("tfile")
with mock.patch('mitmproxy.master.Master.load_flow') as mck:
tf.write(data.getvalue())
tctx.configure(rf, rfile=str(tf))
mck.assert_not_awaited()
rf.running()
await asyncio.sleep(0)
mck.assert_awaited()
async def test_stdin(self, stdin, data, corrupt_data):
rf = readfile.ReadFileStdin()
with taddons.context(rf):
with mock.patch('mitmproxy.master.Master.load_flow') as mck:
stdin.buffer = data
mck.assert_not_awaited()
await rf.load_flows(stdin.buffer)
mck.assert_awaited()
stdin.buffer = corrupt_data
with pytest.raises(exceptions.FlowReadException):
await rf.load_flows(stdin.buffer)
def __init__(self, proxy_events, listen_port, include_path):
self.opts = options.Options()
self.opts.listen_port = listen_port
self.opts.confdir = include_path
self.master = master.Master(self.opts)
proxy_events.set_proxy(self)
self.master.addons.add(proxy_events)
self.master.addons.add(termlog.TermLog())
self.master.addons.add(*addons.default_addons())
self.master.addons.add(keepserving.KeepServing(),
readfile.ReadFileStdin())
def test_stdin(self, stdin, load_flow, data, corrupt_data):
rf = readfile.ReadFileStdin()
with taddons.context() as tctx:
stdin.buffer = data
tctx.configure(rf, rfile="-")
assert not load_flow.called
rf.running()
assert load_flow.called
stdin.buffer = corrupt_data
tctx.configure(rf, rfile="-")
with pytest.raises(exceptions.OptionsError):
rf.running()
def __init__(
self,
options: options.Options,
with_termlog=True,
with_dumper=True,
) -> None:
super().__init__(options)
self.errorcheck = ErrorCheck()
if with_termlog:
self.addons.add(termlog.TermLog())
self.addons.add(*addons.default_addons())
if with_dumper:
self.addons.add(dumper.Dumper())
self.addons.add(keepserving.KeepServing(), readfile.ReadFileStdin(),
self.errorcheck)
