diff options
| author | Mickaël Schoentgen <contact@tiger-222.fr> | 2019-01-06 15:06:30 +0100 |
|---|---|---|
| committer | Mickaël Schoentgen <contact@tiger-222.fr> | 2019-01-06 15:06:30 +0100 |
| commit | c03b07930ccf41b696ae02c363d116ba602313d3 (patch) | |
| tree | 66cb0f48d6de421a1f93f196716da1db81990eaa /test | |
| parent | 889987aa0a7f4852758ed09f70fe5d30f733a6d3 (diff) | |
| download | mitmproxy-c03b07930ccf41b696ae02c363d116ba602313d3.tar.gz mitmproxy-c03b07930ccf41b696ae02c363d116ba602313d3.tar.bz2 mitmproxy-c03b07930ccf41b696ae02c363d116ba602313d3.zip | |
Fix ResourceWarning: unclosed file, prevent resource leaks
Signed-off-by: Mickaël Schoentgen <contact@tiger-222.fr>
Diffstat (limited to 'test')
| -rw-r--r-- | test/bench/benchmark.py | 3 | ||||
| -rw-r--r-- | test/mitmproxy/addons/test_session.py | 3 |
2 files changed, 4 insertions, 2 deletions
diff --git a/test/bench/benchmark.py b/test/bench/benchmark.py index 84ec6005..076ad6c9 100644 --- a/test/bench/benchmark.py +++ b/test/bench/benchmark.py @@ -31,7 +31,8 @@ class Benchmark: stdout=asyncio.subprocess.PIPE ) stdout, _ = await traf.communicate() - open(ctx.options.benchmark_save_path + ".bench", mode="wb").write(stdout) + with open(ctx.options.benchmark_save_path + ".bench", mode="wb") as f: + f.write(stdout) ctx.log.error("Proxy saw %s requests, %s responses" % (self.reqs, self.resps)) ctx.log.error(stdout.decode("ascii")) backend.kill() diff --git a/test/mitmproxy/addons/test_session.py b/test/mitmproxy/addons/test_session.py index 20feb69d..97351426 100644 --- a/test/mitmproxy/addons/test_session.py +++ b/test/mitmproxy/addons/test_session.py @@ -68,7 +68,8 @@ class TestSession: os.remove(path) con = sqlite3.connect(path) script_path = pkg_data.path("io/sql/session_create.sql") - qry = open(script_path, 'r').read() + with open(script_path) as f: + qry = f.read() with con: con.executescript(qry) blob = b'blob_of_data' |