Improve XZ repair test

fox-it · Jul 3, 2024 · 81fe85e · 81fe85e
1 parent bd0f0a9
commit 81fe85e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/test_compression.py b/tests/test_compression.py
@@ -12,6 +12,8 @@
     xz,
 )
 
+import pytest
+
 
 def test_lz4_decompress():
     assert (
@@ -267,6 +269,9 @@ def test_xz_repair_checksum():
             "2972e8fd62b18ee300013a8020000000deadbeefdeadbeef020000000004595a"
         )
     )
-    repaired = xz.repair_checksum(buf)
 
+    with pytest.raises(lzma.LZMAError, match="Corrupt input data"):
+        lzma.decompress(buf.getvalue())
+
+    repaired = xz.repair_checksum(buf)
     assert lzma.decompress(repaired.read()) == b"test" * 1024