diff options
Diffstat (limited to 'tests/test_compression.py')
| -rw-r--r-- | tests/test_compression.py | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/tests/test_compression.py b/tests/test_compression.py index 6f443cc..a81e80f 100644 --- a/tests/test_compression.py +++ b/tests/test_compression.py @@ -160,6 +160,18 @@ ADUdSd6zBOkOpekGFH46zix9wE9VT65OVeV479//7uUAAA== except gemato.exceptions.UnsupportedCompression: raise unittest.SkipTest('lzma compression unsupported') + def test_lzma_legacy_as_xz(self): + """ + Test that the class rejects mislabel files. + """ + if gemato.compression.lzma is None: + raise unittest.SkipTest('xz compression unsupported') + + with io.BytesIO(base64.b64decode(self.BASE64)) as f: + with self.assertRaises(gemato.compression.lzma.LZMAError): + with gemato.compression.open_compressed_file('xz', f, "rb") as xz: + xz.read() + class XZCompressionTest(unittest.TestCase): BASE64 = b''' @@ -213,3 +225,15 @@ dGhlIGxhenkgZG9nAADjZCTmHjHqggABLxeBCEmxH7bzfQEAAAAABFla self.assertEqual(xz.read(), TEST_STRING) except gemato.exceptions.UnsupportedCompression: raise unittest.SkipTest('xz compression unsupported') + + def test_xz_as_lzma_legacy(self): + """ + Test that the class rejects mislabel files. + """ + if gemato.compression.lzma is None: + raise unittest.SkipTest('xz compression unsupported') + + with io.BytesIO(base64.b64decode(self.BASE64)) as f: + with self.assertRaises(gemato.compression.lzma.LZMAError): + with gemato.compression.open_compressed_file('lzma', f, "rb") as lzma: + lzma.read() |