diff options
author | Aldo Cortesi <aldo@corte.si> | 2018-05-08 11:00:20 +1200 |
---|---|---|
committer | Aldo Cortesi <aldo@corte.si> | 2018-05-08 11:00:20 +1200 |
commit | 9b51393b6ddaec52298cdcf8e800312f41f809fd (patch) | |
tree | 8a1be4b2dd6bc1514f1bcf8010b6a42216f35815 | |
parent | f7d7e31f06dadcf0f7801ab1db055f8609c88a71 (diff) | |
download | mitmproxy-9b51393b6ddaec52298cdcf8e800312f41f809fd.tar.gz mitmproxy-9b51393b6ddaec52298cdcf8e800312f41f809fd.tar.bz2 mitmproxy-9b51393b6ddaec52298cdcf8e800312f41f809fd.zip |
benchmark: keep track of requests/responses seen by proxy
-rw-r--r-- | test/bench/benchmark.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/test/bench/benchmark.py b/test/bench/benchmark.py index 8d208088..84ec6005 100644 --- a/test/bench/benchmark.py +++ b/test/bench/benchmark.py @@ -11,6 +11,15 @@ class Benchmark: self.pr = cProfile.Profile() self.started = False + self.resps = 0 + self.reqs = 0 + + def request(self, f): + self.reqs += 1 + + def response(self, f): + self.resps += 1 + async def procs(self): ctx.log.error("starting benchmark") backend = await asyncio.create_subprocess_exec("devd", "-q", "-p", "10001", ".") @@ -23,6 +32,7 @@ class Benchmark: ) stdout, _ = await traf.communicate() open(ctx.options.benchmark_save_path + ".bench", mode="wb").write(stdout) + ctx.log.error("Proxy saw %s requests, %s responses" % (self.reqs, self.resps)) ctx.log.error(stdout.decode("ascii")) backend.kill() ctx.master.shutdown() |