Fix resource warning when using the CLI

Author: rhpvorderman

src/zlib_ng/gzip_ng.py

@@ -431,6 +431,7 @@ def main():
             if yes_or_no not in {"y", "Y", "yes"}:
                 sys.exit("not overwritten")
 
+    out_buffer = None
     if args.compress:
         if args.file is None:
             in_file = sys.stdin.buffer
@@ -470,6 +471,8 @@ def main():
             in_file.close()
         if out_file is not sys.stdout.buffer:
             out_file.close()
+        if out_buffer is not None and out_buffer is not sys.stdout.buffer:
+            out_buffer.close()
 
 
 if __name__ == "__main__":  # pragma: no cover

tests/test_gzip_ng.py

@@ -14,7 +14,6 @@
 import os
 import re
 import shutil
-import subprocess
 import sys
 import tempfile
 import zlib
@@ -28,21 +27,6 @@
 DATA = b'This is a simple test with gzip_ng'
 COMPRESSED_DATA = gzip.compress(DATA)
 TEST_FILE = str((Path(__file__).parent / "data" / "test.fastq.gz"))
-PYPY = sys.implementation.name == "pypy"
-
-
-def run_gzip_ng(*args, stdin=None):
-    """Calling gzip_ng externally seems to solve some issues on PyPy where
-    files would not be written properly when gzip_ng.main() was called. This is
-     probably due to some out of order execution that PyPy tries to pull.
-     Running the process externally is detrimental to the coverage report,
-     so this is only done for PyPy."""
-    process = subprocess.Popen(["python", "-m", "zlib_ng.gzip_ng", *args],
-                               stdout=subprocess.PIPE,
-                               stderr=subprocess.PIPE,
-                               stdin=subprocess.PIPE)
-
-    return process.communicate(stdin)
 
 
 def test_repr():
@@ -121,12 +105,9 @@ def test_decompress_infile_outfile(tmp_path, capsysbinary):
 def test_compress_infile_outfile(tmp_path, capsysbinary):
     test_file = tmp_path / "test"
     test_file.write_bytes(DATA)
-    if PYPY:
-        out, err = run_gzip_ng(str(test_file))
-    else:
-        sys.argv = ['', str(test_file)]
-        gzip_ng.main()
-        out, err = capsysbinary.readouterr()
+    sys.argv = ['', str(test_file)]
+    gzip_ng.main()
+    out, err = capsysbinary.readouterr()
     out_file = test_file.with_suffix(".gz")
     assert err == b''
     assert out == b''
@@ -189,12 +170,9 @@ def test_compress_infile_out_file(tmp_path, capsysbinary):
     test.write_bytes(DATA)
     out_file = tmp_path / "compressed.gz"
     args = ['-o', str(out_file), str(test)]
-    if PYPY:
-        out, err = run_gzip_ng(*args)
-    else:
-        sys.argv = ['', *args]
-        gzip_ng.main()
-        out, err = capsysbinary.readouterr()
+    sys.argv = ['', *args]
+    gzip_ng.main()
+    out, err = capsysbinary.readouterr()
     assert gzip.decompress(out_file.read_bytes()) == DATA
     assert err == b''
     assert out == b''
@@ -206,12 +184,9 @@ def test_compress_infile_out_file_force(tmp_path, capsysbinary):
     out_file = tmp_path / "compressed.gz"
     out_file.touch()
     args = ['-f', '-o', str(out_file), str(test)]
-    if PYPY:
-        out, err = run_gzip_ng(*args)
-    else:
-        sys.argv = ['', *args]
-        gzip_ng.main()
-        out, err = capsysbinary.readouterr()
+    sys.argv = ['', *args]
+    gzip_ng.main()
+    out, err = capsysbinary.readouterr()
     assert gzip.decompress(out_file.read_bytes()) == DATA
     assert err == b''
     assert out == b''
@@ -254,14 +229,11 @@ def test_compress_infile_out_file_inmplicit_name_prompt_accept(
     test.write_bytes(DATA)
     out_file = tmp_path / "test.gz"
     out_file.touch()
-    if PYPY:
-        out, err = run_gzip_ng(str(test), stdin=b"y\n")
-    else:
-        sys.argv = ['', str(test)]
-        mock_stdin = io.BytesIO(b"y")
-        sys.stdin = io.TextIOWrapper(mock_stdin)
-        gzip_ng.main()
-        out, err = capsysbinary.readouterr()
+    sys.argv = ['', str(test)]
+    mock_stdin = io.BytesIO(b"y")
+    sys.stdin = io.TextIOWrapper(mock_stdin)
+    gzip_ng.main()
+    out, err = capsysbinary.readouterr()
     assert b"already exists; do you wish to overwrite" in out
     assert err == b""
     assert gzip.decompress(out_file.read_bytes()) == DATA
@@ -271,13 +243,9 @@ def test_compress_infile_out_file_no_name(tmp_path, capsysbinary):
     test = tmp_path / "test"
     test.write_bytes(DATA)
     out_file = tmp_path / "compressed.gz"
-    args = ['-n', '-o', str(out_file), str(test)]
-    if PYPY:
-        out, err = run_gzip_ng(*args)
-    else:
-        sys.argv = ['', '-n', '-o', str(out_file), str(test)]
-        gzip_ng.main()
-        out, err = capsysbinary.readouterr()
+    sys.argv = ['', '-n', '-o', str(out_file), str(test)]
+    gzip_ng.main()
+    out, err = capsysbinary.readouterr()
     output = out_file.read_bytes()
     assert gzip.decompress(output) == DATA
     assert err == b''