Find private keys within gzip-compresssed files

This commit is contained in:
Joe Wreschnig 2025-04-25 10:08:34 +02:00
parent 5c514f85cc
commit e2c9870bc2
3 changed files with 27 additions and 1 deletions

View file

@ -1,5 +1,7 @@
from __future__ import annotations
import gzip
import pytest
from pre_commit_hooks.detect_private_key import main
@ -26,3 +28,16 @@ def test_main(input_s, expected_retval, tmpdir):
path = tmpdir.join('file.txt')
path.write_binary(input_s)
assert main([str(path)]) == expected_retval
@pytest.mark.parametrize(('input_s', 'expected_retval'), TESTS)
def test_main_gzip(input_s, expected_retval, tmpdir):
path = tmpdir.join('file.txt.gz')
path.write_binary(gzip.compress(input_s))
assert main([str(path)]) == expected_retval
def test_main_gz_not_gzip(tmpdir):
path = tmpdir.join('file.txt.gz')
path.write_binary(b'not a sensitive value nor gzip')
assert main([str(path)]) == 0